Current File : //opt/cloudlinux/venv/lib/python3.11/site-packages/wmt/__pycache__/db.cpython-311.pyc
�
tf� � � � d dl Z d dlZd dlZd dlmZmZ d dlmZmZmZm Z m
Z
m Z m Z d dl
mZ d dlmZ d dlmZ d dlmZmZ e� � Z G d� d e� � Z G d
� d e� � Zd � Zd
� Zd� Zd� Z dd�Z ej defd�� � Z dS )� N)�datetime� timedelta)�Column�Boolean�DateTime�Integer�String�
create_engine�event)�declarative_base)� Inspector)�Session)�WMT_DB�RETENTION_TIME_DAYSc � � e Zd ZdZdZ eed�� � Z ee� � Z ee
d� d�� � Z ee � � Z
eed�� � Z eed�� � 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.3-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__r r �idr �websiter � create_dater � is_finished�
response_code�response_time_msr r r r r ! 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���r r c � � e Zd ZdZdZ eed�� � Z eed�� � Z ee
d� d�� � Z ee d� � � Z
d
S ) � DomainAlertsz0
alert_time is saved in local TZ format
�
domain_alertsTr )r c �( � t j � � S r r r r r r zDomainAlerts.<lambda>9 s � �(�,�.�.� r r F)r N)r" r# r$ r% r& r r r' r r( r �
alert_timer � is_resolvedr r r r. r. 1 su � � � � � �� � $�M� ���T� *� *� *�B��f�V�4�(�(�(�G����*@�*@��M�M�M�J��&��%�0�0�0�K�K�Kr r. c �� � t j � � t t �� � z
}t | � � 5 }|� t � � � t j |k � � � � � |� t � � � t j |k � � � � � d d d � � d S # 1 swxY w Y d S )N)�days) r �todayr r �
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!t j � | � � d S d S r )�
is_db_present�Base�metadata�
create_all)r: s r �create_db_if_not_existrC I s8 � � �� � � )� �
� � ��(�(�(�(�(�)� )r c �� � t j � t � � sdS t j | � � }d� |� � � D � � }t |� � dk S )NFc � � g | ]}|��S r r )�.0�tables r �
<listcomp>z!is_db_present.<locals>.<listcomp>R s � �
G�
G�
G��e�
G�
G�
Gr r )�os�path�isfiler r
� from_engine�get_table_names�len)r: �database_inspection�tabless r r? r? N s^ � �
�7�>�>�&� !� !� ��u�#�/��7�7��
G�
G�!4�!D�!D�!F�!F�
G�
G�
G�F� �v�;�;��?�r c �0 � | � d� � d S )NzPRAGMA journal_mode = WAL)�execute)� dbapi_con�
con_records r �setup_wal_moderU V s � �
���1�2�2�2�2�2r Fc � �� dt � ��| r�� d���fd�}t d|d�� � }t j |dt � � t |� � |S )Nzfile:z?mode=roc �0 �� t j � d�� � S )NT)�uri)�sqlite3�connect)�connection_strings �r r z setup_database.<locals>.<lambda>^ s �� �g�o�&7�T�B�B�B� r z sqlite:////F)�creator�echorZ )r r
r �listenrU rC )�readonlyr\ r: r[ s @r �setup_databaser` Z sz �� �(��(�(��� ;� 0� :� :� :��B�B�B�B�G�
��w�U�� � �F�
�L���N�3�3�3��6�"�"�"� �Mr �returnc # �� K � t | �� � } |V � |� � � n# |� � � � xY w |� � � dS # |� � � w xY w)z<Provide a transactional scope around a series of operations.)�bindN)r �commit�rollback�close)r: r< s r r6 r6 g s� � � � � �6�"�"�"�G���
�
�
����������������
���� �
�
�
��������
�
�������s �- �A �A�A � A4)F) rI �
contextlibrY r r �
sqlalchemyr r r r r r
r �sqlalchemy.ext.declarativer � sqlalchemy.engine.reflectionr
�sqlalchemy.ormr �wmt.common.constr r r@ r r. r= rC r? rU r` �contextmanagerr6 r r r �<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
Mini Shell