Rツ B d dl Z d dlZd dlZd dlZd dlmZmZmZmZm Z m
Z
d dl m Z m
Z
dZ ej e Z G d de Z G d de Z G d d
e
Zd Zd e
d
ededefdZ G d de
Zd Zd Z G d d Z e Z dS ) N) BlobField CharField DateFieldForeignKeyField IntegerFieldPeeweeException)instanceModeliQ c e Zd ZdZd Zd ZdS )
FilenameFieldz/
Class to store file names in database
c * t j | S N)osfsencodeselfvalues Y/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/model/simplification.pydb_valuezFilenameField.db_value {5!!! c * t j | S r )r fsdecoder s r python_valuezFilenameField.python_value r r N)__name__
__module__ __qualname____doc__r r r r r r s< " " "" " " " "r r c e Zd ZdZd ZdS )
ScanPathField
list_of_filesc > t |t r| j S |S r )
isinstancelistREALTIME_SCAN_PATH_STUBr s r r zScanPathField.db_value& s" eT " " 0/ / r N)r r r r& r r r r r! r! # s) - r r! c e Zd ZdZdS )
ModelErrorzf
Model exception. Please use this one from other modules instead
PeeweeException directly
N)r r r r r r r r( r( , s
Dr r( c K || S )z0
Fake run_in_executor() test (DEF-4541)
r )loopcbargss r run_in_executorr- 5 s 2t9r tablenum_days max_countreturnc 4 t | dd }|s"t d | t j |t z z
}| | j | j |
| j |k }|
| j |
}|S )z
Removes records that is older that *num_days* days and
all others that are out of range *max_count* from *table*.
Returns count of rows deleted.
timestampFz#No 'timestamp' column in table {!r})getattr
ValueErrorformattime POSIX_DAYselectr3 order_bydesclimitwheredeletenot_inexecute)r. r/ r0
has_timestamp
end_save_timeto_keep
deleted_counts r remove_old_and_truncaterE < s E;66M N>EEeLLMMM IKK(Y"66M
U_%% %/&&(( ) ) y u . / / U_ 3 3G < <==EEGG r c e Zd ZdZ G d d Z ed Z edd Ze de
fd Z e d d
Z dS ) EulazKeeps track of updates and acceptions of end user license agreement.
Admins will be asked to accept EULA if the latest version is not accepted
yet.
c e Zd Zej ZdZdS ) Eula.MetaeulaN)r r r r dbdatabasedb_tabler r r MetarI ^ s ;r rN T) primary_keyN)nulldefaultr1 c t t | | j | j d d }|d u S )N ) nextiterr9 r= acceptedis_nullr: updatedr< )cls
unaccepteds r is_acceptedzEula.is_acceptedg sp
s|++--..#+&&q
T!!r c | t j | j d S )N)rV )updater7 r= rV rW r@ )rY s r acceptz Eula.acceptt sI
DIKK
((.. L " "
')))))r )r1 N)
r r r r rN r rX r rV classmethodboolr[ r^ r r r rG rG W s
iD)))G|t444H
"D
"
"
" [
" [ r rG c B d t j | d D S )Nc g | ]\ }}|S r r ).0_objs r
z get_models..| s, As r c j t j | o t | t o
| t k S r )inspectisclass
issubclassr
)re s r z get_models.. s1 ,, 3&& u r )rh
getmembersmodules r
get_modelsro { s= (
r c t j t j t | d d S )NT)safe)r rK connect
create_tablesro rm s r rs rs s? K K j 0 0t <<<<
r OrderByNc S r r )nodess r rk z,ApplyOrderBy.resolve_nodes.. s % r )r$ r rel_modelr4 )_modelrv modelcustom_order_bynoderz s @r
resolve_nodeszApplyOrderBy.resolve_nodes s !+6? C C OF !%D99 &HGO[----HHJJE 5+t44D r column_namesc |d |dd }}t | | }g }|D ]M}|r4t || D ]}| | 8| | N|S )z
:param model: peewee.Model or peewee.ForeignKeyField
:param column_names: list
:return: list
r rS N)ru r get_nodesappend)r} r rv restrz result
node_or_modelr s r r zApplyOrderBy.get_nodes s )O\!""-=T **5+>> " - -M
- ( 2 2=$ G G ( (DMM$''''(
m,,,,
r c g }|D ]j}t ||j d }|D ]2}| |j r| n| 3k |j | S )z
:param order_by: list
:param model: peewee.Model or peewee.ForeignKeyField
:param query_builder: peewee.Query
:return: peewee.Query with applied order_by
.)ru r rv splitr r; r: )r r: r}
query_builderordersorderrz r s r __call__zApplyOrderBy.__call__ s C CE **5%2C2I2I#2N2NOOE
C
C
UZ AdiikkkTBBBB
C &}%v..r N)
r r r staticmethodstrtupler r% r r r r r ru ru s} 3 5 \& t \$
/
/
/
/
/r ru ) rh loggingr r7 peeweer r r r r r defence360agent.modelr r
r8 getLoggerr loggerr r! r( r- intrE rG ro rs ru apply_order_byr r r r s 2 1 1 1 1 1 1 1
8 $ $ " " " " "I " " " I ,/ 6! ! ! ! !5 ! ! !H = = =
5/ 5/ 5/ 5/ 5/ 5/ 5/ 5/p r