1#@!#!123s

��JFIF�� 1#@!#!123s
: / opt / saltstack / salt / extras-3.10 / bakauth / __pycache__ /
Filename : sess.cpython-310.pyc
back
o

�����<�f�����������������������@���s����d�Z�ddlmZmZ�ddlZddlZddlmZ�ddlm	Z	�ddl
Z
dZdZe�
d��ej��G�d	d
��d
ej�ZG�dd��dej�ZG�d
d��de�ZG�dd��de
j�Zddd�dedededededeedf�dedeeef�fdd�ZdS�)z8Handles automatic retries for Backup Authority API calls�����)�Union�AnyN)�Retry)�HTTPAdapter��������zurllib3.connectionpoolc�������������������@���sP���e�Zd�ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZ
dZd
ZdZdZdZdZdS�)�StatuszBackup Authority Error Codes���r�������r����������������������������	����
������������
�������N)�__name__�
__module__�__qualname__�__doc__�REQUEST_EXCEPTION�OKAY�ERRORZARGS_MISSINGZUNHANDLED_EXCEPTIONZSQLALCHEMY_ERRORZAMP_DOWNZWRONG_SERVER_CLASSZAUTH_FAILEDZINTERNAL_NO_QUOTAZPRODUCTION_ONLYZVDED_AMP_MISSINGZLOOKUP_MISSINGZ
AUTH_MIGRATEDZNO_REPLICANTZINTERNAL_VPS��r���r����//opt/saltstack/salt/extras-3.10/bakauth/sess.pyr������s$����r���c�������������������@���s���e�Zd�ZdZdZdZdZdS�)�MdsStatez&Possible states for vznode MDS backupsr���r
���r���N)r���r���r���r���r����WARNZCRITr���r���r���r���r ���%���s
����r ���c�����������������������s.���e�Zd�ZdZ						d��fdd�	Z���ZS�)�LoggedRetryzLogs on HTTP retriesNc��������������������s,���|d�urt��d||��t���||||||�S�)Nzbakauth::%s: %s)�logging�debug�super�	increment)�self�method�url�response�error�_pool�_stacktrace��	__class__r���r���r&���0���s
���	�zLoggedRetry.increment)NNNNNN)r���r���r���r���r&����
__classcell__r���r���r.���r���r"���-���s�����r"���c�����������������������s*���e�Zd�ZdZdedef��fdd�Z���ZS�)�BakAuthSessionz,Custom requests.Session for bakauth requests�retries�log_retriesc��������������������s>���t�������|r	tnt}|��dt|||||dgdd�d���d�S�)N�https://i���g�������?)�total�read�connect�status�status_forcelist�backoff_factor)�max_retries)r%����__init__r"���r����mountr���)r'���r2���r3���Zretry_classr.���r���r���r<���C���s���
���zBakAuthSession.__init__)r���r���r���r����int�boolr<���r0���r���r���r.���r���r1���@���s����r1���T)�authr3����bakauth_host�uri�data�timeoutr2���r@���r3����returnc��������������
���C���s����z)t�||��}|jd|���d|�d����|||d�}W�d����n1�s#w���Y��W�n�tjjyD�}	�z
tjt|	�fW��Y�d}	~	S�d}	~	ww�z|�	��}
W�n�t
y\���tjd|���f�Y�S�w�zt|
d��}W�n�t
yq���tj}Y�nw�||
d�fS�)	a���Performs a Backup Authority API request

    Args:
        bakauth_host (str): bakauth server fqdn
        uri (str): endpoint uri to use
        data (dict): POST form data
        timeout (int): seconds before raising BakAuthDown
        retries (int): number of automatic retries
        auth (tuple | None): HTTP auth info

    Returns:
        (Status, data): Status response (enum) and returned data
    r4���z:4002/�/)r)���r@���rD���rC���NzInvalid JSON from auth server: r8���rC���)r1����post�lstrip�requests�
exceptions�RequestExceptionr���r����str�json�
ValueErrorr���)rA���rB���rC���rD���r2���r@���r3����session�raw�exc�retr8���r���r���r���rG���U���s2���������
�rG���)r����typingr���r���r#����enumZurllib3.util.retryr���Zrequests.adaptersr���rI����DEFAULT_TIMEOUT�DEFAULT_RETRIES�	getLogger�setLevelr����Enumr���r ���r"����Sessionr1���rL����dictr>����tupler?���rG���r���r���r���r����<module>���sB����������
��
	�