File: //opt/saltstack/salt/lib/python3.10/site-packages/vultr/__pycache__/utils.cpython-310.pyc
o
;j� � @ sL d Z ddlZddlZddlZdZG dd� de�ZG dd� de�Z dd � Z
dS )
zHelper classes� Nzhttps://api.vultr.comc @ s e Zd ZdZdS )�
VultrErrorzVultr custom exceptionN)�__name__�
__module__� __qualname__�__doc__� r r �?/opt/saltstack/salt/lib/python3.10/site-packages/vultr/utils.pyr s r c @ sF e Zd ZdZdd� Zdd� Zddd�Zdd d
�Zd d � Zddd�Z dS )� VultrBasez Base class for Vultr inheritancec C s t | _|| _| �d� d S )N� )� API_ENDPOINT� api_endpoint�api_key�set_requests_per_second)�selfr
r r r �__init__ s zVultrBase.__init__c C s || _ d| j | _dS )z&Adjusts the request/second at run-timer
N)�req_per_second� req_duration)r r r r r r s z!VultrBase.set_requests_per_secondNc C s0 t |t�st� }| jr| j|d< tj||dd�S )zAPI GET request helperr
�< )�params�timeout)�
isinstance�dictr
�requests�get)r �urlr r r r �_request_get_helper s
z VultrBase._request_get_helperc C s" | j rd| j i}tj|||dd�S )zAPI POST helperr
r )r �datar )r
r Zpost)r r r �queryr r r �_request_post_helper# s
z VultrBase._request_post_helperc
C s\ z |dkr | � ||�W S |dkr| �||�W S td| �� tjy- } zt|��d}~ww )zAPI request helper method�POST�GETzUnsupported method %sN)r r r r ZRequestException� RuntimeError)r r r �method�exr r r �_request_helper) s ��zVultrBase._request_helperr c C s� t � � }|�d�s
d| }| �| j| ||�}|jdkrU|jdkr%td��|jdkr.td��|jdkr7td��|jd krCtd
|j ��|jd krLtd ��|jd
krUtd��t � � | }|| jk rht �| j| � |jro|� � S t
� d�S )zAPI request / call method�/�� i� z6Invalid API location. Check the URL that you are usingi� z\Invalid or missing API key. Check that your API key is present and matches your assigned keyi� z^Invalid HTTP method. Check that the method (POST|GET) matches what the documentation indicatesi� zPRequest failed. Check the response body for a more detailed description. Body:
i� z0Internal server error. Try again at a later timei� z\Rate limit hit. API requests are limited to an average of 1/s. Try your request again later.z{}) �time�
startswithr$ r Z status_coder �textr �sleep�json� json_module�loads)r �pathr r" Z_startZrespZ_elapsedr r r �request4 s.
�
zVultrBase.request)N)Nr )
r r r r r r r r r$ r/ r r r r r s
r c C s& t | t�r | �� nt� } | �|� | S )zMerges updates into params)r r �copy�update)r Zupdatesr r r �
update_params^ s
r2 ) r r r' r+ r, r r! r �objectr r2 r r r r �<module> s P