.piohQ 0 d dl Z d dlZd dlZd dlmZmZ d dlmZmZ d dl m
c m c m Z
d dlmZ d dlmZmZ d dlmZ d dlmZ d dlmZ d d lmZ ej e Z G d
d e Z d Z ddee de!de!fdZ"de!fdZ#dee de!fdZ$dS ) N)ListOptional) BooleanField CharField) MessageType)Modelinstance)run_in_executor)execute_iterable_expression
update_config)
MalwareHitc e Zd ZdZ G d d Z ed Z edd Ze de
e d e fd
Z
e d ee d e fd
ZdS ) MyImunifyz Secure-site related settingsc e Zd Zej ZdZdS )MyImunify.Meta myimunifyN)__name__
__module__ __qualname__r dbdatabasedb_table T/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/myimunify/model.pyMetar s ; r r T)uniqueF)nulldefaultuserreturnc \ ||dk rdS | |ddi \ }}|j S )z%Get SecureSite protection by usernameNrootT
protectionF)r! defaults)
get_or_creater% )clsr! perm_s r get_protectionzMyImunify.get_protection! s? <46>>4## u8M#NNa r usersstatusc | fd|D | j gg | j i d S )Nc g | ]}|d S ))r! r% r ).0r! r- s r
z5MyImunify.update_users_protection... s! D D Ddd&
1
1 D D Dr )conflict_targetpreserveupdate) insert_many on_conflictr! r% execute)r( r, r- s `r update_users_protectionz!MyImunify.update_users_protection+ sd D D D De D D D
+ XJNF+
')))))r N)r r r __doc__r r r! r r% classmethodr strboolr+ r r8 r r r r r s && 9D ! ! !D 5%888J (3- D [ DI t [ r r c K t j |d }|r0| t j | d {V d S d S )NT)r! cleanup)hits)r malicious_selectprocess_messager MalwareCleanupTask)sinkr! r? s r malware_cleanuprD 6 sj &D$ ? ? ?D N"";#At#L#L#LMMMMMMMMMMMN Nr Fr, r- force_config_updatec K t d fd d {V d sd|rt ddii g}n fdD }r| fdD z
}t j | d {V d S )Nc : t S N)r r8 )r- r, s r z)update_users_protection..A s 11%@@ r LOGPROACTIVE_DEFENCEmodec : g | ]}t d dii| S )rK rL r )r0 r! proactive_moderC s r r1 z+update_users_protection..O sH
$v~&>?
r c 0 g | ]}t | S r )rD )r0 r! rC s r r1 z+update_users_protection..Y s# @@@$/$--@@@r )r
r
asynciogather)rC r, r- rE tasksrN s ``` @r r8 r8 < s @@@@@ N
$v~&>?
A
@@@@%@@@@
.%
r c K t j d{V }t | || d{V dS )z#Set protection status for all usersN)hp HostingPanel get_usersr8 )rC r- panel_userss r #set_protection_status_for_all_usersrX ^ s[ ))3355555555K
!$ V
<
<<<<<<<<<.expressionp s9 ##%%++""?33
r zAdd permissions to users %sFN)
setr selectr! itertoolschaintuplesloggerinfor listr8 r< )rC r, rW myimunify_usersr^ r_ users_to_adds r
sync_usersrj d s e**K &&y~66O)/?+A+A+C+CDEEO% 3O G / AAA
$J_0E0EFFF 0L : 1<@@@%dE5999999999 6o!6!66r )F)%rP rb loggingtypingr r peeweer r +defence360agent.subsys.panels.hosting_panelsubsyspanels
hosting_panelrT "defence360agent.contracts.messagesr defence360agent.modelr r $defence360agent.model.simplificationr
defence360agent.utilsr defence360agent.utils.configr
imav.malwarelib.modelr getLoggerr re r rD r; r< r8 rX rj r r r ry s ! ! ! ! ! ! ! ! * * * * * * * * 8 8 8 8 8 8 8 8 8 8 8 8 : : : : : : 1 1 1 1 1 1 1 1 @ @ @ @ @ @ = = = = = = 6 6 6 6 6 6 , , , , , , 8 $ $ FN N N GL ! !c !$( !?C ! ! ! !D=D = = = = 7$s) 7 7 7 7 7 7 7r