File: //usr/local/CyberCP/aiScanner/__pycache__/scheduled_views.cpython-310.pyc
o
��h&0 � @ s� d dl mZmZ d dlmZ d dlmZ d dlmZ d dl m
Z
d dl Z eddg�d d
� �Z edd g�d d
� �Z
eedg�dd� ��Zedg�dd� �ZdS )� )�render�redirect)� JsonResponse)� csrf_exempt)�require_http_methods)�
loadLoginPageN�GET�POSTc
C s� �z�| j d }ddlm} ddlm} ddlm} |jj|d�}|� |�}| j
dkr�|d dkr6|j� � }n |� |�}|jj
|d
�}g } |D ]^}
| �i d |
j�d |
j�d
|
j�d|
j�d|
j�d|
j�d��d|
j�d|
j�d|
j�d|
jr{|
j�� nd�d|
jr�|
j�� nd�d|
j �d|
j �d|
j �d|
j �d |
j �d |
j!�� �� qFt"d | d ��W S | j
d k�r�t#�$| j%�} g d!�} | D ]}
|
| vs�| |
s�t"d"d#|
� �d$�� W S q�t&| d
t'�r�t(| d
�dkr�t"d"d%d$��W S |d dk�r dd&l)m*} t+|jj
|d'�j,d(d d)��}t+| d
�}|�-|��s t"d"d*d$��W S dd+l.m.} z |�/| d d��0� }W n t1�yA t"d"d,d$�� Y W S w ||| d | d | d || �dd �| �dd �| �dd"�| �d d �d-� }|�2| d
� | �d��ry|�3| d � | d d.k�r�d| v �r�t4| d �|_n| d d/v �r�d| v �r�t4| d �|_|�5� t"d |jd0��W S W dS t6�y� t"d"d1d$�� Y S t7�y� } zt"d"t8|�d$��W Y d}~S d}~ww )2zManage scheduled scans�userIDr ��
Administrator� ��
ScheduledScan��
ACLManager��pkr �admin)� admin__in�id�name�domains� frequency� scan_type� time_of_day�%H:%M� day_of_week� day_of_month�status�last_runN�next_run�email_notifications�notification_emails�notify_on_threats�notify_on_completion�notify_on_failure�
created_atT)�success�scheduled_scansr )r r r r r FzMissing required field: �r( �errorz$At least one domain must be selected)�Websites)r �domain)�flatz6You do not have access to some of the selected domains)�datetimezInvalid time format) r r r r r r" r$ r% r& �weekly)�monthly� quarterly)r( r �Not authenticated)9�session�loginSystem.modelsr �modelsr � plogical.aclr �objects�get� loadedACL�method�all�loadUserObjects�filter�appendr r � domain_listr r r �strftimer r r r � isoformatr! r" �notification_email_listr$ r% r& r' r �json�loads�body�
isinstance�list�len�websiteFunctions.modelsr, �set� values_list�issubsetr/ �strptime�time�
ValueError� set_domains�set_notification_emails�int�save�KeyError� Exception�str)�requestr
r r r r �
currentACLr) � user_admins� scan_data�scan�data�required_fields�fieldr, � user_domains�requested_domainsr/ �time_obj�scheduled_scan�e� re �//usr/local/CyberCP/aiScanner/scheduled_views.py�scheduledScans s�
����� ��� �
� � �
����
� � �
�
�< ��rg �DELETEc
C s� z�| j d }ddlm} ddlm} ddlm} |jj|d�}|� |�}z"|jj|d�}|d dkrE|�
|�} |j | vrEt d
d d ��W W S W n |j
yX t d
d
d �� Y W S w | jdkr�i d|j�d|j�d|j�d|j�d|j�d|j�d��d|j�d|j�d|j�d|jr�|j�� nd�d|jr�|j�� nd�d |j �d |j �d |j �d |j �d |j �d!|j!�� �}
t d"|
d#��W S | jd$kr�|�"� t d%d"i�W S W dS t#y� t d
d&d �� Y S t$y� } zt d
t%| �d ��W Y d} ~ S d} ~ ww )'z'Get or delete a specific scheduled scanr
r r r
r r r �r r F�$Access denied to this scheduled scanr* �Scheduled scan not foundr r r r r r r r r r r r Nr! r" r# r$ r% r& r' T)r( rc rh r( r3 )&r4 r5 r r6 r r7 r r8 r9 r: r= r r � DoesNotExistr; r r r@ r r r rA r r r r rB r! r" rC r$ r% r& r' �deleterU rV rW ) rX �scan_idr
r r r r rY rc rZ r[ rd re re rf �scheduledScanDetaily s�
��
����� ��� �
� � �
����
�
� ��ro c
C s8 zq| j d }ddlm} ddlm} ddlm} |jj|d�}|� |�}z"|jj|d�}|d dkrE|�
|�} |j | vrEt d
d d ��W W S W n |j
yX t d
d
d �� Y W S w |jdkrbd|_nd|_|�� t d|jd��W S ty� t d
dd �� Y S ty� }
zt d
t|
�d ��W Y d}
~
S d}
~
ww )z,Toggle scheduled scan status (active/paused)r
r r r
r r r ri r Frj r* rk �active�pausedT)r( r r3 N)r4 r5 r r6 r r7 r r8 r9 r: r= r r rl r rT rU rV rW ) rX rn r
r r r r rY rc rZ rd re re rf �toggleScheduledScan� s8
��
��rr c C s� z�| j d }ddlm} ddlm}m} ddlm} |jj |d�}|�
|�}z"|jj |d�} |d dkrG|� |�}
| j |
vrGt
d
d d ��W W S W n |jyZ t
d
d
d �� Y W S w |jj| d��d�dd� } g } | D ]2}
| �|
j|
j�� |
j|
j|
j|
j|
jt|
j�|
j |
j |
j r�|
j �� nd|
j r�|
j �� ndd� � qmt
d| d��W S t y� t
d
dd �� Y S t!y� } zt
d
t"|�d ��W Y d}~S d}~ww )z*Get execution history for a scheduled scanr
r r r
)r �ScheduledScanExecutionr r ri r Frj r* rk )rc z-execution_timeN� ) r �execution_timer �domains_scanned� total_scans�successful_scans� failed_scans�
total_cost�scan_ids�
error_message�
started_at� completed_atT)r( �
executionsr3 )#r4 r5 r r6 r rs r7 r r8 r9 r: r= r r rl r> �order_byr? r ru rB r �scanned_domainsrw rx ry �floatrz � scan_id_listr| r} r~ rU rV rW )rX rn r
r r rs r r rY rc rZ r �execution_data� executionrd re re rf �scheduledScanExecutions� sR
��
� ��r� )�django.shortcutsr r � django.httpr � django.views.decorators.csrfr � django.views.decorators.httpr �loginSystem.viewsr rD rg ro rr r� re re re rf �<module> s
o
: '