File: //home/arjun/projects/buyercall/buyercall/blueprints/user/__pycache__/models.cpython-310.pyc
o
:�e�� � @ sH d dl Z d dlZd dlmZ d dlmZ d dlZd dlZd dl Z d dl
Z
d dl Z d dl m
Z
d dlmZ d dlmZ d dlmZ d dlZd dlmZmZ d d lmZ d d
lmZ d d l m Z m Z d d l m Z d d
l!m"Z" d dl#m$Z$m%Z% d dl&m'Z' d dl(m)Z) d dl*m+Z+ e�,e-�ZdZ.G dd� de e$j/�Z0G dd� dee e$j/�Z1dS )� N)� OrderedDict)�md5)� current_app)� UserMixin)�UUID)�URLSafeTimedSerializer)�or_�and_)�text)�hybrid_property)�
ResourceMixin�
AwareDateTime)�Agent)� Appointment)�db�bcrypt)�
ReportUserTie)�convert_to_deltatime)�
relativedeltai�Q c @ s� e Zd ZdZejejejddd�ddd�Zejejejddd�dd �Z ejd
e�
� d d d
�Z e dd� �Z
e dd� �Ze dd� �Ze dd� �Ze dd� �ZdS )�UserExternalApiAutoPostTie� user_external_api_access_tiezusers.id� users_id_user_access_tie_fkey)�nameT)� primary_key�indexz!external_api_service_providers.id�(service_provider_id_user_access_tie_fkey�r �
is_allowedF�0��nullable�server_defaultc C s t j�t j|k��� }|S )z�
Returns a list of existing user/external service provider ties.
:return: list of user/external api service provider tie objects
)r �query�filter�user_id�all)�clsr$ �ties� r( �B/home/arjun/projects/buyercall/buyercall/blueprints/user/models.py�get_existing_ties2 s z,UserExternalApiAutoPostTie.get_existing_tiesc C sn d}ddl m}m} tj�|��|��|j|k��|j |k|j
�� � }|r5t j�t j
|kt j|jk�� � }|S )z�
Returns an existing user/external service provider tie.
:return: a user/external api service provider tie object
N� ��0ExternalApiServiceProvidersPartnershipAccountTie�ExternalApiServiceProviders)�partnership.modelsr- r. r �sessionr" �joinr# r �partnership_account_id�active�firstr r$ � external_api_service_provider_id�id)r& r$ r2 �service_provider_name�service_provider_tier- r. �service_providerr( r( r) �!get_service_provider_access_state= s(
��
��z<UserExternalApiAutoPostTie.get_service_provider_access_statec
C s� ddl m}m} tj�|��|��|j|k��|j |k��
� }|rit j�t j |kt j
|jk��
� }|r?||_tj�tj�|_n#t � } || _ |j| _
|| _tj�tj�| _tj�tj�| _tj�| � tj�� dS dS )z<
Sets a user/external service provider tie.
r+ r, N)r/ r- r. r r0 r" r1 r# r r2 r4 r r$ r5 r6 r �datetime�now�pytz�utc�
updated_on�
created_on�add�commit)
r& r$ r2 r7 �valuer- r. r9 r8 �new_tier( r( r) �!set_service_provider_access_stateX s8 �
�� �z<UserExternalApiAutoPostTie.set_service_provider_access_statec C sl ddl m}m} ddlm} tj�|��|�� |j
|k|j dk��t �� t j
|kt jdk��|j��� }|S )z�
Returns a list of service providers the user has access to.
:return: a external api service provider list
r+ r, r )�FormLogT)r/ r- r. �&buyercall.blueprints.form_leads.modelsrF r r0 r" r1 r# r2 r3 r r$ r �order_byr r% )r& r$ r2 r- r. rF �service_providersr( r( r) � get_allowed_service_providers{ s
��
z8UserExternalApiAutoPostTie.get_allowed_service_providersc C s� ddl m}m} ddlm} g }tj�|j|j ��
|�� |j |k|j
dk��
t�� tj|ktjdk��� }|D ] } | d | d g}
|�|| d �} |
�| � |�|
� q7|S )z�
Returns a list of service providers the user has access to including if form logs exist.
:return: a external api service provider list
r+ r, r )�ExternalApiFormLeadPostTieT� )r/ r- r. rG rK r r0 r" r6 r r1 r# r2 r3 r r$ r r% �existing_posts_for_lead�append) r& r$ r2 � form_lead_idr- r. rK �resultrI �provider� new_entry�
logs_existr( r( r) �1get_allowed_service_providers_with_form_log_count� s$
��
zLUserExternalApiAutoPostTie.get_allowed_service_providers_with_form_log_countN)�__name__�
__module__� __qualname__�
__tablename__r �Column�Integer�
ForeignKeyr$ r5 �Booleanr � classmethodr* r: rE rJ rT r( r( r( r) r " s. ���
"
r c s� e Zd Zeg d��Zeg d��ZdZdZej ej
dd�Z ej e dd�de
jdd�Zej ej
ejd d
d
d �ddd �Zej ej
ejd
d
d
d �ddd �Zejedddd�Zejeddd�Zejeddd�Zej ejeddi�dddd�Zej de� � ddd�Z ej e� � dd �Z ej e� � dd dd �Z ej e�!d �ddd �Z"ej e�!d!�dddd"d#�Z#ej e�!d$�dd"d�Z$ej d%e� � dd d�Z%ej d&e� � ddd�Z&ej d'e� � ddd�Z'ej d(e� � ddd�Z(ej d)e� � ddd�Z)ej d*e� � dd d�Z*ej ej
dd+d,�Z+e� e,� �Z-e� e�!d-��Z.e� e,� �Z/e� e�!d-��Z0ej e�!d$�ddd"d�Z1ej e�!d$�ddd"d�Z2ej ej ddd�Z3ej e�
� dd.d,�Z4ej e� � dd dd �Z5e6d/d0� �Z7e6d1d2� �Z8e8j9d3d2� �Z8ej e�!d4�dd"d�Z:ej e�!d5�dd"d�Z;e� ej
�Z<e� e�!d$��Z=e� e�!d$��Z>ej ej ddd6d7�Z?ej ej ddd6d7�Z@ej ej ddd6d7�ZAej ej ddd6d7�ZBej ej ddd6d7�ZCej ej
ejd8d
d
d �ddd �ZDej e�!d9�dd:d�ZEej e� � dd6d�ZFej ej ddd6d7�ZGej e�!d;�dd"d�ZHej e� � dd6d�ZIej e� � dd<d�ZJ� fd=d>�ZKeLd?d@� �ZMeLdAdB� �ZNeLdCdD� �ZOeLdEdF� �ZPeLdGdH� �ZQeLdIdJ� �ZReLdKdL� �ZSeLdMdN� �ZTeLdOdP� �ZUeLdQdR� �ZVeLdSdT� �ZWeLdUdV� �ZXeLdWdX� �ZYeLd�dZd[��ZZeLd�d\d]��Z[eLd�d^d_��Z\d`da� Z]dbdc� Z^d�ddde�Z_d�dgdh�Z`didj� Zae6dkdl� �Zbe6dmdn� �Zce6dodp� �Zde6dqdr� �Zee6dsdt� �Zfe6dudv� �Zge6dwdx� �Zhe6dydz� �Zie6d{d|� �Zje6d}d~� �Zke6dd�� �Zle6d�d�� �ZmeLd�d�� �ZneLd�d�� �Zoe6d�d�� �Zpd�d�� Zqd�d�� ZreLd�d�d���Zs� ZtS )��User))�guest�Guest)�member�Member)�admin�Admin)�agentr )�partner�Partner)�
limitsysadminzLimited System Admin)�sysadminz System Admin) )�none�None)�sales�Sales)zinternet saleszInternet Sales)�bdc�BDC)�service�Service)�
management�
Management)zclient relationshipzClient Relationship)�finance�Finance)�other�Other�users)&�sid�partnership_idr2 z agent.sidzagent.timezonezreports.report_idzappointments.sid�roler3 �deactivated_on�is_deactivated�email�full_feature_accessr@ �
sign_in_count�current_sign_in_on�current_sign_in_ip�last_sign_in_on�last_sign_in_ip� firstname�lastname�appointment_enabled�appointment_interval� phonenumber�company� extension�title�
department�
leads_onboard�agents_onboard�outbound_onboard�inbound_onboard�locale�two_factor_auth� dark_mode� user_avatar�support_notification� push_notification_subscriptionTr )�as_uuid)�unique�defaultr zpartnerships.id�CASCADE)�onupdate�ondelete)r r zpartnership_accounts.id�userF)�backref�uselist�passive_deleteszall, delete-orphan)�back_populates�cascade�partnership_account)r� r� r �
role_typesra )r r r! � is_active�1r )r r )r r! r� � )r� r � � )r� r r r! � �
is_tos_agreedr �inbound_routing_access�outbound_routing_access� forms_access�external_api_accessr )r r� �- � c C s | j rdS dS )z@
Whether or not the user account is a BDC user.
TF)� external_bdc��selfr( r( r) � is_bdc_user$ s zUser.is_bdc_userc C s6 | j s| jsd S d�| j sdn| j | jsd�S | j�S )Nz{} {}r� )r� r� �formatr� r( r( r) r . s ��z User.namec C s4 | j s| jsd S | j s
dS | js| j d d S | jS )Nr� )r� r� )r& r( r( r) r 7 s $� � �false)r r� r! z partnership_account_groups.id� �eni� �truec s. t t| �jdi |�� t�|�dd��| _d S )N�passwordr� r( )�superr^ �__init__�encrypt_password�getr� )r� �kwargs�� __class__r( r) r� u s z
User.__init__c C sL t j�tt j|kt j|k���� }|dur$d|_t� � |_
t j �
� dS dS )zb
Deactivate user. The is_active flag will be set to false.
:return: bool
NFT)r^ r" r# r r6 r2 r4 r3 r; r< r| r r0 rB )r& r6 r2 r� r( r( r) �
deactivate{ s "
zUser.deactivatec
C s^ t �| �}
|
d ur-| dvrd} t ||||||||| |
| |
dd�
}tj�|� tj�� dS dS )N�r_ ra re rf ra F)
r� r� r� r� r� r~ r� r� r{ �usernamer2 r� �
tos_agreementT)r^ r� r r0 rA rB )r& r� r� r� r� r� r~ r� r� r{ r� r2 r� � new_password�new_userr( r( r) �create� s
�
z User.createc
C s~ t j�tt j|kt j|k���� } | dur=|| _|| _|| _ || _
|| _ |
| _ || _
| | _| dvr3d} | | _tj�� dS dS )zC
Update the user's details.
:return: bool
Nr� ra TF)r^ r" r# r r6 r2 r4 r� r� r� r� r~ r� r� r� r{ r r0 rB )
r& r6 r2 r� r� r� r� r� r~ r� r� r{ r� r( r( r) �update� s "
z User.updatec C sN |st d�S d�|�}tj�|�tj�|�tj�|�tj�|�f}t t|� �S )z�
Search a resource by 1 or more fields.
:param query: Search query
:type query: str
:return: SQLAlchemy filter
r� z%{0}%) r
r� r^ r~ �iliker� r� r� r )r& r" � search_query� search_chainr( r( r) �search� s
� z User.searchc C s. t j�t j|kt j|kB ��t j�d���� S )z�
Find a user by their e-mail or username.
:param identity: Email or username
:type identity: str
:return: User instance
F)r^ r" r# r~ r� r} �is_r4 )r& �identityr( r( r) �find_by_identity� s
��zUser.find_by_identityc C s |r t �|d��d�S dS )z�
Hash a plaintext string using bcrypt.
:param plaintext_password: Password in plain text
:type plaintext_password: str
:return: str
� �utf-8N)r �generate_password_hash�decode)r& �plaintext_passwordr( r( r) r� � s zUser.encrypt_passwordc C sD t jd }ztj||dgd�}t�|�d��W S ty! Y dS w )z�
Obtain a user from de-serializing a signed token.
:param token: Signed token.
:type token: str
:return: User instance or None
�
SECRET_KEY�HS256)�
algorithms�
user_emailN)r �config�jwtr� r^ r� r� � Exception)r& �token� private_key�decoded_payloadr( r( r) �deserialize_token� s
�zUser.deserialize_tokenc C sf |j dko|dk}|jdu o|du }|s|r1tj�tj dktj|jktj|jk��� }|dkr1dS dS )�=
Determine whether this user is the last admin account.
:param user: User being tested
:type user: User
:param new_role: New role being set
:type new_role: str
:param new_active: New active status being set
:type new_active: bool
:return: bool
rc TNrL F)r{ r3 r^ r" r# rz r2 �count)r& r� �new_role�
new_active�is_changing_roles�is_changing_active� admin_countr( r( r) �
is_last_admin s
��zUser.is_last_adminc C s\ |j dko|dk}|jdu o|du }|s|r,tj�tj dktj|jk��� }|dkr,dS dS )r� rf TNrL F)r{ r3 r^ r" r# rz r� )r& r� r� r� r� r� �
partner_countr( r( r) �is_last_partner s
��zUser.is_last_partnerc C s6 t j�| ��| j�|��jddidd�}t j�� |S )z�
Override the general bulk_delete method because we need to delete them
one at a time while also deleting them on Stripe.
:param ids: List of ids to be deleted
:type ids: list
:return: int
r3 F�fetch)�synchronize_session)r r0 r" r# r6 �in_r� rB )r& �ids� delete_countr( r( r) � bulk_delete8 s
�
zUser.bulk_deletec C sB d}t j�t j|k��� }|r d|_tj�� |_t j
� � d}|S )z�
Deactivates a user. Sets the is_deactivated to TRUE and sets the deactivated_on date.
:param id: User ID to be deactivated
:type id: int
:return: boolean
FT) r^ r" r# r6 r4 r} r; r< r| r r0 rB )r& r6 rP r� r( r( r) �deactivate_userJ s
zUser.deactivate_userc C s� t �|�}|�� }ddlm} ddlm}m} |j� |j
|j k�� � }|j
dkr5|j� |j
|jk�� � }n |j� |j
|jk�� � }|�|j
||j|j� |S )z�
Generate a token to reset the password for a specific user.
:param identity: User e-mail address or username
:type identity: str
:return: User instance
r )� deliver_password_reset_emailr+ )� Partnership�PartnershipAccountrf )r^ r� �serialize_token� buyercall.blueprints.user.tasksr� r/ r� r� r" r# r6 r2 r4 r{ rz �delayr �logo) r& r� �u� reset_tokenr� r� r� �partner_accountrf r( r( r) �initialize_password_reset] s
�
z User.initialize_password_resetc C s� | j �| j|k��� }tj�|d�}|�� d d }t|�}|tj|j d� }tj�t
|�d�}tj�t
|�d�} |� � }
| � � } t
|�|t
|
�dd� t
| �dd� d�} | � |j|�}
| |
v rcd S d
S ) aR
Check if a user is available for a specific date and time
:param user_id: User's system id
:param date: The request appointment date
:param start_time: The request appointment start time
:type user_id: int
:type date: string
:type start_time: string
:return: True
�%Y-%m-%drL � ��minutes�%H:%M:%SN�����date�weekday�
start_time�end_timeTF)
r" r# r6 r4 r; �strptimer r � timedeltar� �str�time� user_available_appointment_slots)r& r$ r r
r� � format_date� date_weekdayr �format_start_time�format_end_time�final_start_time�final_end_time�requested_appointment�available_slotsr( r( r) � user_appointment_availabilityx s$ �z"User.user_appointment_availabilityNc $ C s� | j �t| j|k| j�d�| j�d����� }tj� t
tj � � �d�}|j
j}t�|�}tj�|�}|�d�}|rUtj� |d�} t
| tdd� �dd� }
tj� |
d�} n|} t
| tdd� �dd� }
tj� |
d�} �zW|�r�g } | }
|
| kr�|
�� d d
}|
|d �}| �|� |
td d �7 }
|
| ksxg }|j
jr�|j
j�� }nd
}|d
ks�|j
jdur�| D ])}|j
jD ]"}|j|d kr�|jr�|j}|j}|d |d ||d�}|�|� q�q�n| D ]}d}d}|d |d ||d�}|�|� q�t j �tt j |kt j�d�t j | kt j | k��� � }g }|D ]"}|j D ]}t
|j �|j �� d d
|j!|j"d�}|�|� �q�qg }|D ]�} | d }|| d k �r�|dk �r�t#|dd� �} | tj$|j%d� } tj� t
| �d�} tj� t
| �d�} | �&� }!| �&� }"t
| d �| d t
|!�dd� t
|"�dd� d�}#|#|v�r�t
|#d �t
|�k�r�|#d |k �r�n|�|#� t
|"�}|| d k �r�|dk �sK�q9|W S W dS t'�y� t(�)d�*t+�,� �� Y dS w )aC
Retrieve a list of start and end time appointment slots for a specific date an user
:param user_id: User's system id
:param date: The date in question
:type user_id: int
:type date: string
:return: list of appointment slot dictionaries for a specific date and user
FTr r r+ )�monthsNi����rL r )r r )�daysr r r )r r �schedule_start_time�schedule_end_timez00:00:00z23:00:00r r r r r r
z2User available appointment slots error message: {})-r" r# r r6 r} r� r� r4 r; r r r �todayre �timezoner= r< �strftimer r rN � schedulesr� � all_hours�dayr� �available_from� available_tor r$ �appointment_dater% �appointment_slotr
r r r
r� r r� �log�errorr� � traceback�
format_exc)$r& r$ r r� �
date_today�account_timezone�tz�time_now�account_time_nowr � future_date�format_future_date�dates_with_weekdays�
start_dater � date_and_day�dates_with_schedules�schedule_count�var_dayr r �available_days� appointments�unavailable_slots� appointmentr' �appointment_time_slotr �
schedule_dater
r r r r r �slotr( r( r) r � s�
��
� �
��� �
��
��
�
�
��K�z%User.user_available_appointment_slotsc C sn |r| j �t| j|k| jdk��}n | j �| j|k�}|r-|r-|�t| j|k| j|k ��}|r3|�� S |�� S )a1
Get users by partnership id.
:param: is_active_only - Boolean (Is only active users needed)
:param: count_only - Boolean (Is only count of users needed)
:param: from_date - Datetime
:param: to_date - Datetime
:return: User objects or objects count
T)r" r# r rz r3 r@ r� r% )r& �pid�is_active_only�
count_only� from_date�to_daterx r( r( r) �get_by_partnership s
� � zUser.get_by_partnershipc
C s� |rF|dkr t jtjd�}|jdd�}| j||||d�S |dkrDt jtjd�}|jd }|jdd�}|j|dd�}|} | j|||| d�S d S | j||d�S )
z^
:param date_range_expr: type:string, possible values: this_month, last_month
�
this_month)r. rL )r# )rB rC rD �
last_month)�monthr# )rB N)�pendulumr r= r> �replacerE rH )
r& r@ rB �date_range_exprr � first_day� pre_month�first_day_this_month�first_day_pre_month�last_day_pre_monthr( r( r) � get_by_partnership_datetime_expr s
�z%User.get_by_partnership_datetime_exprc C s | j S )z�
Return whether the user account is active, this satisfies
Flask-Login by overwriting the default value.
:return: bool
)r3 r� r( r( r) r� 2 s zUser.is_activec C s: t jd }t|�}t| j�t| j��� g}|�|�� d�S )as
Return the user's auth token. Use their password as part of the token
because if the user changes their password we will want to invalidate
all of their logins across devices. It is completely fine to use
md5 here as nothing leaks.
This satisfies Flask-Login by providing a means to create a token.
:return: str
r� r� )
r r� r r r6 r r� � hexdigest�dumpsr� )r� r� �
serializer�datar( r( r) �get_auth_token; s
zUser.get_auth_tokenc C s |r t �| j|�S dS )a+
Ensure a user is authenticated, and optionally check their password.
:param with_password: Optionally check their password
:type with_password: bool
:param password: Optionally verify this as their password
:type password: str
:return: bool
T)r �check_password_hashr� )r� �
with_passwordr� r( r( r) �
authenticatedM s
zUser.authenticated� c C s< t jd }| jtj�� tj|d� d�}tj||dd�}|S )a
Sign and create a token that can be used for things such as resetting
a password or other tasks that involve a one off token.
:param expiration: Seconds until it expires, defaults to 1 hour
:type expiration: int
:return: JSON
r� )�seconds)r� �expr� )� algorithm)r r� r~ r; �utcnowr
r� �encode)r� �
expirationr� �payloadr� r( r( r) r� \ s
�zUser.serialize_tokenc C s< | j d7 _ | j| _| j| _tj�tj�| _|| _| � � S )a
Update various fields on the user that's related to meta data on his
account, such as the sign in count and ip address, etc..
:param ip_address: IP address
:type ip_address: str
:return: SQLAlchemy commit results
rL )
r� r� r� r� r� r; r< r= r> �save)r� �
ip_addressr( r( r) �update_activity_trackings s z User.update_activity_trackingc C s d}| j dkr d}t| |�S )N� partnershiprc r� )r{ �getattr)r� �depr( r( r) �"partnership_or_partnership_account� s
z'User.partnership_or_partnership_accountc C s: ddl m} |j�|j| jk��� }|s|j�d�}|jS �Nr+ �r� rL ) r/ r� r" r# r6 rz r4 r� ry �r� r� re r( r( r) �partnership_sid� s
zUser.partnership_sidc C s6 ddl m} d }|j�|j| jk��� }|r|j}|S )Nr+ �r� )r/ r� r" r# r6 r2 r4 ry )r� r� �partnership_account_sidr� r( r( r) rn � s z User.partnership_account_sidc C sL | j r| j jr | j jS | j jr| j jjr| j jjS | jr$| jjr$| jjS d S �N)r� � subscriptionre r� r( r( r) rp � s
zUser.subscriptionc C � | j dur | j dkr dS dS )zS
Whether the user account is used in a partnership account or not.
Nr TF)r2 r� r( r( r) �is_partnership_account_user� � z User.is_partnership_account_userc C rq )zK
Whether the user account is used in a partnership or not.
Nr TF)rz r� r( r( r) �is_partnership_user� rs zUser.is_partnership_userc C s^ ddl m} | jdkr-| jr-| jr-| jdkr-|j�t|j| jk|j| jk���� }|r-dS dS )z�
Whether the user account is used to view/acccess/change partnership account groups.
Checks: Must be an admin role, must have a partnership account ID and must belong to a group.
r+ ��PartnershipAccountGroupTierc r TF) r/ rv r{ � partnership_account_group_idr2 r" r# r r� )r� rv �group_tie_existsr( r( r) �is_admin_user_with_groups� s
���z User.is_admin_user_with_groupsc
C s� t jtjd tjd d�}ddlm} zG| jdkrX| jrX| jrX| jdkrX|j �
t |j| jk|j| jk��� � }|rX|dkrXd�
| j�}|�|�d krXt|�|��}|dkrXW d
S W d S tyr } z t�d � W Y d
}~d S d
}~ww )zL
Whether the user account is viewing a partnership account.
�REDIS_CONFIG_URL�REDIS_CONFIG_PORT��host�portr+ ru rc r �partner-user-view:{}rL TFz,Error retrieving viewed partnership account.N)�redis� StrictRedisr r� r/ rv r{ rw r2 r" r# r r� r� r6 �exists�intr� r� r( r) )r� �redis_dbrv rx �keyrP �er( r( r) �is_viewing_partnership� s8 � ����
��zUser.is_viewing_partnershipc
C s� t jtjd tjd d�}d�| j�}d}|�|�dkrDz
t|�|��}W |S t yC } zt
� d�|�� d}W Y d}~|S d}~ww |S ) z�
Whether the user account is currently viewing (acting on behalf) of a partnership account.
Returns -1 if not viewing, partnership account id if viewing.
rz r{ r| r ���rL z6Error retrieving viewed partnership account. Error: {}N) r� r� r r� r� r6 r� r� r� r� r( r) )r� r� r� rP r� r( r( r) �'get_user_viewing_partnership_account_id� s � ���z,User.get_user_viewing_partnership_account_idc
C �� t jtjd tjd d�}ddlm} z8d}d�| j�}|�|�dkrCt |�
|��}|d krF|j � |j|k��
� }|d
urI|j}W |S W |S W |S W |S tyh } zt�d �|�� d}W Y d
}~|S d
}~ww ) �>
Returns the viewed partnership account name.
rz r{ r| r+ rm r� r rL r N�9Error retrieving viewed partnership account name. Error: )r� r� r r� r/ r� r� r6 r� r� r� r" r# r4 r r� r( r) �r� r� r� rP r� r2 �accountr� r( r( r) �)get_user_viewing_partnership_account_name �4 � � � ����z.User.get_user_viewing_partnership_account_namec
C r� ) z4
Returns the viewed partnership id.
rz r{ r| r+ rm r� r rL r Nz1Error retrieving viewed partnership id. Error: {})r� r� r r� r/ r� r� r6 r� r� r� r" r# r4 rz r� r( r) r� r( r( r) � get_user_viewing_partnership_id+ r� z$User.get_user_viewing_partnership_idc
C s. t jtjd tjd d�}ddlm} zfd}d�| j�}|�|�dkrkt |�
|��}|dkrn|j � |j|k��
� }|d urq|jrK|jjrK|jj}W |S |jjr\|jjjr\|jjj}W |S |jrt|jjrw|jjj}W |S W |S W |S W |S W |S W |S ty� } zt�d
�|�� d}W Y d }~|S d }~ww ) r� rz r{ r| r rm r� r rL Nr� )r� r� r r� �'buyercall.blueprints.partnership.modelsr� r� r6 r� r� r� r" r# r4 rp �planre r� r( r) r� r( r( r) �6get_user_viewing_partnership_account_subscription_planH sR � �
� � ���
�����z;User.get_user_viewing_partnership_account_subscription_planc
C s� t jtjd tjd d�}d�|�}z|�||� |�|t� d}W |S ty@ } zt �
d�|�� d}W Y d}~|S d}~ww ) z�
Sets the partnership account id of the account the user is viewing.
Set -1 if not viewing, partnership account id if viewing.
rz r{ r| r Tz3Error setting viewed partnership account. Error: {}FN) r� r� r r� r� �set�expire�DAYSr� r( r) )r& r$ r2 r� r� rP r� r( r( r) �$set_user_viewing_partnership_accounto s �
���z)User.set_user_viewing_partnership_accountc
C s ddl m} d}tj�tj|k��� }|r�|jdkr�|jr�|j r�|j dkr�zM|j�t
|j |j k|j|jk���� }|j�t
|j |k|j|jk���� }|rd|jdkrg|rj|jdkrm|j|jkrpd}W |S W |S W |S W |S W |S W |S t y� } zt �
d�|�� W Y d}~|S d}~ww |S ) ze
Whether the user account is in the same group as the passed partnership account id.
r+ ru Frc r TzJError checking if user is in the same partnership account group. Error: {}N)r/ rv r^ r" r# r6 r4 r{ rw r2 r r� r( r) r� ) r& r$ r2 rv rP r� �group_tie_user�group_tie_otherr� r( r( r) �is_in_same_group� s^ ����
�������������zUser.is_in_same_groupc C st | j dur8ddlm} |j�|j| j k��� }|r |jS | jdur8ddlm } |j�|j| jk��� }|r8|jS dS )zJ
Retrieve the user business type from partnership account
Nr+ rm rj r� )
r2 r/ r� r" r# r6 r4 �
business_typerz r� )r� r� r� r� re r( r( r) r� � s&
�
�zUser.business_typec C �8 ddl m} |j�|j| jk��� }|s|j�d�}|S ri )r/ r� r" r# r6 rz r4 r� rk r( r( r) �get_partnership� �
zUser.get_partnershipc C r� )Nr+ rm rL )r/ r� r" r# r6 r2 r4 r� )r� r� r� r( r( r) �get_partnership_account� r� z User.get_partnership_accountc C s: |d u rg S dd� t j�| j��| j�|���� D �}|S )Nc S s g | ]}t |d ��qS )r )r )�.0�uidr( r( r) �
<listcomp>� s z*User.get_sid_by_emails.<locals>.<listcomp>)r r0 r" ry r# r~ r� r% )r& �
recipients�
users_sidsr( r( r) �get_sid_by_emails� s *zUser.get_sid_by_emailsro )TFNN)FN)Tr� )rZ )urU rV rW r �ROLE�
DEPARTMENTrX �serialize_onlyr rY rZ r6 r �uuid�uuid4ry r[ rz r2 � relationshipr re r �reportsr r: �Enumr{ r\ r3 �DateTimer| r} �Stringr� r~ r� r� r r� r� r� r� r� r
r� r� r� r� r� r� r� r� r� r r� r �
expressionr� r� r� r� r� r� r� r� r� �two_factor_auth_onboardrw r� r� r� r� r� r� r� r] r� r� r� r� r� r� r� r� r� r� r� r r r rE rQ r� rV rY r� rd rh rl rn rp rr rt ry r� r� r� r� r� r� r� r� r� r� r� �
__classcell__r( r( r� r) r^ � sj
������� ��� �� �
�
�
�
�
���
� �
l
&
"
r^ )2r; r* � collectionsr �hashlibr �loggingr( r� rI r� r= �flaskr � flask_loginr � sqlalchemy.dialects.postgresqlr � itsdangerousr r� �
sqlalchemyr r �sqlalchemy.sqlr
�sqlalchemy.ext.hybridr � buyercall.lib.util_sqlalchemyr r
�"buyercall.blueprints.agents.modelsr �(buyercall.blueprints.appointments.modelsr �buyercall.extensionsr r �#buyercall.blueprints.reports.modelsr �buyercall.lib.util_datetimer �dateutil.relativedeltar � getLoggerrU r� �Modelr r^ r( r( r( r) �<module> s: