File: //proc/thread-self/root/usr/lib/python3.9/site-packages/qrcode/__pycache__/main.cpython-39.pyc
a
��;\ 7 � @ sh d dl mZmZmZ d dlmZ d dlZd dlmZ ddd�Z dd� Z
d d
� Z d d � Z G d
d� d�Z
dS )� )� constants�
exceptions�util)� BaseImageN)� bisect_leftc K s t f i |��}|�| � |�� S �N)�QRCode�add_data�
make_image)�data�kwargsZqr� r
�//usr/lib/python3.9/site-packages/qrcode/main.py�make s
r c C s | dk s| dkr t d| ��d S )N� �( z*Invalid version (was %s, expected 1 to 40))�
ValueError)�versionr
r
r �_check_version s �r c C s t | �dkrtd| ��d S )Nr z1Invalid box size (was %s, expected larger than 0))�intr )�sizer
r
r �_check_box_size s �r c C sF | d u r d S t | t�s&tdt| � ��| dk s6| dkrBtd| ��d S )Nz+Invalid mask pattern (was %s, expected int)r � z+Mask pattern should be in range(8) (got %s))�
isinstancer � TypeError�typer )� mask_patternr
r
r �_check_mask_pattern s
��r c @ s� e Zd Zdejddddfdd�Zdd� Zd)d d
�Zd*d d
�Zdd� Z dd� Z
d+dd�Z dd� Z d,dd�Z
d-dd�Zd.dd �Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� ZdS )/r N�
� c C sj t |� |ot|�| _t|�| _t|�| _t|�| _t|� || _|| _|d ur^t |t
�s^J �| � � d S r ) r r r �error_correction�box_size�borderr r �
image_factory�
issubclassr �clear)�selfr r r! r" r# r r
r
r �__init__' s
zQRCode.__init__c C s d| _ d| _d| _g | _dS )z*
Reset the internal data.
Nr )�modules�
modules_count�
data_cache� data_list)r& r
r
r r% : s z QRCode.clear� c C sR t |tj�r| j�|� n.|r6| 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.
)ZminimumN)r r ZQRDatar+ �append�extendZoptimal_data_chunksr* )r& r �optimizer
r
r r C s �zQRCode.add_dataTc C sJ |s| j du r | j| j d� | jdu r8| �d| �� � n| �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)r �best_fitr �makeImpl�best_mask_pattern)r& Zfitr
r
r r U s
z QRCode.makec C s� t | j� | jd d | _d g| j | _t| j�D ]4}d g| j | j|< t| j�D ]}d | j| |< qRq2| �dd� | �| jd d� | �d| jd � | �� | �� | �||� | jdkr�| � |� | j
d u r�t � | j| j
| j�| _
| �| j
|� d S )Nr � r r )r r r) r( �range� setup_position_probe_pattern� setup_position_adjust_pattern�setup_timing_pattern�setup_type_info�setup_type_numberr* r Z create_datar r+ �map_data)r& �testr �row�colr
r
r r3 c s(
�zQRCode.makeImplc C s� t dd�D ]�}|| dks
| j|| kr*q
t dd�D ]�}|| dks4| j|| krTq4d|krt|dkrt|dks�|dks�d|kr�|dkr�|dks�|dks�d|kr�|dkr�d|kr�|dkr�d| j|| || < q4d| j|| || < q4q
d S ) N���� r � � r TF�r6 r) r( )r& r> r? �r�cr
r
r r7 ~ s2 ��������z#QRCode.setup_position_probe_patternc C s� |du r d}t |� t�|�}t�� }| jD ]2}|�|jd� |�t|�||j � |�|� q,t|�}t tj
| j ||�| _ | j dkr�t
�� �|t�| j �ur�| j| j d� | j S )zD
Find the minimum size required to fit in the data.
Nr r �) r0 )r r Zmode_sizes_for_versionZ BitBufferr+ Zput�mode�len�writer ZBIT_LIMIT_TABLEr r r ZDataOverflowErrorr2 )r& r1 Z
mode_sizes�bufferr Z needed_bitsr
r
r r2 � s$
�
zQRCode.best_fitc C sJ d}d}t d�D ]4}| �d|� t�| j�}|dks<||kr|}|}q|S )z7
Find the most efficient mask pattern.
r rA T)r6 r3 r �
lost_pointr( )r& Zmin_lost_point�pattern�irL r
r
r r4 � s zQRCode.best_mask_patternc C s� |du rddl }|j}|�� s&td��| jdu r8| �� | j}|�dd|d d d � t|�D ]J}|�d � t|�D ](}| j | | r�|�d
� qz|�d � qz|�d � qd|�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� rC r z[0m
z[1;47m [40mz z[1;47m [0m
) �sys�stdout�isatty�OSErrorr* r r) rJ r6 r( �flush)r& �outrQ �modcountrE rF r
r
r � print_tty� s$
zQRCode.print_ttyFc sP |du r>ddl }|jdk r8ddl}|�|jj�|j�}n|j}|rR|�� sRtd���jdu rd�� � �j
�dd� dD �}|r�d� � r�|� � � ��fd d
�}t �j
��j
d �D ]�}|r� r�|��j
d k r�|�d
� |�d� t �j
��j
�D ].} ||| �||d | �d > }
|�||
� q�|�r8|�d� |�d� q�|�� 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 )rC r rO c S s g | ]}t �|��d ��qS )�cp437)�sixZint2byte�decode)�.0�coder
r
r �
<listcomp>� s �z&QRCode.print_ascii.<locals>.<listcomp>)� �� �� �� Tc sP � r"�j r"t| |���j kr"dS t| |�dk s>t| |��krBdS �j| | S )Nr r )r" �max�minr( )�x�y��invertrW r& r
r �
get_module� s
� z&QRCode.print_ascii.<locals>.get_modulerC r z [48;5;232mz [38;5;255mz[0m�
)rQ � version_info�codecs� getwriterrR �encodingrS rT r* r r) �reverser6 r" rJ rU ) r& rV Zttyrh rQ rl Zcodesri rE rF �posr
rg r � print_ascii� s>
�
zQRCode.print_asciic K s� t | j� | jdu r | �� |dur4t|t�sRJ �n | j}|du rRddlm} |}|| j | j
| jfi |��}t | j
�D ].}t | j
�D ] }| j | | r�|�
||� q�qv|S )zu
Make an image from the QR Code data.
If the data has not been compiled yet, make it first.
Nr )�PilImage)r r! r* r r$ r r# Zqrcode.image.pilrr r" r) r6 r( Zdrawrect)r& r# r rr ZimrE rF r
r
r r
s&
��zQRCode.make_imagec C s� t d| jd �D ].}| j| d d ur(q|d dk| j| d< qt d| jd �D ].}| jd | d urhqP|d dk| jd |< qPd S )NrA rB rC r rD )r& rE rF r
r
r r9 , s zQRCode.setup_timing_patternc C s� t �| j�}tt|��D ]�}tt|��D ]�}|| }|| }| j| | d urPq(tdd�D ]r}tdd�D ]b}|dks�|dks�|dks�|dks�|dkr�|dkr�d| j|| || < qhd| j|| || < qhqZq(qd S )N���� rC r TF)r Zpattern_positionr r6 rI r( )r& rp rN �jr> r? rE rF r
r
r r8 7 s ��z$QRCode.setup_position_adjust_patternc C s� t �| j�}td�D ]>}| o,||? d@ dk}|| j|d |d | j d d < qtd�D ]>}| ot||? d@ dk}|| j|d | j d d |d < q\d S )N� r rt rA )r ZBCH_type_numberr r6 r( r) )r& r= �bitsrN �modr
r
r r; N s & zQRCode.setup_type_numberc C s" | j d> |B }t�|�}td�D ]f}| o8||? d@ dk}|dk rR|| j| d< q |dk rn|| j|d d< q || j| jd | d< q td�D ]v}| o�||? d@ dk}|dk r�|| jd | j| d < q�|dk r�|| jd d| d d < q�|| jd d| d < q�| | j| jd d< d S )Nrt � r rB rA � )r r Z
BCH_type_infor6 r( r) )r& r= r r rw rN rx r
r
r r: Y s"
zQRCode.setup_type_infoc
C s d}| j d }d}d}t�|�}t|�}tj�| j d dd�D ]�} | dkrR| d8 } | | d f}
|
D ]p} | j| | d u rbd} ||k r�|| |? d@ dk} ||| �r�| } | | j| | < |d8 }|dkrb|d7 }d}qb||7 }|dk s�| j |kr^||8 }| }q>q^q>d S )Nr@ r r r rs rB F)r) r � mask_funcrI rZ ZmovesZxranger( )
r& r r Zincr> ZbitIndexZ byteIndexr{ Zdata_lenr? Z col_rangerF Zdarkr
r
r r< x s6
zQRCode.map_datac C s� | j du r| �� | js | jS t| j�| jd }dg| g| j }dg| j }| jD ]}|�|| | � qV|dg| g| j 7 }|S )z�
Return the QR Code as a multidimensonal array, including the border.
To return the array without a border, set ``self.border`` to 0 first.
NrC F)r* r r" r( rI r- )r& �widthr] Zx_border�moduler
r
r �
get_matrix� s
zQRCode.get_matrix)r, )T)N)N)NFF)N)�__name__�
__module__� __qualname__r ZERROR_CORRECT_Mr' r% r r r3 r7 r2 r4 rX rq r
r9 r8 r; r: r< r~ r
r
r
r r % s, �
5
-r )N)Zqrcoder r r Zqrcode.image.baser rZ Zbisectr r r r r r r
r
r
r �<module> s