File: //usr/local/CyberCP/aiScanner/__pycache__/models.cpython-310.pyc
o
��h 0 � @ s� d dl mZ d dlmZ d dlZddlmZ G dd� dej�ZG dd � d ej�Z G d
d � d ej�Z
G d d
� d
ej�Z G dd� dej�Z dS )� )�models)�
AdministratorN� )�ScanStatusUpdatec @ s| e Zd ZdZejeejdd�Zej dddd�Z
ej ddd d
�Z ej
d d �Zejdd
�Zejdd�ZG dd� d�Zdd� ZdS )�AIScannerSettingsz>Store AI scanner configuration and API keys for administrators�ai_scanner_settings�� on_delete� related_name� T)�
max_length�blank�null�
� � ��
max_digits�decimal_places�defaultF�r �� auto_now_add��auto_nowc @ � e Zd ZdZdS )zAIScannerSettings.Metar N��__name__�
__module__� __qualname__�db_table� r! r! �&/usr/local/CyberCP/aiScanner/models.py�Meta � r# c C s d| j j� �S )NzAI Scanner Settings for )�admin�userName��selfr! r! r" �__str__ s zAIScannerSettings.__str__N)r r r �__doc__r �
OneToOneFieldr �CASCADEr% � CharField�api_key� DecimalField�balance� BooleanField�is_payment_configured�
DateTimeField�
created_at�
updated_atr# r) r! r! r! r" r s r c @ s e Zd ZdZg d�Zg d�Zejeej dd�Z
ej ddd�Z ej d d
�Z
ej d ed d
�Zej d edd
�Zejddddd�Zejdd�Zejdd�Zejddd�Zejddd�Zejddd�Zejdd�Zejddd�ZG dd� d�Z dd� Z e dd � �Z e d d � �Z d d � Z!d!d"� Z"d#S )$� ScanHistoryz Store scan history and results�)�pending�Pending)�running�Running)� completed� Completed)�failed�Failed)� cancelled� Cancelled�)�fullz Full Scan)�quickz
Quick Scan)�customz Custom Scan� scan_historyr �d T�r �uniquer �r � rC �r �choicesr r8 r � )r r r r
r r �r
r r �r r
c @ � e Zd ZdZdgZdS )zScanHistory.Meta�ai_scanner_historyz -started_atN�r r r r �orderingr! r! r! r" r# 7 �
r# c C s d| j � d| j� d| j� d�S )NzScan z - � (�))�scan_id�domain�statusr' r! r! r" r) ; s zScanHistory.__str__c C �2 | j rzt�| j �W S tjy g Y S w g S )zParse findings JSON)�
findings_json�json�loads�JSONDecodeErrorr' r! r! r" �findings> � �zScanHistory.findingsc C s2 | j rzt�| j �W S tjy i Y S w i S )zParse summary JSON)� summary_jsonr] r^ r_ r' r! r! r" �summaryH ra zScanHistory.summaryc C � t �|�| _dS )zSet findings from list/dictN)r] �dumpsr\ )r( �
findings_listr! r! r" � set_findingsR � zScanHistory.set_findingsc C rd )zSet summary from dictN)r] re rb )r( � summary_dictr! r! r" � set_summaryV rh zScanHistory.set_summaryN)#r r r r* �SCAN_STATUS_CHOICES�SCAN_TYPE_CHOICESr �
ForeignKeyr r, r% r- rX rY � scan_typerZ r/ �cost_usd� IntegerField�
files_scanned� issues_found� TextFieldr\ rb �
error_messager3 �
started_at� completed_atr# r) �propertyr` rc rg rj r! r! r! r" r6 s2
r6 c @ s� e Zd ZdZejddd�Zejeej dd�Z
ejdd�Z ejd d�Z e�
� Zej
dd
�Zejdd �ZG d d
� d
�Zdd� Zdd� ZdS )�FileAccessTokenz-Temporary tokens for file access during scansrG TrH �
access_tokensr r rJ i� r r c @ r )zFileAccessToken.Meta�ai_scanner_file_tokensNr r! r! r! r" r# e r$ r# c C s d| j � d| j� �S )Nz
Access token z for )�tokenrY r' r! r! r" r) h s zFileAccessToken.__str__c C s ddl m} |�� | jkS )Nr ��timezone)� django.utilsr} �now�
expires_at)r( r} r! r! r" �
is_expiredk s zFileAccessToken.is_expiredN)r r r r* r r- r{ rm r6 r, rF rY �wp_pathr3 r� r4 r1 � is_activer# r) r� r! r! r! r" rx [ s rx c sx e Zd ZdZg d�Zg d�Zg d�Zeje ej
dd�Z ej ddd �Z
ejd
d �Zej d ed
d�Zej d edd�Zejdd �Zejdddd�Zejdddd�Zej d edd�Zejddd�Zejddd�Zejdd�Zejdd�Z ej dd�Z ejddd�Z ej dd�Z ej d d�Z!ej dd�Z"G d d � d �Z#d d � Z$e%d!d"� �Z&e%d#d$� �Z'd%d&� Z(d'd(� Z)d)d*� Z*� fd+d,�Z+� Z,S )-�
ScheduledScanz#Store scheduled scan configurations))�daily�Daily)�weekly�Weekly)�monthly�Monthly)� quarterly� QuarterlyrB ))�active�Active)�paused�Paused)�disabled�Disabled�scheduled_scansr �� z Name for this scheduled scan)r � help_textz JSON array of domains to scan)r� rK r� rL rC z!Time of day to run the scan (UTC)Tz1Day of week for weekly scans (0=Monday, 6=Sunday))r r
r� z%Day of month for monthly scans (1-31)r� rP r r r z JSON array of email addresses�r
r� Fc @ rQ )zScheduledScan.Meta�ai_scanner_scheduled_scansz -created_atNrS r! r! r! r" r# � rU r# c C s d| j � d| j� d�S )NzScheduled Scan: rV rW )�name� frequencyr' r! r! r" r) � � zScheduledScan.__str__c C r[ )zParse domains JSON)�domainsr] r^ r_ r' r! r! r" � domain_list� ra zScheduledScan.domain_listc C r[ )z Parse notification emails JSON)�notification_emailsr] r^ r_ r' r! r! r" �notification_email_list� ra z%ScheduledScan.notification_email_listc C rd )zSet domains from listN)r] re r� �r( r� r! r! r" � set_domains� rh zScheduledScan.set_domainsc C rd )z!Set notification emails from listN)r] re r� )r( �
email_listr! r! r" �set_notification_emails� rh z%ScheduledScan.set_notification_emailsc C sH ddl m} ddlm}m} ddl}|�� }| jdkr5|j| jj | jj
ddd�}||kr3||dd�7 }|S | jd kr_| j |� � }|dkrI|d
7 }|||d� }|j| jj | jj
ddd�}|S | jd kr�|j
}|j} t| j|�|| �d �}
|j|
| jj | jj
ddd �}||kr�| d
kr�|d7 }d} n| d7 } t| j|�|| �d �}
|j|| |
d�}|S | jdk�r|j| jj | jj
ddd�}|j} |j
}| d7 } | d
kr�|d7 }| d
8 } t| jp�d|�|| �d �}
|j|| |
d�}||k�r| d7 } | d
k�r|d7 }| d
8 } t| j�pd|�|| �d �}
|j|| |
d�}|S ||dd� }|S )z*Calculate next run time based on frequencyr r| )�datetime� timedeltaNr� )�hour�minute�second� microsecondr )�daysr� � r� )�dayr� r� r� r� � )�year�monthr� r� � )�weeks)r~ r} r� r� �calendarr r� �replace� time_of_dayr� r� � day_of_week�weekdayr� r� �min� day_of_month�
monthrange) r( r} r� r� r� r �next_run�
days_aheadr� r� r� r! r! r" �calculate_next_run� sb
8
�0
� �
�z ScheduledScan.calculate_next_runc s0 | j r| jdkr
| �� | _ t� j|i |�� dS )z#Override save to calculate next_runr� N)r� rZ r� �super�save)r( �args�kwargs�� __class__r! r" r� s
zScheduledScan.save)-r r r r* �FREQUENCY_CHOICESrl �STATUS_CHOICESr rm r r, r% r- r� rs r� r� rn � TimeFieldr� rp r� r� rZ r3 �last_runr� r4 r5 r1 �email_notificationsr� �notify_on_threats�notify_on_completion�notify_on_failurer# r) rw r� r� r� r� r� r� �
__classcell__r! r! r� r" r� p sB
Fr� c @ s� e Zd ZdZg d�Zejeejdd�Z ej
dd�Z ej dedd �Z
ejdd
d �Zejd d
�Zejd d
�Zejd d
�Zejdddd�Zejddd �Zejddd�Zej
ddd�Zej
ddd�ZG dd� d�Zdd� Ze dd� �Z e dd � �Z d d � Z d d � Z d!S )"�ScheduledScanExecutionz.Track individual executions of scheduled scansr7 �
executionsr Tr rK r8 rL z'JSON array of domains that were scannedr� r r r rN r r z JSON array of scan IDs createdrO rP c @ rQ )zScheduledScanExecution.Meta� ai_scanner_scheduled_executionsz-execution_timeNrS r! r! r! r" r# ! rU r# c C s d| j j� d| j� �S )Nz
Execution of z at )�scheduled_scanr� �execution_timer' r! r! r" r) % r� z ScheduledScanExecution.__str__c C r[ )zParse domains scanned JSON)�domains_scannedr] r^ r_ r' r! r! r" �scanned_domains( ra z&ScheduledScanExecution.scanned_domainsc C r[ )zParse scan IDs JSON)�scan_idsr] r^ r_ r' r! r! r" � scan_id_list2 ra z#ScheduledScanExecution.scan_id_listc C rd )z Set scanned domains from listN)r] re r� r� r! r! r" �set_scanned_domains< rh z*ScheduledScanExecution.set_scanned_domainsc C rd )zSet scan IDs from listN)r] re r� )r( r� r! r! r" � set_scan_ids@ rh z#ScheduledScanExecution.set_scan_idsN)!r r r r* r� r rm r� r, r� r3 r� r- rZ rs r� rp � total_scans�successful_scans� failed_scansr/ �
total_costr� rt ru rv r# r) rw r� r� r� r� r! r! r! r" r�
s.
r� )
� django.dbr �loginSystem.modelsr r] �
status_modelsr �Modelr r6 rx r� r� r! r! r! r" �<module> s B