File manager

�

b(�h���
�dd
lZdd
l
Z
dd
lZddlmZmZ
ddlmZmZmZm	Z	m
Z
mZmZ
ddl
mZ
ddlmZ
ddlmZ
ddlmZmZ
e��ZGd�d	e��ZGd
�de��Zd�Zd
�Zd�Zd�Zdd�
Ze
jdefd���Zd
S)�N)�datetime�	timedelta)�Column�Boolean�DateTime�Integer�String�
create_engine�event)
�declarative_base)
�	Inspector)
�Session)�WMT_DB�RETENTION_TIME_DAYSc��eZ
dZd
ZdZeed���Zee�
�
Z	ee
d�d���Zee�
�
Z
eed���Zeed���ZdS)	�ScrapeResultz2
    created_date is saved in local TZ format
    �
scrape_resultT�
�primary_keyc�(�t
j
��S�
N�r�now����/builddir/build/BUILDROOT/cl-web-monitoring-tool-3.1.8-1.el8.cloudlinux.x86_64/opt/cloudlinux/venv/lib/python3.11/site-packages/wmt/db.py�<lambda>zScrapeResult.<lambda>)s��8�<�>�>�r��default�index)
�nullableN)�__name__�
__module__�__qualname__�__doc__�
__tablename__rr�idr	�websiter�create_dater�is_finished�
response_code�response_time_msrrrrr!s���������$�M�	���T�	*�	*�	*�B��f�V�n�n�G��&��+A�+A��N�N�N�K��&��/�/�K��F�7�T�2�2�2�M��v�g��5�5�5���rrc��eZ
dZd
ZdZeed���Zeed���Z	ee
d�d���Zeed�	��Z
d
S)�DomainAlertsz0
    alert_time is saved in local TZ format
    �
domain_alertsTr)
r c�(�t
j
��SrrrrrrzDomainAlerts.<lambda>9s��(�,�.�.�rrF)
rN)r"r#r$r%r&rrr'r	r(r�
alert_timer�is_resolvedrrrr.r.1su��������$�M�	���T�	*�	*�	*�B��f�V�4�(�(�(�G����*@�*@��M�M�M�J��&��%�0�0�0�K�K�Krr.c
��
�t
j
��tt�
�
�
z
}
t	|�
�
5}|�t�
�
�tj|
k�
�
�	��
|�t�
�
�tj|
k�
�
�	��
ddd��
dS#1swxYw

Y

dS)N)
�days)r�todayrr�
session_scope�queryr�filterr)�deleter.r1)�engine�
n_days_ago�sessions   r�cleanup_old_datar=>s�����!�!�I�3F�$G�$G�$G�G�J�	�v�	�	��'��
�
�l�#�#�
�V�L�,�z�9�
:�
:�
�V�X�X�X��
�
�l�#�#�
�V�L�+�j�8�
9�
9�
�V�X�X�X�
��������������������s�BC!�!C%�(
C%c
�f�t
|�
�
s!tj�|�
�

dSdSr)�
is_db_present�Base�metadata�
create_all)
r:s
 r�create_db_if_not_existrCIs8���� � �
)��
� � ��(�(�(�(�(�
)�
)rc
���tj
�t�
�
sd
St	j|�
�
}
d�|
���D��}t|�
�
dkS)NFc
��g|]}
|
��Srr)�.0�tables  r�
<listcomp>z!is_db_present.<locals>.<listcomp>Rs��
G�
G�
G��e�
G�
G�
Grr)�os�path�isfilerr
�from_engine�get_table_names�len)r:�database_inspection�tabless   rr?r?Ns^��
�7�>�>�&�!�!�
��u�#�/��7�7��
G�
G�!4�!D�!D�!F�!F�
G�
G�
G�F��v�;�;��?�rc�0�|�d
�
�

dS)NzPRAGMA journal_mode = WAL)
�execute)�	dbapi_con�
con_records  r�setup_wal_moderUVs��
���1�2�2�2�2�2rFc
���d
t���|r��d���f
d�}
td|
d���}tj|dt��
t|�
�

|S)Nzfile:z?mode=roc�0�
�t
j
�d
���S)NT)
�uri)�sqlite3�connect)
�connection_strings
�rrz setup_database.<locals>.<lambda>^s���g�o�&7�T�B�B�B�rzsqlite:////F)�creator�echorZ)rr
r�listenrUrC)�readonlyr\r:r[s   @r�setup_databaser`Zsz���(��(�(���
;�0�:�:�:��B�B�B�B�G�
��w�U����F�
�L���N�3�3�3��6�"�"�"��Mr�returnc
#��K
�t
|�
�
�
}
	|
V�

|
�
��
n#
|
���
�xYw
	|
���
dS#|
���
wxYw
)z<Provide a transactional scope around a series of operations.)
�bindN)r�commit�rollback�close)r:r<s  rr6r6gs������6�"�"�"�G���
�
�
����������������
����	�
	�
�
��������
�
�������s�-�
A�A�A�A4)
F) rI�
contextlibrYrr�
sqlalchemyrrrrr	r
r�sqlalchemy.ext.declarativer�sqlalchemy.engine.reflectionr
�sqlalchemy.ormr�wmt.common.constrrr@rr.r=rCr?rUr`�contextmanagerr6rrr�<module>rn
s�
�


�

�	�	�	���������(�(�(�(�(�(�(�(�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
8�7�7�7�7�7�2�2�2�2�2�2�"�"�"�"�"�"�8�8�8�8�8�8�8�8������

6�

6�

6�

6�

6�4�

6�

6�

6� 

1�

1�

1�

1�

1�4�

1�

1�

1�
�
�
�
)�
)�
)�

�
�
�

3�

3�

3�

�

�

�

��
�

�W�

�

�

��
�

�

�

r