File: //home/arjun/projects/buyercall/buyercall/blueprints/widgets/__pycache__/models.cpython-310.pyc
o
we�C � @ s d dl Z d dlZd dlmZ d dlmZ d dlmZ d dl m
Z
m Z m Z d dl
mZ d dlmZ d dlmZ d d lmZ d d
lmZmZmZ d d lmZ d d lm Z d d
l m Z d dl m Z G dd� deej!�Z"G dd� dej!�Z#G dd� deej!�Z$dd� Z%dS )� N)�jsonify)� current_user)�
ResourceMixin)�backref�column_property� load_only)�
postgresql)�association_proxy)�
flag_modified)�Index)�and_�func�select)�db)�User)�Lead)�Agentc @ s� e Zd ZdZedddd�fZejejdd�Z ejej
dddd�Z ej e
d d
�Zed d �Zejejejd
ddd�ddd�Zeje�
d�ddd�Zeje�
d�dd�Zeje�
d�ddd�Ze�dejejdddd��Zejdejddd�Zejejdddd �Zeje�
d �dddd �Z ejejdddd �Z e e e �!e
j �g��"e
j#e k��Z$ejejejd ddd�ddd�Z%ejejejd ddd�ddd�Z&d d!� Z'd"d#� Z(d$d%� Z)d&d'� Z*e+dLd)d*��Z,e+dLd+d,��Z-e+d-d.� �Z.e+d/d0� �Z/e0d1d2� �Z1e0d3d4� �Z2e2j3d5d4� �Z2e0d6d7� �Z4e0d8d9� �Z5e0d:d;� �Z6e0d<d=� �Z7e+d>d?� �Z8e+d@dA� �Z9e+dBdC� �Z:e+dDdE� �Z;e0dFdG� �Z<e0dHdI� �Z=dJdK� Z>d(S )M�Widget�widgets�uq_widgets_guid�guidT)�unique�� primary_keyF)�index�nullabler �widget�r � assignments�agent�partnership_accounts.id�CASCADE��onupdate�ondelete�r r � i, �r �2 �phone_number_idzphonenumbers.idzSET NULL�optionsz{}�r �server_default�true)r �defaultr, � )r r r zusers.idc C s* g }| j D ] }|�|jj� qtt|��S �N)�channels�append� channel_type�name�list�set)�self�
channel_types�channel� r: �E/home/arjun/projects/buyercall/buyercall/blueprints/widgets/models.py�get_channel_typesV s
zWidget.get_channel_typesc C s$ | j D ] }|jj|kr| S qd S r0 )r1 r3 r4 )r7 �c_typer9 r: r: r; �get_channel_by_type\ s
�zWidget.get_channel_by_typec C s ddl m} |�| j�}|jS )Nr )�PartnershipAccount)�'buyercall.blueprints.partnership.modelsr? � get_by_id�partnership_account_id�partnership_id)r7 r? �partnership_accountr: r: r; �get_partnership_idb s zWidget.get_partnership_idc C s( ddl m}m} |�| j�}|�|j�S )Nr )r? � Partnership)r@ r? rF rA rB �get_sid_from_idrC )r7 r? rF rD r: r: r; �get_partnership_sidg s zWidget.get_partnership_sidNc C sP |r&g }| j �| jdktjtjk��� }|D ]
}||�� vr#|�|� q|S g S )NT) �query�filter�
is_options_v3r rB r �allr< r2 )�cls�type_r �_widgets�_widgetr: r: r; �get_unallocated_typel s ��
�zWidget.get_unallocated_typec
C s� ddl m} d}d}z!tdi |��}|j�|j|k��� }|r#|j|_|� � W ||fS t
yL } zt � d�
|�� t|�}W Y d}~||fS d}~ww )zb
Return widget id if widget was created successfully.
:return: widget id
r ��PhoneN� Error creating widget. Error: {}r: )�(buyercall.blueprints.phonenumbers.modelsrS r rI rJ �sid�first�id�
inbound_id�save� Exception�log�error�format�str)rM �params�phonenumber_idrS r r] �phone�er: r: r; �
api_createx s
���zWidget.api_createc
C s� ddl m} d}z2tdi |��}|j�|j|k|j|jk��� }|r%||_t j
� |� t j
� � |j
}t j
�� W |S tyU } zt�d�|�� W Y d}~|S d}~ww )zf
Return widget guid if widget was created successfully.
:return: widget guid
r rR ���rT Nr: )rU rS r rI rJ rX rB rW �inboundr �session�add�flushr �commitr[ r\ r] r^ )rM r` ra rS � widget_guidr rb rc r: r: r; �api_create_new� s,
��
���zWidget.api_create_newc
C s� ddl m} d}z�tj�ttj|ktj|k���� }|r�|r!||_ |r&||_
|dv r-||_ |j r�|j }t
�|� |r�|d }|rV|d }|
rJ|
|d< |rP||d< |rV||d< |r\||d < | rb| |d
< |rh||d < |
rn|
|d < | rt| |d
< | rz| |d< |r�||d< |r�||d< |r�||d< |r�||d< t|d� |j�|j|k|j|jk��� }|r�||_tj�� d}W |S W |S ty� } zt
�d�|�� W Y d}~|S d}~ww )z[
Return True if widget was updated successfully.
:return: Boolean
r rR F)TF�agentsrX � contactUsing�fullName�addHiddenInformation�hiddenInformation�language� recordCalls� voicemail�voicemailMessage� retryRouting�
routeRandomly�routeInSequence�routeSimultaneously�_optionsTz Error updating widget. Error: {}N)rU rS r rI rJ r rB r rW r4 �type�enabledr* r\ r] r
rX rY r rg rj r[ r^ ) rM rB r r4 r{ ra r| �add_hidden_informationrr �hidden_information� record_callsrt �voicemail_message�agent_id�agent_contact_using�agent_full_name�route_randomly�route_in_sequence�route_simultaneously�
retry_routingrS �resultr �result_options�result_option_agents� first_agentrb rc r: r: r; �
api_update� s�
�
��
�J���zWidget.api_updatec C s t dd� | jD ��S )zC Checks if the agent schedule allows calling at this time.
c s s � | ]}|j V qd S r0 )�
available_now)�.0r r: r: r; � <genexpr> s � z*Widget.agents_available.<locals>.<genexpr>)�anyrm �r7 r: r: r; �agents_available s zWidget.agents_availablec C sn dd� t | jdd� d�D �}|| jd< | j| jd< z
| jd | jd < W | jS ty6 d
| jd < Y | jS w ) Nc S s" g | ]
}t |j|jj|jd ��qS ))rX ro rn )�dictr� r � full_name�
contact_using)r� �asgr: r: r; �
<listcomp> s ����"Widget.options.<locals>.<listcomp>c S � | j S r0 )�order)�xr: r: r; �<lambda> s z Widget.options.<locals>.<lambda>)�keyrm �
widgetType�
notifyNone�enableNotificationsF)�sortedr rz r{ �KeyError)r7 rm r: r: r; r* s �
� �zWidget.optionsc C s� |sd S dd� | j D �D ]}tj�|� q t|�dg ��D ]\}}|d |d }}| j �t| |||d�� q |�dd�| _|d |d
< || _ d S ) Nc S s g | ]}|�qS r: r: �r� r� r: r: r; r� - s r� rm rX rn )r r� r� r� r� r r� r� )
r r rg �delete� enumerate�getr2 �AgentAssignmentr{ rz )r7 �optr� �ir �id_r� r: r: r; r* ' s
�
c C s� t � }| jd r
|S tj�tjdktj�d�tj| jk�� � }|r$|j
nd}dd� | j D �}| jd rS| jd r?|r?|� |� | jd rI|�
|� |�
t| jd
�� | jd rx| jd rd|rd|� |� | jd
rn|�
|� |�
t| jd �� |S )z6 Return a list of notification emails for the widget. r� �adminFNc S s g | ]}|j �qS r: )�email)r� �ar: r: r; r� L s z!Widget.emails.<locals>.<listcomp>�notifyAllLeads� notifyAllMe�notifyAllAgents�notifyAllCustom�notifyMissedLeads�notifyMissedMe�notifyMissedAgents�notifyMissedCustom)r5 r* r rI rJ �role�is_deactivated�is_rB rW r� rm r2 �extend� split_emails)r7 r� r� �my_email�
agents_emailsr: r: r; �emails? s2
��
z
Widget.emailsc C s< d}d| j v r | j d r | j d du s| j d du r d}|S )�? Returns whether or not the lead is an ADF widget lead
F� notifyAdfTr- )r* �r7 r� r: r: r; �is_adf] s
&z
Widget.is_adfc C r� )r� )rY r� r: r: r; r) i s zWidget.phone_number_idc C s( t � }| jd r|�t| jd �� |S )zB Return a list of adf notification emails for the widget.
r� �notifyAdfCustom)r5 r* r� r� r� r: r: r; �adf_notification_emailso s
z Widget.adf_notification_emailsc C s� d}|rt j�td���t j|kt j|k��� }n|r/t j�td���t j|kt j|k��� }|r>t j
� |� t j
� � dS dS )z< Deletes widget. Provide None where not applicable.
NrX r TF)
r rI r* r rJ rX rB rW r r rg r� rj �rM rX r rB r r: r: r; r� z s0 �
���
��
z
Widget.deletec C s~ d}|rt j�tdd���t j|kt j|k��� }n|r1t j�tdd���t j|kt j|k��� }|r=d|_ t
j � � dS dS )z< Disable widget. Provide None where not applicable.
NrX r| r FT�
r rI r* r rJ rX rB rW r r| r rg rj r� r: r: r; �disable� �$ ����
zWidget.disablec C s~ d}|rt j�tdd���t j|kt j|k��� }n|r1t j�tdd���t j|kt j|k��� }|r=d|_ t
j � � dS dS )z; Enable widget. Provide None where not applicable.
NrX r| r TFr� r� r: r: r; �enable� r� z
Widget.enablec C s. t j�t jj|d�t j|k��� }|rdS dS )N)r TF)r rI rJ r1 r� r{ rW )rM �type_id� widget_idr r: r: r; � is_widget_channel_type_exist� s " z#Widget.is_widget_channel_type_existc C � | j �d�S )z6 Return the date/time this widget was created
�%Y-%m-%d %H:%M:%S)�
created_on�strftimer� r: r: r; �created_datetime� � zWidget.created_datetimec C r� )z6 Return the date/time this widget was updated
r� )�
updated_onr� r� r: r: r; �updated_datetime� r� zWidget.updated_datetimec C s t | j| j| j| j| j| jd�S )N)rX r4 �user_idr� r� r* )r rX r4 r� r� r� r* r� r: r: r; r � s �zWidget.jsonifyr0 )?�__name__�
__module__� __qualname__�
__tablename__r �__table_args__r �Column�IntegerrX �Stringr � relationshipr �leadsr rm �
ForeignKeyrB r4 � descriptionr{ rY r �JSONrz �BooleanrK r� r| r r r
�count�whererk �
lead_count�
created_by�
updated_byr< r> rE rH � classmethodrQ rd rl r� �propertyr� r* �setterr� r� r) r� r� r� r� r� r� r� r r: r: r: r; r s� ��
���
�
�
�"����
]
r c @ s� e Zd ZdZejeeddd�d�Zeje eddd�d�Z
ej ej ej
dddd�d d
�Zej ej ej
d ddd�d d
�Z ej e�d �d
dd�Ze� e� � �ZdS )r� �
widget_agentsr rL )�cascader z
widgets.idr! r" Tr z agents.id�
Frb r+ N)r� r� r� r� r r� r r r r r r� r� r� r� r� r� r� r� r: r: r: r; r� � s2 � ����r� c @ sT e Zd ZdZeje�d�dd�Zejejej dddd�ddd �Z
ejejdd
�Z d S ) �
WidgetFile� widget_files�$ Tr r r! r" Fr% r' N) r� r� r� r� r r� r� r r� r� rB �urlr: r: r: r; r� � s �r� c C s dd� t �d| �D �S )z� Split a string of email addresses into individual emails.
:param email_str: A comma- or semicolon-separated list of emails.
:type email_str: str
:return: A list of emails
:rtype: list
c S s g | ]}|r|� � �qS r: )�stripr� r: r: r; r� s z split_emails.<locals>.<listcomp>z,|;)�re�split)� email_strr: r: r; r� s r� )&r� �loggingr\ �flaskr � flask_loginr � buyercall.lib.util_sqlalchemyr �sqlalchemy.ormr r r �sqlalchemy.dialectsr � sqlalchemy.ext.associationproxyr �sqlalchemy.orm.attributesr
�sqlalchemy.schemar �
sqlalchemyr r
r �buyercall.extensionsr � buyercall.blueprints.user.modelsr �!buyercall.blueprints.leads.modelsr �"buyercall.blueprints.agents.modelsr �Modelr r� r� r� r: r: r: r; �<module> s, C