File: //home/arjun/projects/buyercall/buyercall/blueprints/agents/__pycache__/models.cpython-310.pyc
o
we�H � @ s6 d dl Z d dlZd dlmZ d dlmZ d dlZd dlmZmZ d dl m
Z
d dl m Z d dl
mZ d dlmZ d d lmZ e �e�Zd
ZG d d � d eej�Ze�d
ejdeje� d�dd�ejdeje� d�dd��Z G dd� deej�Z e�de�deje� d��e�deje� d���Z G dd� deej�Z dS )� N)� OrderedDict)�datetime)�or_�and_)�UUID)�hybrid_property)�text)�db)�
ResourceMixinz
US/Easternc @ s� e Zd ZdZejejdd�Zejedd�de j
dd�Z ejeje� d�dd�Z
ejejdd�Zejejddd �Zeje�d
�dd d �Zeje�d
�dd
d �Zejejej dddd�ddd �Zdd� Zedd� �Zedd� �Zedd� �Zedd� �ZdS ) �
AgentSchedule�agent_schedulesT�� primary_key��as_uuid��unique�default�index� agents.idF��nullable�r r � z08:00:00�r �server_defaultz17:00:00�partnership_accounts.id�CASCADE��onupdate�ondeletec C s( || _ || _|| _|| _|| _|| _d S �N)�day�available_from� available_to� is_active�partnership_account_id�agent_id)�selfr" r# r$ r% r& r' � r) �D/home/arjun/projects/buyercall/buyercall/blueprints/agents/models.py�__init__1 s
zAgentSchedule.__init__c G s. |D ]
}t j�tdi |��� qt j�� dS )zk
Return whether or not the agent schedule was created successfully.
:return: bool
TNr) )r �session�addr �commit)�cls�daysr" r) r) r* �create9 s
zAgentSchedule.createc C s2 t j�t�j|d��||||d�� t j�� d S )N)�sid)r% r# r$ r& )r r, �queryr � filter_by�updater. )r/ � schedule_idr# r$ r% r& r) r) r* r5 F s ��zAgentSchedule.updatec
C s� z5t j�tt j|kt j|k���� }|d ur3|d ur ||_|d ur$||_|d ur+||_ t
j � � W dS W dS t
yQ } zt�d�|j�� W Y d }~dS d }~ww )NTFz(Error updating agent schedule. Error: {})r r3 �filterr �idr& �firstr# r$ r% r r, r. � Exception�log�error�format�message)r/ r6 r# r$ r% r& �schedule�er) r) r* �
api_updateP s.
���
��zAgentSchedule.api_updatec C sp t j�t��ttj|ktj|k���� }|d ur6t j�t ��t j
|jk�� � }|D ]}d|_ q)t j�
� dS dS )NFT)r r, r3 �Agentr7 r r8 r& r9 r r' �allr% r. )r/ r' r& � agent_foundr �agent_scheduler) r) r* �
deactivaten s ��
zAgentSchedule.deactivateN)�__name__�
__module__� __qualname__�
__tablename__r �Column�Integerr8 r �uuid�uuid4r2 �
ForeignKeyr' �Booleanr% r" �Stringr# r$ r& r+ � classmethodr1 r5 rA rF r) r) r) r* r s> � ���
r � agent_groups�group_idr Tr
r' c s e Zd Zeg d��ZdZejejdd�Z eje
dd�de j dd�Z
ejddd �Zejd
dd d d
d�Zejejejdddd�ddd�Zejejejdddd�ddd�Zeje�d�dddd�Zeje�d�ddd�Ze�e�d��Zeje�d�ddd�Zeje�d�ddd�Zeje�d�ddd�Zejejddd�Ze�e�d��Z eje�� ddd�Z ejej ddd d �Z eje� � dd �Z!eje� � dd dd �Z"ejd e#e e#j$j%ke e#j$j&kd!d"�Z'eje�� de(e(d �Z)ejejddd�Z*e�ej �Z+� fd#d$�Z,e-d%d&� �Z.e-d'd(� �Z/e-d)d*� �Z0e-d+d,� �Z1e-d-d.� �Z2e-d/d0� �Z3e-d1d2� �Z4e-d3d4� �Z5e6d5d6� �Z7e6d7d8� �Z8e6d9d:� �Z9e6d;d<� �Z:e;d=d>� �Z<e;d?d@� �Z=e=j>dAd@� �Z=e;dBdC� �Z?e;dDdE� �Z@e;dFdG� �ZAe=jBdHd@� �Z=e=j>dId@� �Z=e;dJdK� �ZCeCj>dLdK� �ZC� ZDS )MrB ) )�none�None)�sales�Sales)zinternet saleszInternet Sales)�bdc�BDC)�service�Service)�
management�
Management)zclient relationshipzClient Relationship)�finance�Finance)�other�Other�agentsTr
r r �Lead�agent)�back_populatesr zall, delete-orphan�dynamiczasc(AgentSchedule.day))�backref�cascade�lazy�order_byzusers.idr r r r F� � )r r r r � � N�f)r r r r �0)r r r �groups)� secondary� primaryjoin�
secondaryjoinrh c s t t| �j|i |�� d S r! )�superrB r+ )r( �args�kwargs�� __class__r) r* r+ � s zAgent.__init__c C s� t j�tt jtkt j|k���� } | durGt j| }
|| _|| _ || _
|| _ || _ || _
|| _|
| _|
dur9|
| _| dur@| | _tj�� dS dS �zD
Update the agent's details.
:return: bool
NTF)rB r3 r7 r r8 r& r9 �
DEPARTMENT� firstname�lastname�title�email� phonenumber�mobile� extension� all_hours�
department�timezoner r, r. )r/ r& r r} r~ r� r� r� r� r� r� r� re �get_departmentr) r) r* r5 � s$ "
z Agent.updatec
C sx t j�tt j|kt j|k���� }
|
dur:|
|
_| |
_||
_ ||
_
||
_ | |
_ ||
_
||
_||
_| |
_tj�� dS dS r{ )rB r3 r7 r r8 r& r9 r� � descriptionr} r~ r r� r� r� r� r� r r, r. )r/ r& r' r r} r~ r� r� r� r� r� r� r� re r) r) r* rA � s& ��
zAgent.api_updatec
C s� t j�tt j|kt j|k���� }
|
durb|dur ||
_|dur#||
_|dur*||
_ | dur1| |
_
|
dur8|
|
_ | dur?| |
_ |durF||
_
|durM||
_|durT||
_| dur[| |
_tj�� dS dS r{ )rB r3 r7 r r8 r& r9 r r} r~ r� r� r� r� r� r� r� r r, r. )r/ r& r' r r} r~ r� � mobilenumberr� r� r� r� r� re r) r) r* �
api_update_v2 s: ��
zAgent.api_update_v2c C sR |st d�S d�|�}t ttj�|�tj�|�tj�|�tj�|�tj �|���S )z�
Search a resource by 1 or more fields.
:param query: Search query
:type query: str
:return: SQLAlchemy filter
rm z%{0}%)
r r= r rB r� �iliker} r~ r� r� )r/ r3 � search_queryr) r) r* �searchH s
��z Agent.searchc C s | j �| j�|���� S r! )r3 r7 r2 �in_rC )r/ �sidsr) r) r* �
get_from_sids_ s zAgent.get_from_sidsc C s< |}t di |��}tj�|� tj�� |j}tj�� |S )zb
Return whether or not the agent was created successfully.
:return: bool
Nr) )rB r r, r- �flushr8 r. )r/ �params� agent_paramsre r' r) r) r* r1 c s
z Agent.createc C s^ t j�tt j|kt j|k���� }|dur-d|_d|_t �
� |_ t j
�� t�|j|j�S dS )zc
Deactivate agent. The is_active flag will be set to false.
:return: bool
NFT)rB r3 r7 r r8 r& r9 r� �is_deactivatedr �now�deactivated_onr r, r. r rF )r/ r8 r& re r) r) r* rF w s "
zAgent.deactivatec C sB | j �t| j|k| j|k| j�d����� }|r |j}|S d}|S )zk
Lookup and return agent id based on agent full name comparison.
:return: integer
FN) r3 r7 r � full_namer& r� �is_r9 r8 )r/ r& �
agent_namere r' r) r) r* �reverse_agent_id_lookup� s
���z Agent.reverse_agent_id_lookupc C � | j �d�S )z5 Return the date/time this agent was created
�%Y-%m-%d %H:%M:%S)�
created_on�strftime�r( r) r) r* �created_datetime� � zAgent.created_datetimec C r� )z5 Return the date/time this agent was updated
r� )�
updated_onr� r� r) r) r* �updated_datetime� r� zAgent.updated_datetimec C s | j s | j �d�S dS )z= Return the date/time this agent was deactivated on.
r� rm )r� r� r� r) r) r* �deactivated_on_datetime� s z Agent.deactivated_on_datetimec C s� | j r
tdd� | jD ��S t�t�}| jpt}zt�|�}W n tjy/ t�d| � Y nw t �
tj �}|� |�
|��}|�� d d }d}|�d�}| j�� dksW| jr[d }|S | jD ] }|j|kr{|jr{|j| k rt|jk r{n q^d } |S q^|S )
Nc s s � | ]}|j V qd S r! )�
available_now)�.0�xr) r) r* � <genexpr>� s � z&Agent.available_now.<locals>.<genexpr>zInvalid timezone: '%s'� � Fz%H:%M:%Sr T)�is_group�anyrc �pytzr� �DEFAULT_TIMEZONE�UnknownTimeZoneErrorr; r< r r� �utc� normalize�
astimezone�weekdayr� � schedules�countr� r" r% r# r$ ) r( r� �tz_name�date�
local_dater� � foundresult�time�var_dayr) r) r* r� � s2
�
� � zAgent.available_nowc C sb ddl m} ddlm} |j�|j| jk��� }|r/|j�|j|j k��� }|j
� dd�}|S dS )Nr )�Endpoint)�Phonez+1rm ) �"buyercall.blueprints.mobile.modelsr� �(buyercall.blueprints.phonenumbers.modelsr� r3 r7 r' r8 r9 �
inbound_idr� �replace)r( r� r� � sip_endpoint�number�app_phone_numberr) r) r* �
app_number� s zAgent.app_numberc C s d� | j| j�S )Nz{} {})r= r} r~ r� r) r) r* r� � s zAgent.full_namec C � t j�| jd| j�S �N� �r �func�concatr} r~ �r/ r) r) r* r� � � c C �6 ddl m} |j�t|j| jk|j| jk���� }|S )Nr )�Contact) �$buyercall.blueprints.contacts.modelsr� r3 r7 r r& �agent_assignedr� rC )r( r� �leadsr) r) r* �
contact_leads� �
��zAgent.contact_leadsc C r� )Nr )rd ) �!buyercall.blueprints.leads.modelsrd r3 r7 r r& r' r8 rC )r( rd �callsr) r) r* �outbound_calls� r� zAgent.outbound_callsc C r� )Nr )�Message) � buyercall.blueprints.sms.modelsr� r3 r7 r r& r' r8 rC )r( r� �msgsr) r) r* �outbound_messages� r� zAgent.outbound_messagesc C s2 d|v r|� dd�\| _| _d S |d| _| _d S )Nr� r� rm )�splitr} r~ )r( �valuer) r) r* r� s c C r� r� r� r� r) r) r* r� r� c C s | j sdS dS )NrB �Group)r� r� r) r) r* �type
s z
Agent.typec C s t j| jdkdfgdd�S )N�tr� rB )�else_)r �caser� r� r) r) r* r� s )ErG rH rI r r| rJ r rK rL r8 r rM rN r2 � relationshipr� r� rO �user_idr& rQ r} r~ r r� r� r� r� r� r� rP r� �DateTimer� r� �agent_groups_table�crT r' rc r� r� �
contact_countr� r+ rR r5 rA r� r� r� r1 rF r� �propertyr� r� r� r� r r� r� �
expressionr� r� r� �setterr� �
__classcell__r) r) ry r* rB � s�
������
� �
�
�
0
&
rB �agent_team_association�team_idzteam.idc @ s� e Zd ZdZejejdd�Zejedd�de j
dd�Z eje� d�ddd�Z
eje� � ddd�Zejd ed
d �Zejejd d
�Zejejejdddd�dd d�Zejejejdddd�dd d�Zedd� �Zedd� �Zedd� �ZdS )�Team�teamTr
r r � rm r rB �teams)rs rh Fr r r r r zpartnerships.idc C �$ | j �| j|k��� }dd� |D �S )Nc S � g | ]}|j �qS r) �rc �r� r� r) r) r* �
<listcomp>4 � z.Team.get_agents_by_team_id.<locals>.<listcomp>)r3 r7 r8 rC �r/ �tidr� r) r) r* �get_agents_by_team_id1 � zTeam.get_agents_by_team_idc C r� )Nc S r� r) r� r� r) r) r* r� 9 r� z/Team.get_agents_by_team_sid.<locals>.<listcomp>)r3 r7 r2 rC r� r) r) r* �get_agents_by_team_sid6 r� zTeam.get_agents_by_team_sidc C sH | j �| j|k��� }g }|D ]
}|�dd� |jD �� qdd� |D �S )Nc S s g | ]}|�qS r) r) �r� re r) r) r* r� @ s z3Team.get_agent_sids_by_team_sid.<locals>.<listcomp>c S s g | ]}t |j��qS r) )�strr2 r� r) r) r* r� A s )r3 r7 r2 rC �extendrc )r/ r� r� rc r� r) r) r* �get_agent_sids_by_team_sid; s
z Team.get_agent_sids_by_team_sidN)rG rH rI rJ r rK rL r8 r rM rN r2 rQ �namer� r� � agent_team_association_tablerc rP r% rO r& �partnership_idrR r� r� r r) r) r) r* r� s0 �� ��
r� )!�loggingrM � collectionsr r r� �
sqlalchemyr r � sqlalchemy.dialects.postgresqlr �sqlalchemy.ext.hybridr �sqlalchemy.sqlr �buyercall.extensionsr � buyercall.lib.util_sqlalchemyr
� getLoggerrG r; r� �Modelr �TablerK rL rO r� rB r r� r) r) r) r* �<module> s8
l� �