Team Ai
Apppublic

sameersyed/Defence_FrameWork

sourceHugging Faceupdated 8mo agoView on Hugging Face
0likes
rootbackend__pycache__database.cpython-313.pyc
database.cpython-313.pyc58 linesDownload Raw Back to __pycache__
1�

2/~i���<�SSKrSSKJrJr SSKJr "SS5rg)�N)�generate_password_hash�check_password_hash)�datetimec�V�\rSrSrS
SjrSrSrSrSrSr	Sr3SS	jrSS4jrSr
g)�Database�c�0�XlUR5 g�N)�db_name�init_db)�selfrs  �#D:\Security_app\backend\database.py�__init__�Database.__init__s��������c�B�[R"UR5$r5)�sqlite3�connectr)r
s r�get_connection�Database.get_connection6s�����t�|�|�,�,rc���UR5nUR5nURS5 URS5 URS5 UR5 UR	5 g)NaC7            CREATE TABLE IF NOT EXISTS users (8                id INTEGER PRIMARY KEY AUTOINCREMENT,9                username TEXT UNIQUE NOT NULL,10                email TEXT UNIQUE NOT NULL,11                password_hash TEXT NOT NULL,12                created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP13            )14        a]15            CREATE TABLE IF NOT EXISTS chat_history (16                id INTEGER PRIMARY KEY AUTOINCREMENT,17                user_id INTEGER,18                message TEXT,19                response TEXT,20                created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,21                FOREIGN KEY (user_id) REFERENCES users (id)22            )23        a�24            CREATE TABLE IF NOT EXISTS threat_history (25                id INTEGER PRIMARY KEY AUTOINCREMENT,26                user_id INTEGER NOT NULL,27                threat_type TEXT NOT NULL,28                risk_level TEXT NOT NULL,29                model_type TEXT,30                confidence INTEGER,31                timestamp TIMESTAMP DEFAULT CURRENT_TIMESTAMP,32                FOREIGN KEY (user_id) REFERENCES users(id)33            )34        �r�cursor�execute�commit�close)r
�connrs   rr�Database.init_db
sh���"�"�$�������	����	
�	���	�		
�	����	
�	
���
��35�36�rc�D�UR5nUR5n[U5nURSXU45 UR	5 UR5 g![Ra UR5 gf=f!UR5 f=f)NzCINSERT INTO users (username, email, password_hash) VALUES (?, ?, ?)TF)rrrrrrr�IntegrityError)r
�username�email�passwordrr�
password_hashs       r�create_user�Database.create_user9s����"�"�$�������	�2�8�<�M��N�N�U��-�0�
�
�K�K�M��
�J�J�L���%�%�	���J�J�L�	��
�J�J�L�s#�/A#�#B37�7B
�	B38�39B
�
Bc���UR5nUR5nURSX45 UR5nUR	5 U(a[USU5(aUS$g)NzCSELECT id, password_hash FROM users WHERE username = ? OR email = ?�r)rrr�fetchonerr)r
r!r#rr�users      r�verify_user�Database.verify_userJsg���"�"�$����������Q�
� �	40���� ���41�42���'��Q���:�:���7�N�rc��UR5nUR5nURSU45 UR5nUR	5 U$)Nz.SELECT username, email FROM users WHERE id = ?)rrrr)r)r
�user_idrrr*s     r�get_user_by_id�Database.get_user_by_idYsF���"�"�$����������G�'��T���� ���43�44���rc��UR5nUR5nURSXU45 UR5 UR	5 g)NzFINSERT INTO chat_history (user_id, message, response) VALUES (?, ?, ?)r)r
r.�message�responserrs      r�	save_chat�Database.save_chatcsF���"�"�$����������T�
�x�(�	45�	
���
��46�47�rc��UR5nUR5nURSXX4U45 UR5 UR	5 g)NzlINSERT INTO threat_history (user_id, threat_type, risk_level, model_type, confidence) VALUES (?, ?, ?, ?, ?)r)r
r.�threat_type�48risk_level�49model_type�50confidencerrs        r�save_threat�Database.save_threatnsH���"�"�$����������z�
�:�:�F�	51�	
���
��52�53�rc��UR5nUR5nURSX45 UR5nUR	5 U$)Nz�SELECT threat_type, risk_level, model_type, confidence, timestamp FROM threat_history WHERE user_id = ? ORDER BY timestamp DESC LIMIT ?)rrr�fetchallr)r
r.�limitrr�threatss      r�get_threat_history�Database.get_threat_historyysO���"�"�$����������
V�
��	54��/�/�#���55�56���r)rN)zsecurity_app.db)�general�K)�57)�__name__�58__module__�__qualname__�__firstlineno__rrrr%r+r/r4r;rA�__static_attributes__�rrrrs0���-�*�X�"
��	�	�rr)r�werkzeug.securityrrrrrKrr�<module>rMs���I���r