File: //usr/local/CyberCP/aiScanner/__pycache__/aiScannerManager.cpython-310.pyc
o
��hR� � @ s� d dl Z d dlZd dlZd dlZd dlmZmZ d dlmZmZ d dl m
Z
d dlmZ d dl
mZ d dlmZ d dlmZ d d lmZ d
dlmZmZmZ d dlmZ d d
lmZ G dd� d�ZdS )� N)�datetime� timedelta)�render�redirect)�JsonResponse)�timezone)�settings)�reverse)�messages)�
Administrator� )�AIScannerSettings�ScanHistory�FileAccessToken)�
ACLManager)�CyberCPLogFileWriterc @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d� Zdd
� Z dd� Z
dd� Zdd� Zdd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(S ))�AIScannerManagerz,https://platform.cyberpersons.com/ai-scannerc C s
t | _d S )N)�logging�logger��self� r �0/usr/local/CyberCP/aiScanner/aiScannerManager.py�__init__ s
zAIScannerManager.__init__c
C s� �z
t jj|d�}t�|�}z|�dd�dkrt�� W W S W n ttfy/ | j� d� Y nw t
jj|ddd�d �\}}| �� }|d
dkrTt
j�� �d�dd
� }nt�|�} t
jj| d��d�dd
� }|j}
| j� d|
� �� |jr�| j� d� | �|j�}| j� d|� �� |dur�||_|�� |}
| j� d|
� �� n | j� d|
� �� t�� }| �|�}
zt�||�}| j� dt|�� d|j� �� W n ty� } z| j� dt|�� �� g }W Y d}~nd}~ww | j� d|j� �� |||||
||j|
|d� }| j� d� t|d|�W S t�yM } z2ddl}| j� dt|�� �� | j� d|� � � �� t|ddt|�� �dg g ddd��W Y d}~S d}~ww )zMain AI Scanner page��pk�aiScannerAccessr r zO[AIScannerManager.scannerHome] aiScannerAccess field not found, allowing access� F)�balance�is_payment_configured��admin�defaultsr! z-started_atN�
)� admin__inz/[AIScannerManager.scannerHome] Stored balance: z;[AIScannerManager.scannerHome] Fetching balance from API...z5[AIScannerManager.scannerHome] API returned balance: z3[AIScannerManager.scannerHome] Updated balance to: zP[AIScannerManager.scannerHome] API balance call failed, keeping stored balance: z%[AIScannerManager.scannerHome] Found z websites for z8[AIScannerManager.scannerHome] Error fetching websites: z4[AIScannerManager.scannerHome] Building context for ) r! �scanner_settings�pricing_data�recent_scans�current_balance�websitesr �vps_info� server_ipzM[AIScannerManager.scannerHome] Context built successfully, rendering templatezaiScanner/scanner.htmlz&[AIScannerManager.scannerHome] Error: z*[AIScannerManager.scannerHome] Traceback: z Failed to load AI Scanner page: )�errorr r) r'