File: //home/arjun/projects/buyercall_new/buyercall/buyercall/lib/__pycache__/util_twilio.cpython-38.pyc
U
y�Tf16 � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dl m
Z
d dl m Z e�
e�ZdZdZdd � Zd
d � Zd d
� Zdd� Zdd� Zdd� Zdd� ZG dd� de�ZG dd� de�ZG 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)�dbi i�Q c C sl ddl m} | }tjd }|�|�}t|�}|�|j�}|�|j�}tj� dd�r^t
||�}n
t
||�}|S )Nr ��PartnershipCpaasProviders�CRYPTO_SECRET_KEYZ USE_TEST_APIF) �'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 �I/home/arjun/projects/buyercall_new/buyercall/buyercall/lib/util_twilio.py�account_client s"
��r c
C sV ddl m} |}tjd }|�|�}t|�}|�|j�}|�|j�}t ||| d�} | S )Nr r r )Z account_sid)
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 sl ddl m} ddlm} tjd }| }|�|�}t|�}|�|j �}|�|j
�}|�|j �} |||| �}
|
S )Nr �� Bandwidthr r ) Zbuyercall.lib.bandwidthr r r r
r �!partnership_bandwidth_credentialsr r
r Zcpaas_api_secretZ
cpaas_user_id) r r r r � partnershipr r r Zdecrypted_secretr �clientr r r � bw_clientB s
�r# c
C s| | }ddl m} ddlm} tjd }|�|�}t|�}|�|j �}|�|j
�}|�|j �} |j }
|j
} ||| ||
| �} | S )Nr r r r )r r Z buyercall.lib.bandwidth_api_v2r r
r r r r
r Zcpaas_api_usernameZcpaas_api_passwordZ
cpaas_site_idZcpaas_location_id)
r r! r r r r r Zdecrypted_account_idZdecrypted_api_usernameZdecrypted_api_passwordZsite_idZ location_idr" r r r �bw_dashboard_clientX s&
�r$ c C s( d}| dkrd\}}nd\}}|||fS )NZfemale�es)�es_mxZ esperanza)�en_USZsusanr )�languageZgender�locale�voicer r r � select_voiceq s
r+ c C s� ddl m} d�||�}| dkrFt�� }|�||� |�|dt � nR| dkr�t�� }t|�|�pdd�}|r�|j �
|j |k�� |j
|| i� tj�� 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, Ztime_keyZ
timestamp_nowZtimestamp_thenr r r �update_response_time{ s �rA c C s2 | pd}|d }}d|kr*|� dd�\}}||fS )z6 Split a name into (first name, last name) tuple.
�Unknown� � � ��split)�name� full_nameZ
first_name� last_namer r r �
split_name� s
rK 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 r0 )�.0r> �r@ r r �
<listcomp>� s �z*BusyAgents.filter_from.<locals>.<listcomp>)�redis� StrictRedisr
r )�
all_agentsr rT r � filter_from� s
�zBusyAgents.filter_fromc C s2 t jtjd tjd d�}|�d�| �dd� dS )z Mark an agent as busy.rM rN rO rR iX rE N)rV rW r
r �setexr0 �r> r@ r r r �add� s zBusyAgents.addc C s. t jtjd tjd d�}|�d�| �� dS )z Mark an agent as available.rM rN rO rR N)rV rW r
r �deleter0 r[ r r r �remove� s zBusyAgents.removeN)�__name__�
__module__� __qualname__�__doc__� staticmethodrY r\ r^ r r r r rL � s
rL c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� RedisAttrc C s
|| _ d S �N)�_RedisAttr__name��selfrH r r r �__init__� s zRedisAttr.__init__c C sH t �d�| j|�� |d kr0|j�|j| j� n|j�|j| j|� d S )Nz{} -> {})�log�debugr0 rf r@ �hdel� _lead_key�hset)rh �instance�valuer r r �__set__� s zRedisAttr.__set__c C s |j �|j| j�S re )r@ �hgetrm rf )rh ro �ownerr r r �__get__� s zRedisAttr.__get__N)r_ r` ra ri rq rt r r r r rd � s rd c s( e Zd Z� fdd�Z� fdd�Z� ZS )�BoolAttrc s& t |�r dnd}tt| ��||� d S )N�xrC )�bool�superru rq )rh ro rp � redis_value�� __class__r r rq � s zBoolAttr.__set__c s t t| ��||�}t|�S re )rx ru rt rw )rh ro rs ry rz r r rt � s zBoolAttr.__get__)r_ r` ra rq rt �
__classcell__r r rz r ru � s ru c s4 e Zd Z� fdd�Z� fdd�Z� fdd�Z� ZS )�JsonAttrc s t t| ��|� d S re )rx r} ri rg rz r r ri � s zJsonAttr.__init__c s t �|�}tt| ��||� d S re )�json�dumpsrx r} rq )rh ro �dict_rp rz r r rq � s
zJsonAttr.__set__c s6 t t| ��||�}|sd S t|t�r,|�� }t�|�S re )rx r} rt �
isinstance�bytes�decoder~ �loads)rh ro rs rp rz r r rt � s
zJsonAttr.__get__)r_ r` ra ri rq rt r| r r rz 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_taskr> � 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_{}) r@ r? r0 rm �_callback_cnt_key�
_retry_key� _lock_key� _agents_key�_agent_calls_key)rh r@ r? r r r ri > s zCallStorage.__init__c C s, | j �| jd| j� | j �| jdt � d S )Nr? � )r@ rn rm r? r3 r4 �rh r r r �initI s zCallStorage.initc C s dd� | j �| jdd�D �S )Nc S s g | ] }t |��qS r ��int�rS rv r r r rU O s z.CallStorage.agents_to_call.<locals>.<listcomp>r ���)r@ �lranger� r� r r r �agents_to_callM s zCallStorage.agents_to_callc s$ dd� � � fdd�| j �| j�D �S )Nc S s | � d�}t|d �|d fS )N�_r rE )rG r� )�itemZtplr r r �to_agent_call_tupleS s
z<CallStorage.current_agent_calls.<locals>.to_agent_call_tuplec s g | ] }� |��qS r r r� �r� r r rU V s �z3CallStorage.current_agent_calls.<locals>.<listcomp>)r@ �smembersr� r� r r� r �current_agent_callsQ s
�z CallStorage.current_agent_callsc C s |r | j j| jft|��� d S re )r@ �lpushr� �reversed)rh � agent_idsr r r �set_agents_to_call[ s z CallStorage.set_agents_to_callc C s | j �| j�}|rt|�S d S re )r@ �lpopr� r� )rh rp r r r �next_agent_to_call_ s z CallStorage.next_agent_to_callc C s4 d� ||�}| j�| j|� | j�| jdt � d S )N�{}_{}r� )r0 r@ �saddr� r3 r4 �rh r> r� rp r r r �push_agent_callc s zCallStorage.push_agent_callc
C sJ d� ||�}| �� �, | j�| j|� | j�| j�W 5 Q R � S Q R X d S )Nr� )r0 �lockr@ �sremr� �scardr� r r r �remove_agent_callh s
z CallStorage.remove_agent_callNc sr | � � �` | j�| j�}dd� |D �}� fdd�dd� |D �D �}|rd| jj| jfdd� |D ��� W 5 Q R X |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 re )r� r� r� )rS rp r r r � <genexpr>t s z0CallStorage.clear_agent_calls.<locals>.<genexpr>c s: g | ]2}� r"� t |d �|d �rt |d �|d f�qS )r rE r� )rS �lst�� predicater r rU u s �z1CallStorage.clear_agent_calls.<locals>.<listcomp>c s s | ]}|� d �V qdS )r� NrF r� r r r r� w s c S s g | ]\}}d � ||��qS )r� )r0 )rS r> r� r r r rU { s �)r� r@ r� r� r� )rh r� �valuesZ
agent_sidsr r� r �clear_agent_callsn s
��z CallStorage.clear_agent_callsc C s | j �| jdt d� d S )Nr� r )r@ rZ r� r4 r� r r r �clear_callback_cnt� s z CallStorage.clear_callback_cntc C s t | j�| j��S re )r� r@ r r� r� r r r � callback_cnt� s zCallStorage.callback_cntc C s | j �| j�S re )r@ �incrr� r� r r r �inc_callback_cnt� s z CallStorage.inc_callback_cntc C s | j �| jdt d� d S )Nr. r )r@ rZ r� �HOURSr� r r r �clear_retry_cnt� s zCallStorage.clear_retry_cntc C s | j �| j�S re )r@ r� r� r� r r r �
inc_retry_cnt� s zCallStorage.inc_retry_cntc C s | j j| jdd�S )N�
)�timeout)r@ r� r� r� r r r r� � s zCallStorage.lock)N)5r_ r` ra rb r} r� rd r� r� r� r� r� r� r� r� r r� r> r� r� r� r� r� r� r� r� r� r� r� r� r� r� ru r� r� r� r� ri r� �propertyr� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� � sb
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)r_ r` ra r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� � s r� ) r~ �loggingr1 rV �flaskr r
Z twilio.restr Zbuyercall.lib.util_cryptor �buyercall.extensionsr � getLoggerr_ rj r� r4 r r r# r$ r+ rA rK �objectrL rd ru r} r� r� r r r r �<module> s0
$
1