File: //home/arjun/projects/buyercall/buyercall/blueprints/activity/__pycache__/models.cpython-310.pyc
o
�weL � @ s� d dl Zd dlmZ d dlZd dlmZ d dlmZ d dl m
Z
m Z d dl m
Z
mZ d dlmZ G dd � d ee�ZG d
d � d ee�ZG d d
� d
e ej�ZdS )� N)�Enum)�DateTime)�db)�
AwareDateTime�
ResourceMixin)�tzware_datetime�timedelta_months)�Userc @ s e Zd ZdZdZdZdZdZdS )� ActivityType)�
authorization)�page)�modal)�data�creditN)�__name__�
__module__� __qualname__�
AUTHORIZATION�PAGE�MODAL�DATA�CREDIT� r r �F/home/arjun/projects/buyercall/buyercall/blueprints/activity/models.pyr
s r
c @ s8 e Zd ZdZdZdZdZdZdZdZ dZ
d Z d
Z d Z
d S )
� ActivityName)zlogout.success)z
login.success)z login.failed)zlogin.twofactauth.success)zlogin.twofactauth.failed)zlogin.acceptterms)zlogin.declineterms)�view)�edit)�pdf�downloadN)r r r �LOGOUT�
LOGIN_SUCCESS� LOGIN_FAILED�TWOFACTAUTH_SUCCESS�TWOFACTAUTH_FAILED�LOGIN_ACCEPTTERMS�LOGIN_DECLINETERMS�VIEW�EDIT�PDF�DOWNLOADr r r r r s r c @ sv e Zd ZdZejejdd�Zejejejddddd�ddd �Z eje�
d
�dd d �Z eje�
d
�dd d �Z eje�
d
�dd d �Z
eje�
d
�dd d �Zeje�
d
�dd d �Zeje�
d
�dd d �Zeje�
d
�dd d �Zejejdd
�Zejejejdddd�ddd �Zejejejdddd�ddd �Zed dd��Zedd� �Zedd� �Zedd� �Zedd� �Zed d � �ZdS ) � ActivityLogs�
activity_logsT)� primary_keyzusers.id�activity_logs_user_id_fkey�CASCADE)�name�onupdate�ondeleteF)�index�nullable�@ � )r3 �server_default)r3 zpartnerships.id)r0 r1 zpartnership_accounts.idNc
C s� d}d}t j�t j|k��� } | r| j}| j}z)t||||| �|�| � |�| �
|�| � |�|||d� }
t j
�|
� t j
�� W dS ty\ } zt�dt| � � W Y d} ~ dS d} ~ ww )z5
Create new user activity log entry
N) �user_id�typer/ � description�
ip_address�device_browser� device_os�resource_address� resource_id�partnership_id�partnership_account_idz%Error adding to activity log. Error: )r �query�filter�id�firstr? r@ r* �get_ip_address_from_request� get_device_browser_from_request�get_device_os_from_request�!get_resource_address_from_requestr �session�add�commit� Exception�log�error�str) �clsr7 r8 r/ �requestr> r9 �p_id�pa_id�user� new_activity�er r r �add_log> s4 �
��zActivityLogs.add_logc
C sb z|j �d�du r|j d W S |j d W S ty0 } zt�dt|� � W Y d}~dS d}~ww )z0
Retrieves the user IP address
�HTTP_X_FORWARDED_FORN� REMOTE_ADDRz:Error retrieving user IP address for activity log. Error: r5 )�environ�getrL rM rN rO �rP rQ rV r r r rE ^ s ��z(ActivityLogs.get_ip_address_from_requestc
C sZ z|r|j jd |j j W S W dS ty, } zt�dt|� � W Y d}~dS d}~ww )z<
Retrieves the user device browser details
�|r5 z>Error retrieving user device browser for activity log. Error: N)�
user_agent�browser�versionrL rM rN rO r\ r r r rF n s ��z,ActivityLogs.get_device_browser_from_requestc
C sN z
|r|j jW S W dS ty& } zt�dt|� � W Y d}~dS d}~ww )z7
Retrieves the user device os details
r5 z9Error retrieving user device os for activity log. Error: N)r^ �platformrL rM rN rO r\ r r r rG ~ s
��z'ActivityLogs.get_device_os_from_requestc
C sL z |r|j W S W dS ty% } zt�dt|� � W Y d}~dS d}~ww )z3
Retrieves the requested url path
r5 z=Error retrieving requested url path for activity log. Error: N)�pathrL rM rN rO r\ r r r rH � s ��z.ActivityLogs.get_resource_address_from_requestc C s� ddl m} |� }|t_|�� �- d}d}td�}tj�tj |k��
� }tj�tj |k�� � tj �
� |W d � S 1 s@w Y dS )zP
Delete activity logs 6 months and older.
:return: int
r )�
create_app�< i����N)�
buyercall.apprc r �app� app_contextr r* rA rB �
created_on�count�deleterI rK )rP rc rf � delete_count�expiration_days�limitr r r �delete_old_activity_logs� s
�z%ActivityLogs.delete_old_activity_logs)Nr5 ) r r r �
__tablename__r �Column�IntegerrC �
ForeignKeyr7 �Stringr8 r/ r9 r: r; r< r= r> r? r@ � classmethodrW rE rF rG rH rn r r r r r* " sT ������
r* )�loggingrM �enumr �datetime�
sqlalchemyr �buyercall.extensionsr � buyercall.lib.util_sqlalchemyr r �buyercall.lib.util_datetimer r � buyercall.blueprints.user.modelsr rO r
r �Modelr* r r r r �<module> s