File: //home/arjun/projects/env/lib/python3.10/site-packages/qrcode/__pycache__/main.cpython-310.pyc
o
$weN@ � @ s U d dl Z d dlmZ d dlmZmZmZmZmZm Z m
Z
m Z m Z d dl
mZ d dlmZmZmZ d dlmZ d dlmZ eeee Zi Zeeef ed< dd d
�Z d d � Z d
d� Z dd� Z dd� Z G dd� de�Z!e
ded�Z"e
ded�Z#G dd� dee" �Z$dS )� N)� bisect_left) �Dict�Generic�List�
NamedTuple�Optional�Type�TypeVar�cast�overload)�Literal)� constants�
exceptions�util)� BaseImage)�
PyPNGImage�precomputed_qr_blanksc K s t di |��}|�| � |�� S )N� )�QRCode�add_data�
make_image)�data�kwargs�qrr r �D/home/arjun/projects/env/lib/python3.10/site-packages/qrcode/main.py�make s
r c C s t | �dkrtd| � d���d S )Nr zInvalid box size (was z, expected larger than 0)��int�
ValueError��sizer r r �_check_box_size s �r! c C s t | �dk r td| ��d S )Nr z=Invalid border value (was %s, expected 0 or larger than that)r r r r r �
_check_border% s
��r" c C sN | d u rd S t | t�stdt| �� d���| dk s | dkr%td| � d���d S )NzInvalid mask pattern (was z, expected int)r � z(Mask pattern should be in range(8) (got �))�
isinstancer � TypeError�typer )� mask_patternr r r �_check_mask_pattern, s
��r) c C s dd� | D �S )Nc S s g | ]}|d d � �qS �Nr )�.0�rowr r r �
<listcomp>8 s z!copy_2d_array.<locals>.<listcomp>r )�xr r r �
copy_2d_array7 s r/ c @ sd e Zd ZU eed< eed< eed< eed< eed< eed< eed< eed< eed < d
efd d �Zd
S )�ActiveWithNeighbors�NW�N�NE�W�me�E�SW�S�SE�returnc C � | j S r* )r5 ��selfr r r �__bool__F s z ActiveWithNeighbors.__bool__N)�__name__�
__module__� __qualname__�bool�__annotations__r> r r r r r0 ; s
r0 � GenericImage)�bound�GenericImageLocalc @ sz e Zd ZU eed< dZee ed< dej ddddfdee
e fdd�Z e
d efd
d ��Zejd;d d ��Ze
d
d� �Zejdd� �Zdd� Zd<dd�Zd=dd�Zdd� Zdd� Zd>d d �Zd d � Zd>d d!�Zd?d#d$�Zed>ded d e fd%d&��Z e d>de
e d e fd'd&��Z d>d(d&�Z d)ed*ed e fd+d,�Z d-d.� Z d/d0� Z!d1d2� Z"d3d4� Z#d5d6� Z$d7d8� Z%d)ed*ed e&fd9d:�Z'dS )@r �modulesN�_version�
� �
image_factoryc C sb t |� t|� || _t|�| _t|�| _t|�| _|| _|| _|d ur+t |t
�s+J �| � � d S r* ) r! r" �versionr �error_correction�box_size�borderr( rK �
issubclassr �clear)r= rL rM rN rO rK r( r r r �__init__R s
zQRCode.__init__r: c C s | j d u r | �� tt| j �S r* )rH �best_fitr
r r< r r r rL i s
zQRCode.versionc C s$ |d ur
t |�}t�|� || _d S r* )r r �
check_versionrH )r= �valuer r r rL o s
c C r; r* )�
_mask_patternr< r r r r( v s zQRCode.mask_patternc C s t |� || _d S r* )r) rV )r= �patternr r r r( z s
c C s g g| _ d| _d| _g | _dS )z*
Reset the internal data.
r N)rG �
modules_count�
data_cache� data_listr< r r r rQ s
z QRCode.clear� c C sR t |tj�r
| j�|� n|r| j�tj||d�� n | j�t�|�� 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 �QRDatarZ �append�extend�optimal_data_chunksrY )r= r �optimizer r r r � s
zQRCode.add_dataTc C sL |s| j du r| j| j d� | jdu r | �d| �� � dS | �d| j� dS )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.
N��startF)rL rS r( �makeImpl�best_mask_pattern)r= �fitr r r r � s
z QRCode.makec s� � j d d � _� j tv rtt� j �� _n5� fdd�t� j�D �� _� �dd� � �� jd d� � �d� jd � � �� � �� t� j�t� j < � � ||� � j dkr[� �
|� � j d u rkt �
� j � j� j�� _ � �� j |� d S )NrJ � c s g | ]}d g� j �qS r* )rX )r+ �ir< r r r- � s �z#QRCode.makeImpl.<locals>.<listcomp>r r# )rL rX r r/ rG �range� setup_position_probe_pattern� setup_position_adjust_pattern�setup_timing_pattern�setup_type_info�setup_type_numberrY r � create_datarM rZ �map_data)r= �testr( r r< r rd � s(
�
�zQRCode.makeImplc C s� t dd�D ]u}|| dks| j|| krqt dd�D ]_}|| dks)| j|| kr*qd| kr4dkr:n n|dv sbd| krDdkrJn n|dv sbd| krTdkrnn nd| kr`dkrnn n d| j|| || < qd | j|| || < qqd S )
N���� r � > r rt � rJ TF�ri rX rG )r= r, �col�r�cr r r rj � s 0��z#QRCode.setup_position_probe_patternc C s� |du rd}t �|� t �|�}t �� }| jD ]}|�|jd� |�t|�||j � |�|� qt|�}t t j
| j ||�| _ | j dkrIt
�� �|t �| j �urX| j| j d� | j S )zD
Find the minimum size required to fit in the data.
N� rJ �) rb )r rT �mode_sizes_for_version� BitBufferrZ �put�mode�len�writer �BIT_LIMIT_TABLErM rL r �DataOverflowErrorrS )r= rc �
mode_sizes�bufferr � needed_bitsr r r rS � s$
�
zQRCode.best_fitc C sJ d}d}t d�D ]}| �d|� t�| j�}|dks ||kr"|}|}q|S )z7
Find the most efficient mask pattern.
r rs T)ri rd r �
lost_pointrG )r= �min_lost_pointrW rh r� r r r re � s �zQRCode.best_mask_patternc C s� |du r ddl }|j}|�� std��| jdu r | �� | j}|�dd|d d d � t|�D ]%}|�d � t|�D ]}| j | | rL|�d
� q=|�d � q=|�d � q2|�dd|d d d � |�
� 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[1;47m� ru rJ z[0m
z[1;47m [40mz z[1;47m [0m
) �sys�stdout�isatty�OSErrorrY r rX r� ri rG �flush)r= �outr� �modcountrx ry r r r � print_tty s$
zQRCode.print_ttyFc s$ |du rt j}|r|�� std���jdu r��� �j�dd� dD �}|r(d� � r.|�� dtf� ��fdd � }t �j
��j
d
�D ]G}|r]� rS|��j
d k rX|� d � |� d
� t �j
��j
�D ]}|||�||d |�d > }|� || � qg|r�|� d� |� d� qD|� � 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)
Nr� c S s g | ]
}t |f��d ��qS )�cp437)�bytes�decode)r+ �coder r r r- 5 s z&QRCode.print_ascii.<locals>.<listcomp>)� �� �� �� Tr: c sV � r�j rt| |���j krdS t| |�dk s t| |��kr!dS tt�j| | �S )Nrz r )rO �max�minr
r rG )r. �y��invertr� r= r r �
get_module; s
z&QRCode.print_ascii.<locals>.get_moduleru rz z [48;5;232mz [38;5;255mz[0m�
)
r� r� r� r� rY r rX �reverser ri rO r� r� ) r= r� �ttyr� �codesr� rx ry �posr r� r � print_ascii$ s2
zQRCode.print_asciic K � d S r* r �r= rK r r r r r O s zQRCode.make_imagec K r� r* r r� r r r r S s c K s� t | j� | jdu r| �� |durt|t�sJ �n| j}|du r/ddlm}m } |r-|nt
}|| j | j | jfd| j
i|��}|jrkt| j �D ]#}t| j �D ]}|jr\|j||| d� qN| j
| | ri|�||� qNqG|jrr|�� |S )zu
Make an image from the QR Code data.
If the data has not been compiled yet, make it first.
Nr )�Image�PilImage�qrcode_modules)r )r! rN rY r rP r rK �qrcode.image.pilr� r� r rO rX rG �needs_drawrectri �
needs_context�drawrect_context�drawrect�needs_processing�process)r= rK r r� r� �imrx ry r r r r Y s<
��� ��r, rw c C s0 |dko|t | j�k o|dko|t | j| �k S )Nr )r� rG )r= r, rw r r r �is_constrained� s ���zQRCode.is_constrainedc C s� t d| jd �D ]}| j| d d urq|d dk| j| d< qt d| jd �D ]}| jd | d ur4q(|d dk| jd |< q(d S )Nrs rt ru r rv )r= rx ry r r r rl � s �zQRCode.setup_timing_patternc C s� t �| j�}tt|��D ]\}|| }tt|��D ]O}|| }| j| | d ur(qtdd�D ]9}tdd�D ]1}|dksN|dksN|dksN|dksN|dkrZ|dkrZd| j|| || < q4d| j|| || < q4q-qq d S )N���� ru r TF)r �pattern_positionrL ri r� rG )r= r� rh r, �jrw rx ry r r r rk � s( ����z$QRCode.setup_position_adjust_patternc C s� t �| j�}td�D ] }| o||? d@ dk}|| j|d |d | j d d < q
td�D ] }| o:||? d@ dk}|| j|d | j d d |d < q.d S )N� rz r� rs )r �BCH_type_numberrL ri rG rX )r= rq �bitsrh �modr r r rn � s & &�zQRCode.setup_type_numberc C s" | j d> |B }t�|�}td�D ]3}| o ||? d@ dk}|dk r)|| j| d< q|dk r7|| j|d d< q|| j| jd | d< qtd�D ];}| oT||? d@ dk}|dk rf|| jd | j| d < qH|dk rx|| jd d| d d < qH|| jd d| d < qH| | j| jd d< d S )Nr� � rz rt rs � )rM r �
BCH_type_infori rG rX )r= rq r( r r� rh r� r r r rm � s"
zQRCode.setup_type_infoc
C s d}| j d }d}d}t�|�}t|�}t| j d dd�D ]b} | dkr'| d8 } | | d f}
|
D ]8} | j| | d u rhd} ||k rK|| |? d@ dk} ||| �rS| } | | j| | < |d8 }|dkrh|d7 }d}q0||7 }|dk sv| j |kr~||8 }| }nq.q d S ) Nrr rz r# r r� rt TF)rX r � mask_funcr� ri rG )
r= r r( �incr, �bitIndex� byteIndexr� �data_lenrw � col_rangery �darkr r r rp � s@
���zQRCode.map_datac C s� | j du r | �� | js| jS t| j�| jd }dg| g| j }dg| j }| jD ]}|�|ttt |� | � q+|dg| g| j 7 }|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.
Nru F) rY r rO rG r� r^ r
r rB )r= �widthr� �x_border�moduler r r �
get_matrix s
zQRCode.get_matrixc C s` g }t |d |d �D ] }t |d |d �D ]}|�| �||�o(t| j| | �� qq t|� S )Nrz ru )ri r^ r� rB rG r0 )r= r, rw �contextrx ry r r r �active_with_neighbors s &�z QRCode.active_with_neighbors)r: N)r[ )Tr* )NFF)(r? r@ rA � ModulesTyperC rH r r r
�ERROR_CORRECT_Mr rD rR �propertyrL �setterr( rQ r r rd rj rS re r� r� r r r rF rB r� rl rk rn rm rp r� r0 r� r r r r r N s^
�
�
+�� �
) -r r* )%r� �bisectr �typingr r r r r r r r
r �typing_extensionsr �qrcoder
r r �qrcode.image.baser �qrcode.image.purer rB r� r r rC r r! r" r) r/ r0 rD rF r r r r r �<module> s$
,