o hA@sUddlZddlmZddlmZmZmZmZmZm Z m Z ddl m Z m Z mZddlmZddlmZeeeeZiZeeefed<ddd Zd d Zd d ZddZddZGdddeZededZ ededZ!Gdddee Z"dS)N) bisect_left)Generic NamedTupleOptionalTypeVarcastoverloadLiteral) constants exceptionsutil) BaseImage) PyPNGImageprecomputed_qr_blankscKs tdi|}|||S)N)QRCodeadd_data make_image)datakwargsqrrr>/var/www/html/venv/lib/python3.10/site-packages/qrcode/main.pymakes rcCs t|dkrtd|ddS)NrzInvalid box size (was z, expected larger than 0)int ValueErrorsizerrr_check_box_sizes rcCst|dkr td|dS)Nrz=Invalid border value (was %s, expected 0 or larger than that)rrrrr _check_border!s rcCsN|durdSt|tstdt|d|dks|dkr%td|ddS)NzInvalid mask pattern (was z, expected int)rz(Mask pattern should be in range(8) (got )) isinstancer TypeErrortyper) mask_patternrrr_check_mask_pattern(s r&cCsdd|DS)NcSsg|]}|ddqSNr).0rowrrr 4sz!copy_2d_array..r)xrrr copy_2d_array3sr,c@sdeZdZUeed<eed<eed<eed<eed<eed<eed<eed<eed <d efd d Zd S)ActiveWithNeighborsNWNNEWmeESWSSEreturncC|jSr')r2selfrrr__bool__BszActiveWithNeighbors.__bool__N)__name__ __module__ __qualname__bool__annotations__r;rrrrr-7s r- GenericImage)boundGenericImageLocalc@s|eZdZUeed<dZeeed<dej ddddfdee e fddZ e d efd d Zejd;d d Ze d dZejddZddZdddZddZd>d d!Zd?d#d$Ze d>dedd e fd%d&Ze d>de ed efd'd&Zd>d(d&Zd)ed*ed efd+d,Zd-d.Z d/d0Z!d1d2Z"d3d4Z#d5d6Z$d7d8Z%d)ed*ed e&fd9d:Z'dS)@rmodulesN_version  image_factorycCsbt|t|||_t||_t||_t||_||_||_|dur+t |t s+J| dSr') rrversionrerror_correctionbox_sizeborderr%rH issubclassr clear)r:rIrJrKrLrHr%rrr__init__Ns     zQRCode.__init__r7cCs|jdur |tt|jSr')rEbest_fitrrr9rrrrIes  zQRCode.versioncCs$|dur t|}t|||_dSr')rr check_versionrE)r:valuerrrrIks  cCr8r') _mask_patternr9rrrr%rszQRCode.mask_patterncCst|||_dSr')r&rS)r:patternrrrr%vs cCsgg|_d|_d|_g|_dS)z* Reset the internal data. rN)rD modules_count data_cache data_listr9rrrrN{s z QRCode.clearcCsRt|tjr |j|n|r|jtj||dn |jt|d|_dS)a Add data to this QR Code. :param optimize: Data will be split into multiple chunks to optimize the QR size by finding to more compressed modes of at least this length. Set to ``0`` to avoid optimizing at all. )minimumN)r"r QRDatarWappendextendoptimal_data_chunksrV)r:roptimizerrrrs  zQRCode.add_dataTcCsL|s|jdur|j|jd|jdur|d|dS|d|jdS)z Compile the data into a QR Code array. :param fit: If ``True`` (or if a size has not been provided), find the best fit for the data to avoid data overflow errors. NstartF)rIrPr%makeImplbest_mask_pattern)r:fitrrrrs  z QRCode.makecsjdd_jtvrttj_n5fddtjD_ddjdddjdtjtj< ||jdkr[ |j durkt jjj_ j |dS)NrGcsg|]}dgjqSr')rU)r(ir9rrr*s z#QRCode.makeImpl..rr )rIrUrr,rDrangesetup_position_probe_patternsetup_position_adjust_patternsetup_timing_patternsetup_type_infosetup_type_numberrVr create_datarJrWmap_data)r:testr%rr9rras(        zQRCode.makeImplcCstddD]u}||dks|j||krqtddD]_}||dks)|j||kr*qd|kr4dkr:nn|dvsbd|krDdkrJnn|dvsbd|krTdkrnnnd|kr`dkrnnn d|j||||<qd |j||||<qqdS) Nr>rrqrGTFrfrUrD)r:r)colrcrrrrgs  0z#QRCode.setup_position_probe_patterncCs|durd}t|t|}t}|jD]}||jd|t|||j||qt|}t tj |j |||_ |j dkrIt |t|j urX|j|j d|j S)zD Find the minimum size required to fit in the data. NrG)r_)r rQmode_sizes_for_version BitBufferrWputmodelenwriterBIT_LIMIT_TABLErJrIr DataOverflowErrorrP)r:r` mode_sizesbufferr needed_bitsrrrrPs$     zQRCode.best_fitcCsJd}d}tdD]}|d|t|j}|dks||kr"|}|}q|S)z7 Find the most efficient mask pattern. rrpT)rfrar lost_pointrD)r:min_lost_pointrTrerrrrrbs   zQRCode.best_mask_patterncCs|dur ddl}|j}|std|jdur||j}|dd|dddt|D]%}|d t|D]}|j ||rL|d q=|d q=|d q2|dd|ddd| dS) zz Output the QR Code only using TTY colors. If the data has not been compiled yet, make it first. Nr Not a ttyz rrrGz z z z  ) sysstdoutisattyOSErrorrVrrUr~rfrDflush)r:outrmodcountrurvrrr print_ttys$        zQRCode.print_ttyFc s$|durtj}|r|stdjdurjdddD}|r(dr.|dtffdd }t j j d D]G}|r]rS|j d krX| d | d t j j D]}|||||d |d >}| ||qg|r| d| dqD| dS)z Output the QR Code using ASCII characters. :param tty: use fixed TTY color codes (forces invert=True) :param invert: invert the ASCII characters (solid <-> transparent) NrcSsg|] }t|fdqS)cp437)bytesdecode)r(coderrrr*/sz&QRCode.print_ascii..)Tr7csVrjrt||jkrdSt||dkst||kr!dSttj||S)Nrwr)rLmaxminrrrD)r+yinvertrr:rr get_module5s z&QRCode.print_ascii..get_modulerrrwz z z ) rrrrrVrrUreverserrfrLr~r) r:rttyrcodesrrurvposrrr print_asciis2       zQRCode.print_asciicKdSr'rr:rHrrrrrIzQRCode.make_imagecKrr'rrrrrrNrcKs$|ds|ds|ds|dr|jtjkrtdt|j|jdur,||dur8t |t s7Jn|j }|durMddl m }m}|rK|nt}||j|j|jfd |ji|}|jrt|jD]#}t|jD]}|jrz|j|||d ql|j||r|||qlqe|jr||S) zu Make an image from the QR Code data. If the data has not been compiled yet, make it first. embedded_image_pathembedded_imageembeded_image_path embeded_imagezOError correction level must be ERROR_CORRECT_H if an embedded image is providedNr)ImagePilImageqrcode_modules)r)getrJr ERROR_CORRECT_HrrrKrVrrMr rHqrcode.image.pilrrrrLrUrDneeds_drawrectrf needs_contextdrawrect_contextdrawrectneeds_processingprocess)r:rHrrrimrurvrrrrSsT     r)rtcCs0|dko|t|jko|dko|t|j|kS)Nr)r}rD)r:r)rtrrris_constraineds zQRCode.is_constrainedcCstd|jdD]}|j|ddurq|ddk|j|d<qtd|jdD]}|jd|dur4q(|ddk|jd|<q(dS)Nrprqrrrrs)r:rurvrrrriszQRCode.setup_timing_patterncCst|j}tt|D]\}||}tt|D]O}||}|j||dur(qtddD]9}tddD]1}|dksN|dksN|dksN|dksN|dkrZ|dkrZd|j||||<q4d|j||||<q4q-qq dS)NrrrTF)r pattern_positionrIrfr}rD)r:rrer)jrtrurvrrrrhs( z$QRCode.setup_position_adjust_patterncCst|j}tdD]}| o||?d@dk}||j|d|d|jdd<q tdD]}| o:||?d@dk}||j|d|jdd|d<q.dS)Nrwrrp)r BCH_type_numberrIrfrDrU)r:rnbitsremodrrrrks  & &zQRCode.setup_type_numbercCs"|jd>|B}t|}tdD]3}| o||?d@dk}|dkr)||j|d<q|dkr7||j|dd<q||j|jd|d<qtdD];}| oT||?d@dk}|dkrf||jd|j|d<qH|dkrx||jdd|dd<qH||jdd|d<qH| |j|jdd<dS)Nrrwrqrp )rJr BCH_type_inforfrDrU)r:rnr%rrrerrrrrjs"   zQRCode.setup_type_infoc Csd}|jd}d}d}t|}t|}t|jdddD]b} | dkr'| d8} | | df} | D]8} |j|| durhd} ||krK|||?d@dk} ||| rS| } | |j|| <|d8}|dkrh|d7}d}q0||7}|dksv|j|kr~||8}| }nq.qdS) Nrorwr rrrqTF)rUr mask_funcr}rfrD) r:rr%incr)bitIndex byteIndexrdata_lenrt col_rangervdarkrrrrms@    zQRCode.map_datacCs|jdur ||js|jSt|j|jd}dg|g|j}dg|j}|jD]}||ttt||q+|dg|g|j7}|S)z Return the QR Code as a multidimensional array, including the border. To return the array without a border, set ``self.border`` to 0 first. NrrF) rVrrLrDr}r[rlistr?)r:widthrx_bordermodulerrr get_matrixs   zQRCode.get_matrixcCs`g}t|d|dD] }t|d|dD]}||||o(t|j||qq t|S)Nrwrr)rfr[rr?rDr-)r:r)rtcontextrurvrrractive_with_neighborss &zQRCode.active_with_neighbors)r7N)rX)Tr')NFF)(r<r=r> ModulesTyper@rErrr ERROR_CORRECT_Mr$rArOpropertyrIsetterr%rNrrrargrPrbrrrr rrCr?rrirhrkrjrmrr-rrrrrrJsh          +  3  )rr')#rbisectrtypingrrrrrrr qrcoder r r qrcode.image.baser qrcode.image.purerrr?rrdictrr@rrrr&r,r-rArCrrrrrs" $