File: //home/arjun/projects/env/lib/python3.10/site-packages/faker/__pycache__/generator.cpython-310.pyc
o
'we� � @ s� d dl Zd dlZd dlmZmZmZmZmZm Z m
Z
m Z m Z ddlm
Z
er,ddlmZ e�d�Ze�� Z e Ze� ZG dd� d�ZdS ) � N) �
TYPE_CHECKING�Any�Callable�Dict�Hashable�List�Optional�Type�Union� )�SeedType)� BaseProviderz \{\{\s*(\w+)(:\s*\w+?)?\s*\}\}c @ s� e Zd ZU di iZeeeeef f ed< dZ e
Z deddfdd�Z d e
d
ed
f ddfd d �Zd
eded
fdd�Zded
fdd�Zedejfdd��Zejdejddfdd��Zd1dee dd fdd�Zed1dee ddfdd��Z ded ed edefd d �Z dede fd d!�Z d
ede ddfd"d#�Z d1d$ed%edee ddfd&d'�Z!d1d$ed%ee defd(d)�Z"d1d$ed%ee defd*d+�Z#d,edefd-d.�Z$d/d0� Z%dS )2� Generator� arguments�_Generator__configF�config�returnNc K s0 g | _ tt| j�� �t|�� � �| _t| _d S �N)� providers�dict�listr �items�random�_Generator__random)�selfr � r �H/home/arjun/projects/env/lib/python3.10/site-packages/faker/generator.py�__init__ s
zGenerator.__init__�providerr
c C s\ t |t�r || �}| j�d|� t|�D ]}|�d�r qt||�}t|�r+| �||� qd S )Nr �_) �
isinstance�typer �insert�dir�
startswith�getattr�callable�
set_formatter)r r � method_name�faker_functionr r r � add_provider s
��zGenerator.add_provider�namec s6 z� fdd�| � � D �}|d W S ty Y d S w )Nc s( g | ]}t |d �r|j� �� kr|�qS )� __provider__)�hasattrr, �lower)�.0�p�r+ r r �
<listcomp>3 s ( z&Generator.provider.<locals>.<listcomp>r )�
get_providers�
IndexError)r r+ �lstr r1 r r 1 s
�zGenerator.providerc C � | j S )zReturns added providers.)r �r r r r r3 8 � zGenerator.get_providersc C r6 r �r r7 r r r r < r8 zGenerator.random�valuec C s
|| _ d S r r9 )r r: r r r r @ s
�seedc C s* | j tkr
t�� | _ | j �|� d| _| S )zCalls random.seedT)r r �
random_module�Randomr; �
_is_seeded)r r; r r r �
seed_instanceD s
zGenerator.seed_instancec C s t �|� || _d| _d S )NT)r r; � _global_seedr> )�clsr; r r r r; N s
zGenerator.seed� formatter�args�kwargsc O s | � |�|i |��S )zL
This is a secure way to make a fake from another Provider.
)�
get_formatter)r rB rC rD r r r �formatT s zGenerator.formatc C sR zt | |�W S ty( d| jv r!d|�d| jd ��}t|��td|����w )N�localezUnknown formatter z
with locale )r% �AttributeErrorr )r rB �msgr r r rE Z s
��zGenerator.get_formatterc C s t | ||� dS )z�
This method adds a provider method to generator.
Override this method to add some decoration or logging stuff.
N)�setattr)r r+ rB r r r r' d s zGenerator.set_formatter�group�argumentc C s` || j d vri | j d |< t|t�r || j d |< dS t|t�s%td��|| j d | |< dS )a�
Creates an argument group, with an individual argument or a dictionary
of arguments. The argument groups is used to apply arguments to tokens,
when using the generator.parse() method. To further manage argument
groups, use get_arguments() and del_arguments() methods.
generator.set_arguments('small', 'max_value', 10)
generator.set_arguments('small', {'min_value': 5, 'max_value': 10})
r z/Arguments must be either a string or dictionaryN)r r r �str�
ValueError)r rK rL r: r r r �
set_argumentsk s
zGenerator.set_argumentsc C s> || j d v r|r| j d | �|�}|S | j d �|�}|S )a
Get the value of an argument configured within a argument group, or
the entire group as a dictionary. Used in conjunction with the
set_arguments() method.
generator.get_arguments('small', 'max_value')
generator.get_arguments('small')
r )r �get�r rK rL �resultr r r �
get_arguments s
�zGenerator.get_argumentsc C sF || j d v r |r| j d | �|�}|S | j d �|�}|S d}|S )z�
Delete an argument from an argument group or the entire argument group.
Used in conjunction with the set_arguments() method.
generator.del_arguments('small')
generator.del_arguments('small', 'max_value')
r N)r �poprQ r r r �
del_arguments� s ��zGenerator.del_arguments�textc C s t �| j|�S )a
Replaces tokens like '{{ tokenName }}' or '{{tokenName}}' in a string with
the result from the token method call. Arguments can be parsed by using an
argument group. For more information on the use of argument groups, please
refer to the set_arguments() method.
Example:
generator.set_arguments('red_rgb', {'hue': 'red', 'color_format': 'rgb'})
generator.set_arguments('small', 'max_value', 10)
generator.parse('{{ color:red_rgb }} - {{ pyint:small }}')
)� _re_token�sub�_Generator__format_token)r rV r r r �parse� s zGenerator.parsec C s� t |�� �\}}|r|�d��� nd}|r9z | jd | }W n ty, td|����w t| j|fi |���}nt| �|��}d� |�S )N�:� r zUnknown argument group )
r �groups�lstrip�stripr �KeyErrorrH rM rF �join)r �matchesrB �argument_groupr � formattedr r r �__format_token� s �
zGenerator.__format_tokenr )&�__name__�
__module__� __qualname__r r rM r r �__annotations__r> �Sentinelr@ r r
r r* r r r r3 �propertyr< r= r �setterr r? � classmethodr; rF r rE r' rO rS rU rZ rY r r r r r s0
�
r )r r< �re�typingr r r r r r r r r
r r r
�compilerW r= �
mod_random�objectrj r r r r r �<module> s ,