File: //usr/local/CyberCP/aiScanner/management/commands/__pycache__/run_scheduled_scans.cpython-310.pyc
o
��hYQ � @ sL d dl mZ d dlmZ d dlmZmZ d dlZd dlZG dd� de�ZdS )� )� BaseCommand)�timezone)�datetime� timedeltaNc @ s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dS )�Commandz Run scheduled AI security scansc C sD |j dddd� |j dtdd� |j ddd d� |j d
dd d� d S ) Nz--daemon�
store_truez8Run as daemon, checking for scheduled scans every minute)�action�helpz --scan-id�#Run a specific scheduled scan by ID)�typer z --verbosez3Show detailed information about all scheduled scansz--forcezCForce run all active scheduled scans immediately, ignoring schedule)� add_argument�int)�self�parser� r �G/usr/local/CyberCP/aiScanner/management/commands/run_scheduled_scans.py�
add_arguments s( ���
�zCommand.add_argumentsc O s� |� dd�| _|� dd�| _|d r | j�d� | �� d S |d r7| j�d|d � d�� | �|d � d S |d rG| j�d � | �� d S | j�d
� | �� d S ) N�verboseF�force�daemonz!Starting scheduled scan daemon...�scan_idzRunning scheduled scan ID z...z+Force running all active scheduled scans...z&Checking for scheduled scans to run...) �getr r �stdout�write�
run_daemon�run_scheduled_scan_by_id�force_run_all_scans�check_and_run_scans)r �args�optionsr r r �handle! s zCommand.handlec
C s� z| j �dt�� �d�� d�� | �� t�d� W nA ty+ | j �d� Y d S t y\ } z&| j
�dt |�� �� dd l m
} |�d
t |�� �� t�d� W Y d }~nd }~ww q) z.Run as daemon, checking for scans every minuteTz
[�%Y-%m-%d %H:%M:%S UTCz!] Checking for scheduled scans...�< z
Daemon stopped by userzError in daemon: r ��CyberCPLogFileWriterz [Scheduled Scan Daemon] Error: N)r r r �now�strftimer �time�sleep�KeyboardInterrupt� Exception�stderr�str� plogical.CyberCPLogFileWriterr$ � writeToFile)r �e�loggingr r r r 2 s ���zCommand.run_daemonc C s� ddl m} ddlm} |jjdd�}|�� dkr&| j�d� |� d� dS | j�d |�� � d
�� |� d |�� � d
�� |D ]$}| j�d |j
� d
|j � d�� |� d|j
� d
|j � d�� | � |� q?dS )z0Force run all active scheduled scans immediatelyr ��
ScheduledScanr# �active)�statusz No active scheduled scans foundz6[Scheduled Scan Force] No active scheduled scans foundN�Found z$ active scheduled scans to force runz [Scheduled Scan Force] Found z Force running scheduled scan: � (ID: �)z5[Scheduled Scan Force] Force running scheduled scan: )
�aiScanner.modelsr2 r- r$ �objects�filter�countr r r. �name�id�execute_scheduled_scan)r r2 r0 � active_scans�scanr r r r B s
�zCommand.force_run_all_scansc C s� ddl m} ddlm} t�� }|j�� }| j� d|�
� � �� |� d|�
� � �� |D �]`}| j r�| j� d|j
� d|j� d�� | j� d |jj� �� | j� d
|j� �� | j� d |j� �� | j� d |j� �� | j� d
d�|j�� �� | j� d|j�d�� �� |jr�| j� d|j�d�� �� n| j� d� |jdkr�d|j
� d|j� d|j� d�}| j� d|� �� |� d|� �� q,|jdu �r%d|j
� d|j� d�}| j� d|� �� |� d|� �� | j �r$| j� d � z|�� |_|�� | j� d |j�d�� �� W n t �y# } z| j� d t |�� �� W Y d}~nd}~ww q,|j|k�r�|j| }t |� � d �} t |� � d d �}
t |� � d d! �} d"} | dk�r_| � d#|
� d$| � d%�} n|
� d$| � d%�} d|j
� d|j� d&| � d'|j�d�� �}| j� d(|� �� |� d|� �� q,q,|jj d|d)�}
|
�
� dk�r�| j� d*� |� d+� n| j� d,|
�
� � d-�� |� d.|
�
� � d-�� |
D ]%}| j� d/|j
� d|j� d�� |� d0|j
� d|j� d�� | �!|� �q�dS )1z*Check for scheduled scans that need to runr r1 r# zTotal scheduled scans: z.[Scheduled Scan Check] Total scheduled scans: z
--- Scan Details: r6 z) ---z Owner: z
Frequency: z
Scan Type: z
Status: z Domains: �, z Created: r! z Last Run: z Last Run: Neverr3 zScan "z" (ID: z) is not active (status: r7 u ❌ z[Scheduled Scan Check] Nz) has no next_run scheduledu/ 🔧 Attempting to calculate next run time...u ✅ Next run set to: u$ ❌ Failed to calculate next run: i�Q i r" � zd zh �mz) scheduled to run in z at u ⏰ )r4 �
next_run__ltez.No scheduled scans are due to run at this timezE[Scheduled Scan Check] No scheduled scans are due to run at this timer5 z scans due to runz [Scheduled Scan Check] Found �Running scheduled scan: z/[Scheduled Scan Check] Running scheduled scan: )"r8 r2 r- r$ r r% r9 �allr r r; r. r r< r= �admin�userName� frequency� scan_typer4 �join� domain_list�
created_atr& �last_run�next_run�calculate_next_run�saver* r, r
�
total_secondsr: r> )r r2 r0 r% � all_scansr@ �reasonr/ �time_until_run�days�hours�minutes�time_str� due_scansr r r r W s�
"��
(�� �zCommand.check_and_run_scansc C sj ddl m} z|jj|d�}| j�d|j� �� | �|� W dS |jy4 | j �d|� d�� Y dS w )r
r r1 )r= rE zScheduled scan with ID z
not foundN)
r8 r2 r9 r r r r< r> � DoesNotExistr+ )r r r2 r@ r r r r � s �z Command.run_scheduled_scan_by_idc C s� ddl m}m} ddlm} ddlm} ddlm} ddl m
} |j j |dt
�� d�}�zht
�� |_|�� |_|�� g } |j}
|jD ]'} z|j j| |
d �} | �| � W qC |jyj |�d
| � d |
j� �� Y qCw | s�d |_d
|_t
�� |_ |�� | j � d|j � �� W dS |� | � t!| �|_"|�� |� }
g }d}d}d}| D ]�} z�| j#� d| � �� G dd� d�}||
j$| |j%�}|
�&||
j$�}t'|d�r�ddl(}|�)|j*�+d��}n|}|�d��r|�d�}|�r
|�|� |d7 }d|v r�|t,|d �7 }|�d|� d| � �� n"|d7 }|�d | � �� n|d7 }|�d d �}|�d | � d |� �� W n& t-�yS } z|d7 }t.|�}|�d!| � d |� �� W Y d}~nd}~ww t/�0d"� q�||_1||_2||_3|�4|� |dk�rod#nd#|_t
�� |_ |�� |j5�r�| �6||� | j#� d$|� d%|� d&�� W dS t-�y� } zSd |_t.|�|_t
�� |_ |�� |�d'|j � d t.|�� �� | j � d(|j � d t.|�� �� |j5�r�|j7�r�| �8|t.|�� W Y d}~dS W Y d}~dS W Y d}~dS d}~ww ))zExecute a scheduled scanr )�ScheduledScanExecution� ScanHistory)�AIScannerManager)�
Administrator)�Websitesr# �running)�scheduled_scanr4 �
started_at)�domainrG z[Scheduled Scan] Domain z no longer accessible for user �failedz(No accessible domains found for scanningz)No accessible domains for scheduled scan Ng zStarting scan for domain: c @ s e Zd Zdd� Zdd� ZdS )z3Command.execute_scheduled_scan.<locals>.FakeRequestc S s@ d|i| _ d| _||d�| _dd l}|�||d���d�| _d S )N�userID�POST)rd rJ r �utf-8)�session�methodrg �json�dumps�encode�body)r �admin_idrd rJ rk r r r �__init__� s
���z<Command.execute_scheduled_scan.<locals>.FakeRequest.__init__c S s. zddl m} |�� }|� d�W S Y dS )Nr )�
ACLManagerz:8090zlocalhost:8090)� plogical.aclrq �fetchIP)r rq � server_ipr r r �get_host� s z<Command.execute_scheduled_scan.<locals>.FakeRequest.get_hostN)�__name__�
__module__� __qualname__rp ru r r r r � FakeRequest� s ry �contentrh �successr � �
cost_estimatez+[Scheduled Scan] Successfully started scan z for z+[Scheduled Scan] Failed to get scan ID for �errorz
Unknown errorz*[Scheduled Scan] Failed to start scan for z: z-[Scheduled Scan] Exception starting scan for � � completedzScheduled scan completed: z
successful, z failedz2[Scheduled Scan] Failed to execute scheduled scan z!Failed to execute scheduled scan )9r8 r\ r] �aiScanner.aiScannerManagerr^ �loginSystem.modelsr_ �websiteFunctions.modelsr` r- r$ r9 �creater r% rN rP rO rQ rG rL r �appendr[ r. rH r4 �
error_message� completed_atr+ r r< �set_scanned_domains�len� total_scansr �pkrJ � startScan�hasattrrk �loadsrz �decode�floatr* r, r' r( �successful_scans� failed_scans�
total_cost� set_scan_ids�email_notifications�send_notifications�notify_on_failure�send_failure_notification)r rb r\ r] r^ r_ r` r0 � execution�domains_to_scanrG rd �website�sm�scan_idsr� r� r� ry � fake_request�resultrk �
response_datar � error_msgr/ r r r r> � s� �
�
�"��
�
���z Command.execute_scheduled_scanc
C s� z1d}|j dkr|jrd}n|j dkr$|jrd}n |jr$|jdkr$|j}|r/| �||� W dS W dS tyT } zddlm} |� dt
|�� �� W Y d}~dS d}~ww ) z+Send email notifications for completed scanFre Tr� r r# z.[Scheduled Scan] Failed to send notification: N) r4 r� �notify_on_completion�notify_on_threatsr� �send_execution_notificationr* r- r$ r. r, )r rb r� �
should_notifyr/ r0 r r r r� N s"
� "��zCommand.send_notificationsc C sx z�|j }|s|jjr|jjgng }|sW dS d|j� d�}|j�� }|jdkr6|jdkr/d}nd|j� d�}d |j� d
|� d |j�d �� d
|j � d|j
� d|j� d|j d�dd� |j
�� d|jrfd|j� �nd� dd� |j�� d�}ddlm} d}|�|||� ddlm} | �d |j� d t|�� d �� W dS ty� }
zddlm} | �d t|
�� �� W Y d}
~
dS d}
~
ww ) z*Send notification email for scan executionN� AI Scanner: Scheduled Scan "z " Completedr� r zCompleted SuccessfullyzCompleted with z failuresz/
Scheduled AI Security Scan Report
Scan Name: z
Status: z
Execution Time: r! z
Results:
- Total Domains: z
- Successful Scans: z
- Failed Scans: z
- Total Cost: $z.4fz
Domains Scanned: rA z
zError Message: rB z
Scan IDs: zA
View detailed results in your CyberPanel AI Scanner dashboard.
��
mailUtilities�noreply@cyberpanel.localr# z'[Scheduled Scan] Notification sent for z to z recipientsz4[Scheduled Scan] Failed to send notification email: )�notification_email_listrG �emailr< r4 �titler� �execution_timer& r� r� r� rK �scanned_domainsr� � scan_id_list�plogical.mailUtilitiesr� � SendEmailr- r$ r. r� r* r, ) r rb r� �notification_emails�subject� status_text�messager� �senderr0 r/ r r r r� f sV
��
�� �
� �
��
� $ "��z#Command.send_execution_notificationc
C s� zO|j }|s|jjr|jjgng }|sW dS d|j� d�}d|j� dt�� �d�� d|� d�}d d
lm} d }|� |||� d d l
m } |� d
|j� �� W dS t
yr } zd d l
m } |� dt| �� �� W Y d} ~ dS d} ~ ww )z(Send notification email for scan failureNr� z" Failedz0
Scheduled AI Security Scan Failure
Scan Name: z
Status: Failed
Time: r! z
Error: zG
Please check your CyberPanel AI Scanner configuration and try again.
r r� r� r# z/[Scheduled Scan] Failure notification sent for z<[Scheduled Scan] Failed to send failure notification email: )r� rG r� r< r r% r& r� r� r� r- r$ r. r* r, )
r rb r� r� r� r� r� r� r0 r/ r r r r� � s0 � ��
"��z!Command.send_failure_notificationN)rv rw rx r r r r r r r r> r� r� r� r r r r r s R 9r ) �django.core.management.baser � django.utilsr r r rk r' r r r r r �<module> s