File: //home/arjun/projects/buyercall/buyercall/lib/__pycache__/util_twilio.cpython-310.pyc
o
�we�7 � @ s d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dl m
Z
m Z d dl m
Z
e�e�ZdZdZdd � Zd
d � Zd d
� Zdd� Zdd� Zdd� Zdd� ZG dd� de�ZG dd� de�Z G dd� de �Z G d d � d e �Z G d d � d e�Z G d d!� d!e�Z dS )"� N)� current_app)�Client)� AESCipher�
AESCipherDome)�dbi i�Q c C sv ddl m} | }tjd }|�|�}t|�}|r9|�|j�}|�|j�}tj� dd�r2t
||�}|S t
||�}|S dS )Nr ��PartnershipCpaasProviders�CRYPTO_SECRET_KEY� USE_TEST_APIFT) �'buyercall.blueprints.partnership.modelsr �app�config� partnership_twilio_credentialsr �decrypt�cpaas_account_id�cpaas_api_token�getr ) �partnership_idr �partner� encrypt_key�
partner_cpaas�cipher�decrypted_user_id�decrypted_token�
twilio_client� r �;/home/arjun/projects/buyercall/buyercall/lib/util_twilio.py�account_client s(
�
��r c
C s^ ddl m} |}tjd }|�|�}t|�}|r-|�|j�}|�|j�}t ||| d�} | S dS )Nr r r )� account_sidT)
r r r r
r r r r r r )
�subaccount_sidr r r r r r r r r r r r �subaccount_client/ s
�r c C st ddl m} ddlm} tjd }| }|�|�}t|�}|r8|�|j �}|�|j
�}|�|j �} |||| �}
|
S dS )Nr �� Bandwidthr r T) �buyercall.lib.bandwidthr" r r r r
�!partnership_bandwidth_credentialsr r r �cpaas_api_secret�
cpaas_user_id) r r" r r � partnershipr r r �decrypted_secretr �clientr r r � bw_clientE s"
�r* c C s| | }ddl m} ddlm} tjd }|�|�}t|�}|r<|�|j �}|�|j
�}|�|j �} |j }
||| ||
�} | S dS )Nr r r! r T)
r r � buyercall.lib.bandwidth_api_v2r" r r
r$ r r r �cpaas_api_username�cpaas_api_password�
cpaas_site_id) r r' r r" r r r �decrypted_account_id�decrypted_api_username�decrypted_api_password�site_idr) r r r �bw_dashboard_client^ s&
�r3 c C s( d}| dkr d\}}nd\}}|||fS )N�female�es)�es_mx� esperanza)�en_US�susanr )�language�gender�locale�voicer r r � select_voicex s
r> c C s� ddl m} d�||�}| dkr$t�� }|�||� |�|dt � d S | dkrOt�� }t|�|�p3d�}|rQ|j �
|j |k�� |j
|| i� tj�� d S d S d S )Nr )�Leadz
TIME_{}_{}�ringing� z in-progress)�!buyercall.blueprints.leads.modelsr? �format�time�setnx�expire�DAYS�floatr �query�filter�id�update�response_time_secondsr �session�commit)� call_status�agent_id�lead_id�redis_dbr? �time_key�
timestamp_now�timestamp_thenr r r �update_response_time� s
��rW c C s2 | pd}|d}}d|v r|� dd�\}}||fS )z6 Split a name into (first name, last name) tuple.
�Unknown� � � ��split)�name� full_name�
first_name� last_namer r r �
split_name� s
rb c @ s4 e Zd ZdZedd� �Zedd� �Zedd� �ZdS ) �
BusyAgentszE Track the agents that are currently busy in a call with a user.
c s, t jtjd tjd d�� � fdd�| D �S )zv Return only those agents from `all_agents` who are not busy.
:param all_agents: a list of agent ids
�REDIS_CONFIG_URL�REDIS_CONFIG_PORT��host�portc s g | ] }� � d �|��s|�qS )�BUSY_{})r rC )�.0rQ �rS r r �
<listcomp>� s ��z*BusyAgents.filter_from.<locals>.<listcomp>)�redis� StrictRedisr r
)�
all_agentsr rk r � filter_from� s
�zBusyAgents.filter_fromc C s2 t jtjd tjd d�}|�d�| �dd� dS )z Mark an agent as busy.rd re rf ri iX r[ N)rm rn r r
�setexrC �rQ rS r r r �add� s zBusyAgents.addc C s. t jtjd tjd d�}|�d�| �� dS )z Mark an agent as available.rd re rf ri N)rm rn r r
�deleterC rr r r r �remove� s zBusyAgents.removeN)�__name__�
__module__� __qualname__�__doc__� staticmethodrp rs ru r r r r rc � s
rc c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� RedisAttrc C s
|| _ d S �N)�_RedisAttr__name��selfr^ r r r �__init__� s
zRedisAttr.__init__c C sJ t �d�| j|�� |d u r|j�|j| j� d S |j�|j| j|� d S )Nz{} -> {})�log�debugrC r} rS �hdel� _lead_key�hset)r �instance�valuer r r �__set__� s zRedisAttr.__set__c C s |j �|j| j�S r| )rS �hgetr� r} )r r� �ownerr r r �__get__� � zRedisAttr.__get__N)rv rw rx r� r� r� r r r r r{ � s r{ c s( e Zd Z� fdd�Z� fdd�Z� ZS )�BoolAttrc s& t |�rdnd}tt| ��||� d S )N�xrY )�bool�superr� r� )r r� r� � redis_value�� __class__r r r� � s zBoolAttr.__set__c s t t| ��||�}t|�S r| )r� r� r� r� )r r� r� r� r� r r r� � s zBoolAttr.__get__)rv rw rx r� r� �
__classcell__r r r� r r� � s r� c s4 e Zd Z� fdd�Z� fdd�Z� fdd�Z� ZS )�JsonAttrc s t t| ��|� d S r| )r� r� r� r~ r� r r r� � s zJsonAttr.__init__c s t �|�}tt| ��||� d S r| )�json�dumpsr� r� r� )r r� �dict_r� r� r r r� � s
zJsonAttr.__set__c s6 t t| ��||�}|s
d S t|t�r|�� }t�|�S r| )r� r� r� �
isinstance�bytes�decoder� �loads)r r� r� r� r� r r r� � s
zJsonAttr.__get__)rv rw rx r� r� r� r� r r r� r r� � s r� c @ s� e Zd ZdZed�Zed�Zed�Zed�Z ed�Z
ed�Z ed�Z ed �Z
ed
�Zed �Zed �Zed
�Zed�Zed�Zed�Zed�Zed�Zed�Zed�Zed�Zed�Zed�Zed�Z ed�Z ed�Z ed�Z e d �Z!e d �Z"ed �Z#ed �Z$d d!� Z%d"d#� Z&e'd$d%� �Z(e'd&d'� �Z)d(d)� Z*d*d+� Z+d,d-� Z,d.d/� Z-d?d1d2�Z.d3d4� Z/e'd5d6� �Z0d7d8� Z1d9d:� Z2d;d<� Z3d=d>� Z4d0S )@� CallStoragezK A wrapper over a Redis database for storing call-related information.
�state�
call_order�agent_outbound_call�
agent_call_id�call_sid�bridge_call_sid_a�bridge_call_sid_b� caller_name�
caller_numberr � prompt_taskrQ � manual_call� agent_disconnect_webhook_sent�routing_config�routing�
bw_conf_id�bw_conf_member_id�inbound�
start_time� connect_time�first_call_agent_id�
call_cause�cause_description�
agent_list�
pressed_digit�is_group�
is_call_retry�sip_endpoint_id�sip_call_directionc C sd || _ || _d�| j�| _d�| j�| _d�| j�| _d�| j�| _d�| j�| _d�| j�| _d S )NzLEAD:{}zCALLBACK_CNT_{}zRETRY_{}zLOCK_{}z AGENTS_{}zCALLS_{}) rS rR rC r� �_callback_cnt_key�
_retry_key� _lock_key� _agents_key�_agent_calls_key)r rS rR r r r r� E s zCallStorage.__init__c C s, | j �| jd| j� | j �| jdt � d S )NrR � )rS r� r� rR rF rG �r r r r �initP s zCallStorage.initc C s dd� | j �| jdd�D �S )Nc S s g | ]}t |��qS r ��int�rj r� r r r rl V s z.CallStorage.agents_to_call.<locals>.<listcomp>r ���)rS �lranger� r� r r r �agents_to_callT s zCallStorage.agents_to_callc s$ dd� � � fdd�| j �| j�D �S )Nc S s | � d�}t|d �|d fS )N�_r r[ )r] r� )�item�tplr r r �to_agent_call_tupleZ s
z<CallStorage.current_agent_calls.<locals>.to_agent_call_tuplec s g | ]}� |��qS r r r� �r� r r rl ] s ��z3CallStorage.current_agent_calls.<locals>.<listcomp>)rS �smembersr� r� r r� r �current_agent_callsX s
�z CallStorage.current_agent_callsc C s& |r| j j| jgt|��R � d S d S r| )rS �lpushr� �reversed)r � agent_idsr r r �set_agents_to_callb s �z CallStorage.set_agents_to_callc C s | j �| j�}|r
t|�S d S r| )rS �lpopr� r� )r r� r r r �next_agent_to_callf s z CallStorage.next_agent_to_callc C s4 d� ||�}| j�| j|� | j�| jdt � d S )N�{}_{}r� )rC rS �saddr� rF rG �r rQ r� r� r r r �push_agent_callj s zCallStorage.push_agent_callc C sV d� ||�}| �� � | j�| j|� | j�| j�W d � S 1 s$w Y d S )Nr� )rC �lockrS �sremr� �scardr� r r r �remove_agent_callo s
$�z CallStorage.remove_agent_callNc s� | � � �? | j�| j�}dd� |D �}� fdd�dd� |D �D �}|r;| jj| jgdd� |D ��R � W d � |S W d � |S 1 sFw Y |S )z� Delete from Redis, and return a list of (agent_id, call_sid) pairs,
representing the in-progress agent call attempts for the current lead.
c s s& � | ]}t |t�r
|�� n|V qd S r| )r� r� r� )rj r� r r r � <genexpr>{ s �$ z0CallStorage.clear_agent_calls.<locals>.<genexpr>c s: g | ]}� r� t |d �|d �rt |d �|d f�qS )r r[ r� )rj �lst�� predicater r rl | s ���z1CallStorage.clear_agent_calls.<locals>.<listcomp>c s s � | ]}|� d �V qdS )r� Nr\ r� r r r r� ~ s � c S s g | ]
\}}d � ||��qS )r� )rC )rj rQ r� r r r rl � s
��N)r� rS r� r� r� )r r� �values�
agent_sidsr r� r �clear_agent_callsu s"
��
��
��z CallStorage.clear_agent_callsc C � | j �| jdt d� d S )Nr� r )rS rq r� rG r� r r r �clear_callback_cnt� � z CallStorage.clear_callback_cntc C s t | j�| j��S r| )r� rS r r� r� r r r � callback_cnt� s zCallStorage.callback_cntc C � | j �| j�S r| )rS �incrr� r� r r r �inc_callback_cnt� � z CallStorage.inc_callback_cntc C r� )NrA r )rS rq r� �HOURSr� r r r �clear_retry_cnt� r� zCallStorage.clear_retry_cntc C r� r| )rS r� r� r� r r r �
inc_retry_cnt� r� zCallStorage.inc_retry_cntc C s | j j| jdd�S )N�
)�timeout)rS r� r� r� r r r r� � r� zCallStorage.lockr| )5rv rw rx ry r� r� r{ r� r� r� r� r� r� r� r� r r� rQ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� �propertyr� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� � sd
r� c @ sD e Zd ZdZdZdZdZdZdZdZ dZ
d Z d
Z d Z
d Zd
ZdZdS )�InboundCallState�NEW�CALLBACK� LEAD_ONHOLD� AGENT_ONHOLD�CALLBACK_PROMPT�ANSWERED�ONGOING�MISSED�BLOCKED�ERROR�MACHINE�CAPTURED� CALL_ME_BACK� DIGIT_MISSEDN)rv rw rx r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r� � s r� )!r� �loggingrD rm �flaskr r � twilio.restr �buyercall.lib.util_cryptor r �buyercall.extensionsr � getLoggerrv r� r� rG r r r* r3 r> rW rb �objectrc r{ r� r� r� r� r r r r �<module> s2
$
1