File: //usr/local/lib/python3.10/dist-packages/pydantic/v1/__pycache__/_hypothesis_plugin.cpython-310.pyc
o
|��g�9 � @ s U d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z m
Z
m Z m Z m
Z
mZ ddlmZ ddlZddlZddlZddlmZ zddlZW n eyT Y n-w dedefdd �Ze� ej e� � � e�� e� ej e�!d
j"ej#d d d
�e� � � e��� e� ej$e�%dd� e&e'e��D ��� d�(ej)j*ej)j+ej)j,ej)j-ej)j.ej)j/f��0ej)j1d��0ej)j2d��0ej)j3d�Z4e� ej)j5e�6e�%e&ej)j7��e�8e�9dd�e�9dd�e�9dd�e�:� e�;dd�B e�;dd��<dj"�B �ej#e4d d
��� dedefdd�Z=dZ>e� ej?ej#d�(e>�d d
��<e=�� e� ej@ejAdd �� e� ejBejAd d �� e� ejCejAd d �� e� ejDejAd d �� e� ejEe�F� �<ejE�� e� ejGe�H� �<ejG�� e� ejIe�J� � e� ejKe�LejM�e�LejN�B � e� ejOe�LejP�e�LejQ�B � e� ejRe�S� � e� ejTe�H� � e� ejUejVejW�X� ejYdd � d!�� e� ejZejVejW�X� ejYdd � d"�� i Z[e
e\e e\gej]f f e^d#< ed$e ej_j` de ej_j` fd%d&��Zaed$ej_jbdej_jbfd'd&��Zad$e e ej_j` ej_jbf de e ej_j` ej_jbf fd(d&�Zad$e e\ej_jbf de e d)ej]f ge d)ej]f f fd*d+�Zcecejd�d,d-� �Zeecejf�d.d/� �Zgecejh�d0d1� �Ziecejj�d2d3� �Zkecejl�d4d5� �Zmecejn�d6d7� �Zoecejp�d8d9� �Zqerej_js�D ]Zteaet� �qteaej__ae� ejuee� dS ):a�
Register Hypothesis strategies for Pydantic custom types.
This enables fully-automatic generation of test data for most Pydantic classes.
Note that this module has *no* runtime impact on Pydantic itself; instead it
is registered as a setuptools entry point and Hypothesis will import it if
Pydantic is installed. See also:
https://hypothesis.readthedocs.io/en/latest/strategies.html#registering-strategies-via-setuptools-entry-points
https://hypothesis.readthedocs.io/en/latest/data.html#hypothesis.strategies.register_type_strategy
https://hypothesis.readthedocs.io/en/latest/strategies.html#interaction-with-pytest-cov
https://docs.pydantic.dev/usage/types/#pydantic-types
Note that because our motivation is to *improve user experience*, the strategies
are always sound (never generate invalid data) but sacrifice completeness for
maintainability (ie may be unable to generate some tricky but valid data).
Finally, this module makes liberal use of `# type: ignore[<code>]` pragmas.
This is because Hypothesis annotates `register_type_strategy()` with
`(T, SearchStrategy[T])`, but in most cases we register e.g. `ConstrainedInt`
to generate instances of the builtin `int` type which match the constraints.
� N)�Fraction)�Callable�Dict�Type�Union�cast�overload)�lenient_issubclass�s�returnc C s, z
t j| dd� W dS t jy Y dS w )NF)�check_deliverabilityT)�email_validator�validate_email�EmailNotValidError�r
� r �I/usr/local/lib/python3.10/dist-packages/pydantic/v1/_hypothesis_plugin.py�is_valid_email: s �r z{} <{}>z"[A-Za-z0-9_]+( [A-Za-z0-9_]+){0,5}T�� fullmatchc C s( g | ]}|� d �sttjd|� ���qS )�_zmath.)�
startswithr �pydantic�PyObject)�.0�namer r r �
<listcomp>S s ( r �|z#(?:(\d\d?(?:\.\d+)?|100(?:\.0+)?)%)z+(?:(0(?:\.\d+)?|1(?:\.0+)?|\.\d+|\d{1,2}%))zB(?:((?:\d|\d\d|[01]\d\d|2[0-4]\d|25[0-4])(?:\.\d+)?|255(?:\.0+)?))� � �d z{}%� card_numberc C sX dD ]%}t �t�� tj�| | � | | W d � S 1 s"w Y qtd��)N�
0123456789� Unreachable)�
contextlib�suppress� Exceptionr �PaymentCardNumber�validate_luhn_check_digit�AssertionError)r! �digitr r r �add_luhn_digitz s &�r+ )z
4[0-9]{14}z5[12345][0-9]{13}z3[47][0-9]{12}z[0-26-9][0-9]{10,17})�version� � � ��days)� min_value)� max_value� RESOLVERS�typc C � d S �Nr �r5 r r r � _registered� � r9 c C r6 r7 r r8 r r r r9 � r: c C sR t jj�| � t�� D ]\}}t| |�r t�| || �� | S q t d| �d���)Nz
Unknown type z has no resolver to register)
r �types�_DEFINED_TYPES�addr4 �items�
issubclass�st�register_type_strategy�NotImplementedError)r5 � supertype�resolverr r r r9 � s
�.c s � fdd�}|S )Nc s | t vsJ �| t � < | S r7 )r4 )�fr8 r r �inner� s zresolves.<locals>.innerr )r5 rF r r8 r �resolves� s rG c C s� z| j d u r
t�� nt�| j �}W n( ty9 tjddd�}tjt�t�� t�� t� � |t�
� �dd� d�}Y nw t | dd �}tj t
|tj�rL| j jntj|t�� t�� t� dd�B t�� d �S )
NF)�allow_infinity� allow_nanc S s t �| �t �t �� | �B S r7 )r@ �lists� dictionaries�text��xr r r �<lambda>� s z resolve_json.<locals>.<lambda>)�base�extend�
inner_typer � )� ensure_ascii�indent� sort_keys)rR r@ �none� from_typer& �floats� recursive�one_of�booleans�integersrL �getattr�buildsr r � BaseModel�json�dumps)�clsrF �finiterR r r r � resolve_json� s" "
�� �re c C s� | j pd}| j}| jstj||d�S d�|dkr |d nd|p ddkr'|d nd�}|dkr5d|� d�}n|dkr@d |� d
�}n |dksFJ �d |� d �}tj|�� d
d�S )Nr ��min_size�max_size� {{{},{}}}� � �\W.�\Wr �\W(.�\W)?�(\W(.�\W)?)?Tr )�
min_length�
max_length�strip_whitespacer@ �binary�format�
from_regex�encode)rc rg rh �repeats�patternr r r �resolve_conbytes� s
� r{ c s� � j }� j}� jd ur|d u sJ d��� j}� jd ur&|d u s#J d��� j}tj||d� jd�}� jd ur>|�� fdd��}� jd urL|�� fdd��}|S )N� Set `gt` or `ge`, but not both� Set `lt` or `le`, but not bothF)rI �placesc s
| � j k S r7 )�lt��d�rc r r rO �
z$resolve_condecimal.<locals>.<lambda>c s
� j | k S r7 )�gtr� r� r r rO r� )�ge�ler� r r@ �decimals�decimal_places�filter)rc r2 r3 r
r r� r �resolve_condecimal s
r� c s� � j }� j}d}d}� jd ur |d u sJ d��� j}d}� jd ur.|d u s)J d��� j}d}� jd u r=tj||||dd�S |d urOt�|� j �}|rO|d }|d urj|� jks\J d��t� |� j �}|rj|d }t�
||�� � fdd ��S )
NFr| Tr} )� exclude_min� exclude_maxrI r z:Cannot build model with max value smaller than multiple ofc �
| � j S r7 �� multiple_ofrM r� r r rO 6 r� z"resolve_confloat.<locals>.<lambda>) r� r� r� r r� r@ rY �math�ceil�floorr] �map)rc r2 r3 r� r� r r� r �resolve_confloat s0
r� c s� � j }� j}� jd ur|d u sJ d��� jd }� jd ur*|d u s%J d��� jd }� jd u s4� jdkr:t�||�S |d urJt�t |�t � j� �}|d urZt�
t |�t � j� �}t�||�� � fdd��S )Nr| r r} c r� r7 r� rM r� r r rO M r� z resolve_conint.<locals>.<lambda>) r� r� r� r r� r@ r] r� r� r r� r� �rc r2 r3 r r� r �resolve_conint9 s
r� c C s� | j d ur| jd u sJ d��| j }n| jd ur!| jtjdd� }ntjj}| jd ur7| jd u s3J d��| j}n| jd urF| jtjdd� }ntjj}t �
||�S )Nr| r r0 r} ) r� r� �datetime� timedelta�date�minr� r �maxr@ �datesr� r r r �resolve_condateP s
r� c s( | j pd�| j� | jd u r| jstj�� d�S | jd ur-t�| j�}| jr,|�dd� �}nF| jrsd��dkr:�d nd� p>ddkrE� d nd�}�dkrVt�d|� d ��}n �d
krdt�d |� d ��}n�dksjJ �t�d
|� d��}�dkr}� d u r}|S � d u r�|��fdd��S |�� �fdd��S )Nr rf c S s | | � � kS r7 )�stripr r r r rO n � z resolve_constr.<locals>.<lambda>ri rj rk rl rm r rn ro rp rq c s � t | �kS r7 ��lenr )rg r r rO r� c s �t | � ko � kS S r7 r� r �rh rg r r rO � s ) rr rs �regexrt r@ rL rw r� rv )rc �strategyry r r� r �resolve_constrc s2
�� r� )v�__doc__r$ r� � ipaddressra r� � fractionsr �typingr r r r r r �hypothesis.strategies�
strategiesr@ r �pydantic.color�pydantic.types�pydantic.v1.utilsr r
� ImportError�str�boolr rA �EmailStr�emailsr� � NameEmailr_ rv rw r � sampled_from�sorted�vars�join�color� r_hex_short�
r_hex_long�r_rgb�r_rgba�r_hsl�r_hsla�replace�_r_sl�_r_alpha�_r_255�_color_regexes�Colorr[ �COLORS_BY_NAME�tuplesr] rW rY r� r+ �
card_patternsr' �UUID1�uuids�UUID3�UUID4�UUID5� SecretBytesru � SecretStrrL �
IPvAnyAddress� ip_addresses�IPvAnyInterfacerX �
IPv4Interface�
IPv6Interface�
IPvAnyNetwork� IPv4Network� IPv6Network�
StrictBoolr\ � StrictStr�
FutureDater� r� �todayr� �PastDater4 �type�SearchStrategy�__annotations__r; �Tr9 �ConstrainedNumberMetarG � JsonWrapperre �ConstrainedBytesr{ �ConstrainedDecimalr� �ConstrainedFloatr� �ConstrainedIntr� �ConstrainedDater� �ConstrainedStrr� �listr< r5 �Jsonr r r r �<module> s� � ��
�� �� �
&� �� ���(( $ �
��
�