U ;gN@@sUddlZddlmZddlmZmZmZmZmZm Z m Z m Z m Z ddl mZddlmZmZmZddlmZddlmZeeeeZiZeeefed<dd d Zd d Zd dZddZddZ GdddeZ!e dedZ"e dedZ#Gdddee"Z$dS)N) bisect_left) DictGenericList NamedTupleOptionalTypeTypeVarcastoverload)Literal) constants exceptionsutil) BaseImage) PyPNGImageprecomputed_qr_blankscKstf|}|||SN)QRCodeadd_data make_image)datakwargsqrr//tmp/pip-unpacked-wheel-ij2mbwps/qrcode/main.pymakes  rcCs t|dkrtd|ddS)NrzInvalid box size (was z, expected larger than 0)int ValueErrorsizerrr_check_box_size s r"cCst|dkrtd|dS)Nrz=Invalid border value (was %s, expected 0 or larger than that)rr rrr _check_border%s r#cCsN|dkr dSt|ts*tdt|d|dks:|dkrJtd|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|]}|ddqSrr).0rowrrr 8sz!copy_2d_array..r)xrrr copy_2d_array7sr/c@sdeZdZUeed<eed<eed<eed<eed<eed<eed<eed<eed <ed d d Zd S)ActiveWithNeighborsZNWNZNEWmeEZSWSZSEreturncCs|jSr)r3selfrrr__bool__FszActiveWithNeighbors.__bool__N)__name__ __module__ __qualname__bool__annotations__r:rrrrr0;s r0 GenericImage)boundGenericImageLocalc@speZdZUeed<dZeeed<dej ddddfee e dddZ e ed d d Zejdd d d Ze d dZejddZddZd;ddZdd d!Zd?d#d$Zed@ede d%d&d'ZedAe eed%d(d'ZdBd)d'Zeeed*d+d,Zd-d.Z d/d0Z!d1d2Z"d3d4Z#d5d6Z$d7d8Z%eee&d*d9d:Z'dS)CrmodulesN_version ) image_factorycCsbt|t|||_t||_t||_t||_||_||_|dk rVt |t sVt | dSr) r"r#versionrerror_correctionbox_sizeborderr)rG issubclassrAssertionErrorclear)r9rHrIrJrKrGr)rrr__init__Rs    zQRCode.__init__r6cCs|jdkr|tt|jSr)rDbest_fitr rr8rrrrHis zQRCode.versioncCs$|dk rt|}t|||_dSr)rr check_versionrD)r9valuerrrrHos cCs|jSr) _mask_patternr8rrrr)vszQRCode.mask_patterncCst|||_dSr)r*rS)r9patternrrrr)zscCsgg|_d|_d|_g|_dS)z* Reset the internal data. rN)rC modules_count data_cache data_listr8rrrrNsz QRCode.clearcCsRt|tjr|j|n.|r6|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. )ZminimumN)r&rZQRDatarWappendextendZoptimal_data_chunksrV)r9roptimizerrrrs  zQRCode.add_dataTcCsJ|s|jdkr|j|jd|jdkr8|d|n|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)rHrPr)makeImplbest_mask_pattern)r9Zfitrrrrs  z QRCode.makecsjdd_jtkr,ttj_njfddtjD_ddjdddjdtjtj< ||jdkr |j dkrt jjj_ j |dS)NrFcsg|]}dgjqSr)rU)r+ir8rrr-sz#QRCode.makeImpl..rr$)rHrUrr/rCrangesetup_position_probe_patternsetup_position_adjust_patternsetup_timing_patternsetup_type_infosetup_type_numberrVrZ create_datarIrWmap_data)r9testr)rr8rr^s,       zQRCode.makeImplcCstddD]}||dks |j||kr*q tddD]}||dks4|j||krTq4d|krhdkrtnn|dksd|krdkrnn|dksd|krdkrnn0d|krdkrnnd|j||||<q4d |j||||<q4q dS) Nr>rrlrFTFrbrUrC)r9r,colrcrrrrcs@    z#QRCode.setup_position_probe_patterncCs|dkr d}t|t|}t}|jD]2}||jd|t|||j||q.t|}t tj |j |||_ |j dkrt |t|j k r|j|j d|j S)zD Find the minimum size required to fit in the data. NrF)r\)rrQZmode_sizes_for_versionZ BitBufferrWputmodelenwriterZBIT_LIMIT_TABLErIrHrZDataOverflowErrorrP)r9r]Z mode_sizesbufferrZ needed_bitsrrrrPs(      zQRCode.best_fitcCsJd}d}tdD]4}|d|t|j}|dks<||kr|}|}q|S)z7 Find the most efficient mask pattern. rrkT)rbr^r lost_pointrC)r9Zmin_lost_pointrTraryrrrr_s   zQRCode.best_mask_patterncCs|dkrddl}|j}|s&td|jdkr8||j}|dd|dddt|D]J}|d t|D](}|j ||r|d qz|d qz|d qd|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 rmrFz z z z  ) sysstdoutisattyOSErrorrVrrUrwrbrCflush)r9outr|modcountrprqrrr print_ttys$       zQRCode.print_ttyFc s&|dkrtj}|r"|s"tdjdkr4jdddD}|rPdr\|tdfdd }t j j d D]}|rr|j d kr| d | d t j j D].}|||||d |d >}| ||q|r| d| dq| 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) NrzcSsg|]}t|fdqS)cp437)bytesdecode)r+coderrrr-5sz&QRCode.print_ascii..)Tr6csVr"jr"t||jkr"dSt||dks>t||krBdSttj||S)Nrrr)rKmaxminr rrC)r.yinvertrr9rr get_module;s z&QRCode.print_ascii..get_modulermrrz z z ) r|r}r~rrVrrUreverserrbrKrwr) r9rttyrcodesrrprqposrrr print_ascii$s2      zQRCode.print_ascii)rGr7cKsdSrrr9rGrrrrrOszQRCode.make_imagecKsdSrrrrrrrSscKst|j|jdkr||dk r4t|ts^tn*|j}|dkr^ddlm }m }|rZ|nt }||j |j |jfd|ji|}|jrt|j D]F}t|j D]6}|jr|j|||dq|j||r|||qq|jr||S)zu Make an image from the QR Code data. If the data has not been compiled yet, make it first. Nr)ImagePilImageZqrcode_modules)r)r"rJrVrrLrrMrGZqrcode.image.pilrrrrKrUrCZneeds_drawrectrbZ needs_contextZdrawrect_contextZdrawrectZneeds_processingprocess)r9rGrrrZimrprqrrrrYs8   )r,ror7cCs0|dko.|t|jko.|dko.|t|j|kS)Nr)rvrC)r9r,rorrris_constraineds zQRCode.is_constrainedcCstd|jdD].}|j|ddk r(q|ddk|j|d<qtd|jdD].}|jd|dk rhqP|ddk|jd|<qPdS)Nrkrlrmrrn)r9rprqrrrreszQRCode.setup_timing_patterncCst|j}tt|D]}||}tt|D]}||}|j||dk rPq0tddD]r}tddD]b}|dks|dks|dks|dks|dkr|dkrd|j||||<qhd|j||||<qhqZq0qdS)NrmrTF)rZpattern_positionrHrbrvrC)r9rrar,jrorprqrrrrds. z$QRCode.setup_position_adjust_patterncCst|j}tdD]>}| o,||?d@dk}||j|d|d|jdd<qtdD]>}| ot||?d@dk}||j|d|jdd|d<q\dS)Nrrrrk)rZBCH_type_numberrHrbrCrU)r9ribitsramodrrrrgs  & zQRCode.setup_type_numbercCs"|jd>|B}t|}tdD]f}| o8||?d@dk}|dkrR||j|d<q |dkrn||j|dd<q ||j|jd|d<q tdD]v}| o||?d@dk}|dkr||jd|j|d<q|dkr||jdd|dd<q||jdd|d<q| |j|jdd<dS)Nrrrrlrk )rIrZ BCH_type_inforbrCrU)r9rir)rrrarrrrrfs"   zQRCode.setup_type_infoc Csd}|jd}d}d}t|}t|}t|jdddD]} | dkrN| d8} | | df} | D]p} |j|| dkr^d} ||kr|||?d@dk} ||| r| } | |j|| <|d8}|dkr^|d7}d}q^||7}|dks|j|krZ||8}| }q:qZq:dS)Nrjrrr$rrrlF)rUr mask_funcrvrbrC) r9rr)incr,ZbitIndexZ byteIndexrdata_lenroZ col_rangerqZdarkrrrrhs6    zQRCode.map_datacCs|jdkr||js|jSt|j|jd}dg|g|j}dg|j}|jD] }||ttt||qV|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. NrmF) rVrrKrCrvrYr rr>)r9widthrZx_bordermodulerrr get_matrix s   zQRCode.get_matrixcCs`g}t|d|dD]@}t|d|dD](}||||oPt|j||q,qt|S)Nrrrm)rbrYrr>rCr0)r9r,rocontextrprqrrractive_with_neighborss (zQRCode.active_with_neighbors)rX)T)N)N)NFF)N)N)N)(r;r<r= ModulesTyper?rDrrr ZERROR_CORRECT_Mrr@rOpropertyrHsetterr)rNrrr^rcrPr_rrr r rrBr>rrerdrgrfrhrr0rrrrrrNsZ         + )  -r)N)%r|bisectrtypingrrrrrrr r r Ztyping_extensionsr Zqrcoder rrZqrcode.image.baserZqrcode.image.purerr>rrrr?rr"r#r*r/r0r@rBrrrrrs$ ,