File: //home/arjun/projects/env/lib64/python3.10/site-packages/faker/__pycache__/proxy.cpython-310.pyc
o
'wec1 � @ s� d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZm Z m
Z
m Z m Z m
Z
mZmZmZmZ ddlmZ ddlmZ ddlmZ dd lmZmZ dd
lmZ dd lm Z d Z ed
�Z G dd� d�Z G dd� d�Z G dd� d�Z!dS )� N)� OrderedDict)�Random)
�Any�Callable�Dict�List�Optional�Pattern�Sequence�Tuple�TypeVar�Union� )�DEFAULT_LOCALE)�UniquenessException)�Factory)� Generator�random)�SeedType)�choices_distributioni� �RetTypec sV e Zd ZU dZe�d�Zeed< dd� e e
�D �Z dBde e
eee eee
eef f f d e ee d
e e
d e ee d ed
eddfdd�Z� fdd�Zdede
fdd�Zdedef� fdd� Zdedefdd�Zi fdedd fdd �Zd eddfd d �Z e dCd!d"��Z e dDd$d%��Z d&ede fd'd(�Z!d)d*� Z"d+d,� Z#d&ede$ee e ee f fd-d.�Z%e&dEd/e e' ddfd0d1��Z(dEd/e e' ddfd2d3�Z)dEded/e e' ddfd4d5�Z*e de+fd6d7��Z,e,j-d8e+ddfd9d7��Z,e dee fd:d;��Z.e de ee
eef fd<d=��Z/e dee
fd>d?��Z0dee$ee
f fd@dA�Z1� Z2S )F�Fakerz2Proxy class capable of supporting multiple localesz^_cached_\w*_mapping$�
cache_patternc C s" g | ]
}|� d �s|dvr|�qS )�__)�seed�
seed_instancer ��
startswith��.0�attr� r! �D/home/arjun/projects/env/lib/python3.10/site-packages/faker/proxy.py�
<listcomp> s
�zFaker.<listcomp>NT�locale� providers� generator�includes�
use_weighting�config�returnc K sP t � | _d | _t| �| _t| �| _t|t�r |� dd�g}ngt|t
t t f�rLg }|D ] }t|t�s;t
dt|�� d���|� dd�} | |vrJ|�| � q*n9t|t �r�tdd� |�� D ��s^J �t � }
|�� D ]\} } | � dd�}
| |
|
< qet
|
�� �}t
|
�� �| _ntg}|D ]}tj||||fd|i|��| j|< q�|| _t
| j�� �| _d S )N�-�_z The locale "z" must be a string.c s s � | ]
}t |ttf�V qd S �N)�
isinstance�int�float)r �vr! r! r" � <genexpr>: s � z!Faker.__init__.<locals>.<genexpr>r( )r � _factory_map�_weights� UniqueProxy�
_unique_proxy�
OptionalProxy�_optional_proxyr. �str�replace�list�tuple�set� TypeError�append�all�values�items�keysr r �create�_locales�
_factories)�selfr$ r% r& r' r( r) �locales�code� final_locale�odict�kr1 �keyr! r! r" �__init__ sL
��
��� zFaker.__init__c s< t tt| ��� �}| jD ]
}|dd� t|�D �O }q t|�S )Nc S s h | ] }|� d �s|�qS )r, r r r! r! r" � <setcomp>U � z Faker.__dir__.<locals>.<setcomp>)r= �superr �__dir__� factories�dir�sorted)rG �
attributes�factory�� __class__r! r" rR R s
z
Faker.__dir__c C s | j |�dd� S )Nr+ r, )r3 r: )rG r$ r! r! r" � __getitem__X s zFaker.__getitem__r c s |dkr
d}t |��t� �|�S )z�
Handles the "attribute resolution" behavior for declared members of this proxy class
The class method `seed` cannot be called from an instance.
:param attr: attribute name
:return: the appropriate attribute
r zZCalling `.seed()` on instances is deprecated. Use the class method `Faker.seed()` instead.)r> rQ �__getattribute__)rG r �msgrX r! r" r[ [ s zFaker.__getattribute__c C sh t | j�dkrt| jd |�S || jv r d| }t|��| j�|�r*d| }t|��| �|�}t||�S )z�
Handles cache access and proxying behavior
:param attr: attribute name
:return: the appropriate attribute
r r zBProxying calls to `%s` is not implemented in multiple locale mode.z$Cached attribute `%s` does not exist) �lenrF �getattr�generator_attrs�NotImplementedErrorr �match�AttributeError�_select_factory)rG r r\ rW r! r! r" � __getattr__j s
zFaker.__getattr__�memodictc st | j }|�|�� t�| j�� _t�| j�� _t�| j�� _t�| j�� _t| �� _ � fdd�| j j
� � D �� j _
� S )Nc s i | ]}|� j jh�qS r! )r6 � _sentinel)r rL ��resultr! r" �
<dictcomp>� s z&Faker.__deepcopy__.<locals>.<dictcomp>) rY �__new__�copy�deepcopyrE rF r3 r4 r5 r6 �_seenrC )rG re �clsr! rg r" � __deepcopy__} s
zFaker.__deepcopy__�statec C � | j �|� d S r- ��__dict__�update�rG rp r! r! r" � __setstate__� � zFaker.__setstate__r5 c C � | j S r- )r6 �rG r! r! r" �unique� � z Faker.uniquer7 c C rx r- )r8 ry r! r! r" �optional� r{ zFaker.optional� method_namec C sb | � |�\}}t|�dkrd|��}t|��t|�dkr |d S |r*| �||�}|S | �|�}|S )z�
Returns a random factory that supports the provider method
:param method_name: Name of provider method
:return: A factory that supports the provider method
r z"No generator object has attribute r )�_map_provider_methodr] rb � _select_factory_distribution�_select_factory_choice)rG r} rS �weightsr\ rW r! r! r" rc � s
�zFaker._select_factoryc C s t ||tdd�d S )Nr )�lengthr )r r )rG rS r� r! r! r" r � s z"Faker._select_factory_distributionc C s
t �|�S r- )r �choice)rG rS r! r! r" r� � �
z Faker._select_factory_choicec s� d� � d�}t | |�rt| |�S | jr0� fdd�t| j| j�D �}t|� \}}t|�t|�f}n� fdd�| jD �}|df}t| ||� |S )ad
Creates a 2-tuple of factories and weights for the given provider method name
The first element of the tuple contains a list of compatible factories.
The second element of the tuple contains a list of distribution weights.
:param method_name: Name of provider method
:return: 2-tuple (factories, weights)
�_cached_�_mappingc s" g | ]
\}}t |� �r||f�qS r! ��hasattr)r rW �weight�r} r! r" r# � s ��z.Faker._map_provider_method.<locals>.<listcomp>c s g | ] }t |� �r|�qS r! r� )r rW r� r! r" r# � rP N)r� r^ r4 �ziprS r; �setattr)rG r} r �valuerS r� �mappingr! r� r" r~ � s
� zFaker._map_provider_methodr c C s t �|� dS )zs
Hashables the shared `random.Random` object across all factories
:param seed: seed value
N)r r )rn r r! r! r" r � s z
Faker.seedc C s | j D ]}|�|� qdS )zr
Creates and seeds a new `random.Random` object for each factory
:param seed: seed value
N)rF r )rG r rW r! r! r" r � s
�zFaker.seed_instancec C s | j |�dd� �|� dS )z�
Creates and seeds a new `random.Random` object for the factory of the specified locale
:param locale: locale string
:param seed: seed value
r+ r, N)r3 r: r )rG r$ r r! r! r" � seed_locale� s zFaker.seed_localec C s& t | j�dkr
| jd jS d}t|��)a
Proxies `random` getter calls
In single locale mode, this will be proxied to the `random` getter
of the only internal `Generator` object. Subclasses will have to
implement desired behavior in multiple locale mode.
r r zJProxying `random` getter calls is not implemented in multiple locale mode.�r] rF r r` )rG r\ r! r! r" r � s
z Faker.randomr� c C s* t | j�dkr|| jd _dS d}t|��)a
Proxies `random` setter calls
In single locale mode, this will be proxied to the `random` setter
of the only internal `Generator` object. Subclasses will have to
implement desired behavior in multiple locale mode.
r r zJProxying `random` setter calls is not implemented in multiple locale mode.Nr� )rG r� r\ r! r! r" r � s
c C s
t | j�S r- )r; rE ry r! r! r" rH s
z
Faker.localesc C rx r- )r4 ry r! r! r" r� r{ z
Faker.weightsc C rx r- )rF ry r! r! r" rS r{ zFaker.factoriesc C s t | j�� �S r- )r; r3 rB ry r! r! r" rB s z Faker.items)NNNNT)r* r5 )r* r7 r- )3�__name__�
__module__� __qualname__�__doc__�re�compiler r �__annotations__rT r r_ r r
r9 r
r r/ r0 r �boolr rN rR rZ r[ rd ro rv �propertyrz r| r rc r r� r r~ � classmethodr r r r� r r �setterrH r� rS rB �
__classcell__r! r! rX r" r sn
��$�
��
���
� 5 &! "r c @ s\ e Zd Zdefdd�Zddd�Zdedefd d
�Zd d � Z d
d� Z
dede de fdd�Z dS )r5 �proxyc C s || _ i | _t� | _d S r- )�_proxyrm �objectrf �rG r� r! r! r" rN s zUniqueProxy.__init__r* Nc C s
i | _ d S r- )rm ry r! r! r" �clear! r� zUniqueProxy.clear�namec C �( t | j|�}t|�r| �||�S td��)Nz9Accessing non-functions through .unique is not supported.�r^ r� �callable�_wrapr> �rG r� �objr! r! r" rd $ � zUniqueProxy.__getattr__c C � | j �� }|S r- �rs rk ru r! r! r" � __getstate__+ �
zUniqueProxy.__getstate__c C rq r- rr ru r! r! r" rv 2 rw zUniqueProxy.__setstate__�functionc s t �� �� ��fdd��}|S )Nc sv �| t t|�� ��f}�j�|�jh�}�j}tt�D ]}||vr# n� | i |��}qtdtd�d���|� |� |S )Nz Got duplicated values after �,z iterations.)
r< rU rB rm �
setdefaultrf �range�_UNIQUE_ATTEMPTSr �add)�args�kwargsrM � generated�retval�i�r� r� rG r! r" �wrapper6 s
z"UniqueProxy._wrap.<locals>.wrapper)� functools�wraps�rG r� r� r� r! r� r" r� 5 s zUniqueProxy._wrap)r* N)
r� r� r� r rN r� r9 r rd r� rv r r� r! r! r! r" r5 s
r5 c @ sj e Zd ZdZdefdd�Zdedefdd�Zd d
� Z d d � Z
ded
e de f de de
e f fdd�ZdS )r7 zN
Return either a fake value or None, with a customizable probability.
r� c C s
|| _ d S r- )r� r� r! r! r" rN S r� zOptionalProxy.__init__r� r* c C r� )Nz;Accessing non-functions through .optional is not supported.r� r� r! r! r" rd V r� zOptionalProxy.__getattr__c C r� r- r� ru r! r! r" r� ] r� zOptionalProxy.__getstate__c C rq r- rr ru r! r! r" rv d rw zOptionalProxy.__setstate__r� .c s8 t �� �dd�dtdtdtdtt f� �fdd��}|S ) Ng �?)�probr� r� r� r* c sL d| k r
dkst d�� t d���jjt| d �d�r$� |i |��S d S )Nr g �?z prob must be between 0 and 1�d )�chance_of_getting_true)�
ValueErrorr� �booleanr/ )r� r� r� �r� rG r! r" r� h s
�(z$OptionalProxy._wrap.<locals>.wrapper)r� r� r r0 r r r� r! r� r" r� g s ,zOptionalProxy._wrapN)r� r� r� r� r rN r9 r rd r� rv r r r r� r! r! r! r" r7 N s .r7 )"rk r� r� � collectionsr r r �typingr r r r r r r
r r r
r) r �
exceptionsr rW r r& r r �utils.distributionr r� r r r5 r7 r! r! r! r" �<module> s& 0 3