Sindbad~EG File Manager
o
�I�e � @ s� d dl mZmZmZmZmZmZmZmZ d dl m
Z
m Z d dl m
Z
d dlZd dlZd dlmZmZmZ ee
jd ddd �Zee
jd
ddd �ZG d d � d �ZdS )
� )�
create_engine�MetaData�Table�insert�select�update�delete�text)�and_�or_)�appN)� timedelta�date�datetime� DATABASE_URIi T)� pool_recycle�future�DATABASE_URI_FKc @ sd e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dS )�KitModelc
C sP zt � | _td| jtd�| _W d S ty' } z t|� W Y d }~d S d }~ww )N� kc_sessions)�
autoload_with)r �metar � engine_fkr � Exception�print)�self�e� r �D/home/numerotech/scans.numerotech.com/scan_v1/core/model/KitModel.py�__init__
s ��zKitModel.__init__c C s� t �� �3}tdt|� d t|� d | d �}|�|��� }|r.t|j�W d � S W d � d S 1 s:w Y d S )N�4SELECT s.* from kc_sessions s where s.session_key = � and s.conf_id = � and conf_key = 'z';�r �connectr �str�execute� one_or_none�dict�_mapping�r � session_key�
current_dt�conf_id�conf_key�conn�stmt�resultr r r �getDataBySessionKey s
(�$�z KitModel.getDataBySessionKeyc C � t �� }|�� }|�d|||||g� |jr?dd� |jD �}g } |�� D ] }
| �tt||
��� q$|� � |�
� | r=| S d S |� � |�
� d S )NZusp_save_kc_scanned_datac S � g | ]}|d �qS �r r ��.0�columnr r r �
<listcomp>" � z1KitModel.insertkcScannedData1.<locals>.<listcomp>� r �raw_connection�cursor�callproc� description�fetchall�appendr( �zip�close�commit) r �
session_id� delegate_no�del_table_name�kc_del_session_table_name�kc_session_entry_table_name�
connectionr= �columns�results�rowr r r �insertkcScannedData1 � z KitModel.insertkcScannedData1c C r3 )NZusp_get_kc_reportsc S r4 r5 r r6 r r r r9 7 r: z)KitModel.getKITReport.<locals>.<listcomp>r; � r rE r- r. rG rI rJ r= rK rL rM r r r � getKITReport2 rO zKitModel.getKITReportc C sn t �� �)}tdt|� d �}|�|��� }|r$t|j�W d � S W d � d S 1 s0w Y d S )Nz/select * from kc_sessions where session_id = z ;r# )r rE r, r/ r0 r1 r r r �checkSessionIsOpenG s
�$�zKitModel.checkSessionIsOpenc
C s� t �� }|�� }|�d||||g� |jr>dd� |jD �}g }|�� D ] } |�tt|| ��� q#|� � |�
� |r<|S d S |� � |�
� d S )NZusp_get_kc_download_reportc S r4 r5 r r6 r r r r9 W r: z1KitModel.getKITReportDownload.<locals>.<listcomp>r; )
r rE r- rI �rolerJ r= rK rL rM r r r �getKITReportDownloadR s z KitModel.getKITReportDownloadc C s� t �� �5}zt|�}|�|�}|�� W W d � dS ty8 } zt|�W Y d }~W d � S d }~ww 1 s<w Y d S �N�success)r r$ r r& rD r r% )r � insert_stmtr/ r0 rL r r r r �insert_kc_token_datag s
����z KitModel.insert_kc_token_datac C s� t �� �7}tdt|� d t|� d | d | d �}|�|��� }|r2t|j�W d � S W d � d S 1 s>w Y d S )Nr r! r" z' and 'z'' between s.start_time and s.end_time;r# r* r r r �getDataBySessionKeyFrontEndq s
0�$�z$KitModel.getDataBySessionKeyFrontEndc C r3 )NZ usp_get_kc_reports_with_namec S r4 r5 r r6 r r r r9 � r: z1KitModel.getKITReportWithName.<locals>.<listcomp>r; rP r r r �getKITReportWithName{ rO z KitModel.getKITReportWithNamec C s� t �� �h}|dkr=tdt|� d d | d d �}|�|��� }|r+dd� |D �nd }|r<|r3|nd W d � S n'td t|� d
| d �} |�| ��� }
|
rYt|
j�nd }
|
W d � S W d � d S 1 sow Y d S ) Nr z�select id,url,message,host_url,scan_updated_at,date_format(created_at , '%d-%m-%Y,%H:%i:%s') as created_on,conf_id from numerotech_primary_db.logs where url like '%/post_kc_scan_data?conf_id=z%' zH and host_url = 'conferencescan.numerotech.com' and date(created_at) = 'z' z: and scan_updated_at is null order by id desc limit 50 ;c S s g | ]}t |j��qS r )r( r) )r7 �rr r r r9 � s z/KitModel.getKcScanErrorLogs.<locals>.<listcomp>z�select conf_id,del_table_name,fc_del_session_table_name,kc_del_session_table_name,kc_session_entry_table_name,fc_session_entry_table_name from conference where conf_id =r" z ' limit 1;) �enginer$ r r% r& �allr' r( r) ) r r- r. � curr_date�is_column_onlyr/ r0 r1 rL �stmt_2Zresult_1r r r �getKcScanErrorLogs� s0
����
� �"�zKitModel.getKcScanErrorLogsc C sb t �� �#}| j�� �| jjj�|���|�}|� |�}|�
� W d � dS 1 s*w Y dS rU ) r\ r$ �logsr �where�c�id�in_�valuesr& rD )r �ids�datar/ r0 � restult_1r r r �saveKCScanDataFromLogs� s
"
��z KitModel.saveKCScanDataFromLogsN)�__name__�
__module__� __qualname__r r2 rN rQ rR rT rX rY rZ ra rk r r r r r s
r )�
sqlalchemyr r r r r r r r Zsqlalchemy.sqlr
r �corer �jsonr r
r �configr\ r r r r r r �<module> s (
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists