B
    f              
   @   s  d dl Z d dlZd dlmZmZ d dlZd dlZd dlZd dlZd dl	m	Z	m
Z
mZmZ d dlmZ d dlmZ d dlmZ d dlmZmZmZ d dlT d d	lmZ d dlZd dlZd dlZd dlZd d
lmZ d dl Z d dl!Z!d dl"Z"d dl#Z#d dl$m%Z% d dl&m'Z' d dl(m)Z) d dl*Z*dZ+G dd dZ,G dd dZ-G dd dZ.G dd dZ/dd Z0dd Z1dd Z2dd Z3dd  Z4d!d" Z5d#d$d%d&Z6d'd( Z7dod)d*Z8d+d, Z9dpd-d.Z:dqd0d1Z;dre<e)e< e)e< e<d2d3d4Z=e<e<d5d6d7Z>d8d9 Z?d:d; Z@d<d= ZAd>d? ZBd@dA ZCG dBdC dCZDdsdDdEZEdFdG ZFdHdI ZGdJdK ZHdtdLdMZIdNdO ZJe<e<dPdQdRZKe<e<dPdSdTZLdue<e<dUdVdWZMdXdY ZNdvd[d\d]d^ZOd_d` ZPdadb ZQdcdd ZRdedf ZSG dgdh dhZTdidj ZUdwdmdnZVdS )x    N)urlparseparse_qs)datetimedate	timedeltatimezone)Linker)g)Path)timeitProfilerDataCtxprofiler_data_log)*)AES)is_dataclass)OrderedDict)deepcopy)Listz******c               @   s.   e Zd ZdZddddZdd Zdd	 ZdS )
disable_aclut   Отключение проверки прав доступа в ORM для системных частей кодаN)returnc             C   s
   d | _ d S )N)save_acl_flag)self r   ./cmf/util/cmfutil.py__init__"   s    zdisable_acl.__init__c             C   s   dt krt j| _dt _d S )Ndisable_permissionsT)r	   r   r   )r   r   r   r   	__enter__%   s    zdisable_acl.__enter__c             C   s   | j t_d S )N)r   r	   r   )r   excvaluetbr   r   r   __exit__*   s    zdisable_acl.__exit__)__name__
__module____qualname____doc__r   r   r    r   r   r   r   r       s   r   c               @   s.   e Zd ZdZddddZdd Zdd	 ZdS )

run_systemut   Отключение проверки прав доступа в ORM для системных частей кодаN)r   c             C   s   d | _ d | _d S )N)current_usercurrent_acl_admin_mode)r   r   r   r   r   1   s    zrun_system.__init__c             C   s(   t j| _tjt j t j| _dt _d S )NT)r	   r&   cmfappset_current_personZsystem_useracl_admin_moder'   )r   r   r   r   r   5   s    zrun_system.__enter__c             C   s   t j| j | jt_d S )N)r(   r)   r*   r&   r'   r	   r+   )r   r   r   r   r   r   r   r    =   s    zrun_system.__exit__)r!   r"   r#   r$   r   r   r    r   r   r   r   r%   .   s   r%   c               @   s.   e Zd ZdZddddZdd Zdd	 ZdS )
disable_notifyut   Отключение проверки прав доступа в ORM для системных частей кодаN)r   c             C   s
   d | _ d S )N)save_notify_flag)r   r   r   r   r   D   s    zdisable_notify.__init__c             C   s   dt krt j| _dt _d S )Nr,   T)r	   r,   r-   )r   r   r   r   r   G   s    zdisable_notify.__enter__c             C   s   | j t_d S )N)r-   r	   r,   )r   r   r   r   r   r   r   r    L   s    zdisable_notify.__exit__)r!   r"   r#   r$   r   r   r    r   r   r   r   r,   B   s   r,   c               @   sJ   e Zd ZedZddddZdd Zdd	 Zed
d Z	edd Z
dS )enable_import_modez/tmp/eva_app_import.lockN)r   c             C   s   d | _ d | _d S )N)save_import_flagsave_cache_flag)r   r   r   r   r   S   s    zenable_import_mode.__init__c             C   sD   dd l }|jdd| _dtkr(tj| _dt_d|jd< |   d S )Nr   NO_CACHE import_modeT1)osenvirongetr0   r	   r3   r/   import_heartbeat)r   r5   r   r   r   r   W   s    
zenable_import_mode.__enter__c             C   s4   dd l }| jt_| j|jd< | j r0| j  d S )Nr   r1   )	r5   r/   r	   r3   r0   r6   	LOCK_PATHexistsunlink)r   r   r   r   r5   r   r   r   r    `   s
    
zenable_import_mode.__exit__c          	   C   s6   t | jd }|tjtjdd W d Q R X d S )Nzw+)tzz%Y-%m-%d %H:%M:%S %Z%z)openr9   writer   nowr   utcZstrftime)clsfr   r   r   r8   j   s    z#enable_import_mode.import_heartbeatc          	   C   s`   dd l }tj| jsdS t| jd}|j| }W d Q R X t	j
tjd|  d dk S )Nr   Fr)r<   <      )Zdateutil.parserr5   pathr:   r9   r=   parserparsereadr   r?   r   r@   total_seconds)rA   ZdateutilrB   	lock_timer   r   r   import_is_runningo   s    z$enable_import_mode.import_is_running)r!   r"   r#   r
   r9   r   r   r    classmethodr8   rL   r   r   r   r   r.   P   s   	
r.   c             C   s   t tdsd S d|  }|tjkr&d S t }tjjj|dd}|jddd t | }|dkrzt	d|d	d
|  |dkrt
jd|d	d
|  |dkrt tdddkrt
jd|d	d
|  |tj|< t tj|< d S )Nacquired_lockszCmfCacheObjLock:   )timeoutT)Zblockingblocking_timeoutg?zcache_obj_lock_get long time: z0.3zsec    z,DEV: cache_obj_lock_get long time CRITICAL: rE   i  z#DEV: cache_obj_lock_get long time: )hasattrr	   rN   timeAPPREDIS_DBredislockacquiredebugr(   include	cmf_alertr   Ztodayacquired_lock_timigns)obj_idZlock_keyst_timerX   Zacquire_timer   r   r   cache_obj_lock_get   s$    



r`   c              C   s   t tdsd S t } xVtj D ]H}y|  W q" tjjk
rh } zt	d|  W d d }~X Y q"X q"W t |  }|dkrt	d|ddtj
   xDtj D ]6\}}t | }|dkrt	d| d|dd	 qW d S )
NrN   z*cache_obj_lock_release_all release error: g?z%cache_obj_lock_release_all too slow: z0.3zsec. Locks:z(cache_obj_lock_release_all lock >100ms:  Zsec)rS   r	   rT   rN   valuesreleaserW   
exceptionsZ	LockErrorrZ   keysr]   items)r_   rX   eZunlock_timeZ	lock_nameZlock_strK   r   r   r   cache_obj_lock_release_all   s    
&rh   c              C   s8   t tdr4tjr4xtjD ]} tjjj|   qW g t_d S )Ndelayed_redis_events)rS   r	   ri   rU   rV   rW   Zpublish)Zeventr   r   r   emit_delayed_events   s    rj   c             C   s   t t|  S )N)varsmodels)
class_namer   r   r   get_model_by_name   s    rn   c             C   s   t | }tt| S )N)get_class_name_by_idrk   rl   )idrm   r   r   r   get_model_by_id   s    rq   c             C   s   t | dd }|S )N:r   )strsplit)rp   rm   r   r   r   ro      s    ro   F)simplec            O   sH   t | tjjr| j }n|  }t|}|r0|jn|j}||d|i|S )Nrp   )
isinstancer(   rl   	CmfEntityrp   rq   sgetr7   )rp   ru   argskwargsZtuuidmodelr7   r   r   r   get_obj_by_id   s    
r|   c             C   s   t |trtt| }n|}t | tjjr@t | tjjs@t| } t | tr| |j	 dr`| S |j
| ddgd}|r||jS |j
| ddgd}|r|jS dS t| jS )u  
    Находим id объекта. Функция для использования в API и bzPython ручек, которые используют клиенты.
    Реализуем максимально универсальную логику, чтобы клиент мог получить объект по:
    - id
    - code
    - name
    - передать сам объект
    - передать field модели со ссылкой наобъект
    Если передается id - функция отрабатывает максимально быстро, т.к. используется в кешах
    Если передается code или name - будут доп.запросы в БД.
    Сначала ищем по code и только если не находим - по name
    В class_name можно передать имя модели или саму модель
    rr   z--rp   )codefields)namer~   N)rv   rs   rk   rl   r(   r~   CmfType
CmfRelBase
startswithrm   rx   rp   )objrm   r{   Z
loaded_objr   r   r   get_obj_id_by_any   s     

r   c             c   sb   ddl m} ddlm} xDt|j D ]2}t||}t||r(|j	rHq(| rT| |r(|V  q(W dS )u`   Последовательность моделей, с возможно фильтрацией.r   )rl   )BaseModelMetaN)
cmf.includerl   cmf.models.base_modelr   sorted__dict__re   getattrrv   Zabstract)Zmodel_filterrl   r   Z
model_namer{   r   r   r   iter_models   s    
r   c             C   s>   ddl m} ddlm} x | D ]}|j| kr"|jS q"W d S )Nr   )rl   )rw   )r   rl   r   rw   Ziter_subclassesui_namerm   )r   rl   rw   r{   r   r   r   get_class_name_by_ui_name   s    

r   c             O   s   |st ddd t|tkr$t|}|j|ddd| gi|}|s| d k	rd| kr| dd }| rt|d	kr|j|ddd
d| gi|}|S )Nu|   TODO: сделать автоматическое обнаружение моделей и объекта как в bzPythonT)abortfilterr}   =-   ZLIKEz%-)r\   typers   rn   r7   rt   isdigitlen)r}   r{   ry   rz   resZcode_numberr   r   r   get_obj_by_code  s     r   _c                s(   t j| ddd}d fdd|D S )NruT)Zlanguage_codereversedr2   c             3   s*   | ]"}|  s|d kr| n V  qdS )r   N)isalnumlower).0c)sepr   r   	<genexpr>  s    z!translit_strip.<locals>.<genexpr>)transliterateZtranslitjoin)textr   r   r   )r   r   translit_strip  s    r   )r   tags_extendprotocols_extendr   c             C   s   |r d|kr |  dd dd} tj}dddg|d	< d
g|d< dddg|d< dddg|d< tj}|d	 |d |d |d |d |d |d |d |d |d |r|| tj}|r|| tj| tj||ddS )NZdivz<div>z<p>z</div>z</p>srcwidthZheightZimgr   rT   Zcontrolsr}   ZaudioZvideopbrZinsdelZfigureZ
figcaptionT)r   tags
attributes	protocolsstrip)replacebleachZALLOWED_ATTRIBUTESZALLOWED_TAGSappendextendZALLOWED_PROTOCOLSZclean)r   r   r   r   r   r   r   r   r   
html_clean  s0    












r   )r   r   c             C   s    ddd}t |gd}|| S )NFc             S   s.   |  ds| S t| d }d| d< d| d< | S )N)NZhrefZ_blank)NtargetZexternal)Nclass)r7   r   )Zattrsnewr   r   r   r   
set_target=  s    
zlinkify.<locals>.set_target)Z	callbacks)F)r   linkify)r   r   Zlinkerr   r   r   r   <  s    
r   c                s   i  fdd}|S )Nc                 s4   t | t| f}|kr, | ||< | S )N)pickledumpsr   rf   )ry   rz   hash)funcmemoryr   r   memoK  s    zmemoized.<locals>.memor   )r   r   r   )r   r   r   memoizedI  s    r   c             K   s   t jj}d|  }|jddd}| s<tjjddd dS ||sb|	  t
d	 t| d
S t||}t
jj|d kr|	  tjjddd dS ttj|d< ||t| |	  dS )Nz	obj_lock-zobj_lock.lock
   )rP   z(DEV: FATAL lock(): if not lock.acquire()T)r   FuR   DEV: FATAL Попытка lock_ping() на отсутствующую записьi  	locked_byuJ   DEV: FATAL Попытка lock_ping() на чужую блокировку	ping_time)rU   rV   rW   rX   rY   r(   r[   r\   r:   rc   r	   rZ   r   loadsr7   current_personrp   r   r?   r   r@   setr   )keyrz   redis_db_key_lock	lock_infor   r   r   	lock_pingS  s&    



r   c             C   sD   | st dtjj}d|  }||r@t||}|r@|S dS )uh   
    Из редиса заберём структуру, где есть данные
    :return:
    uL   Блокировка объектов без key не реализованаz	obj_lock-N)
ValueErrorrU   rV   rW   r:   r   r   r7   )r   r   r   r   r   r   r   r   l  s    

r   c             C   s  t jj}d|  }|jddd}tjjjt	t
jt	t
j|d}| s`tjjddd d	S ||rt||}t	t
j|d
   |k rtjj|d kr|  tjj|d d}tjd|j  d	S tjd |  dS ||t| |  dS )uT   
    Захват объекта на редактирование
    :return:
    z	obj_lock-zobj_lock.lockr   )rP   )r   Z
lock_startr   lock_timoutz(DEV: FATAL lock(): if not lock.acquire()T)r   Fr   r   )rp   u[   Редактирование временно невозможно.
Редактирует u`   DEV: WARNING lock() попытка захватить свою блокировку дважды)rU   rV   rW   rX   r	   r   rp   r   r   r?   r   r@   rY   r(   r[   r\   r:   r   r   r7   rJ   rc   rl   Z	CmfPersonr   r   r   )r   Zlock_timeoutr   r   rX   r   personr   r   r   rX   }  s0    



rX   c             K   s   t jj}d|  }|jddd}| s<tjjddd dS ||s^tjd	 |	  dS t
||}ttj|d
   |d k rtjj|d kr|	  dS || |	  dS n|| |	  dS dS )u2   
    Снятие захвата
    :return:
    z	obj_lock-zobj_lock.lockr   )rP   z*DEV: FATAL unlock(): if not lock.acquire()T)r   Fus   DEV: WARNING unlock() попытка разблокировать несуществующую блокировкуr   r   r   )rU   rV   rW   rX   rY   r(   r[   r\   r:   rc   r   r   r7   r   r?   r   r@   rJ   r	   r   rp   delete)r   rz   r   r   rX   r   r   r   r   unlock  s,    

 

r   c               @   s0   e Zd ZdZdddddZdd Zd	d
 ZdS )CmfLockuJ   Глобальная блокировка аппа на основе redis  N)r   c             C   s$   ddl m} |jj|||d| _d S )Nr   )rV   )rP   rQ   )r   rV   rW   rX   
redis_lock)r   r   rP   rQ   rV   r   r   r   r     s    zCmfLock.__init__c             C   s    | j  std| j j d S )NzCannot lock )r   rY   ZCmfGetLockErrorr   )r   r   r   r   r     s    
zCmfLock.__enter__c             C   s   | j   d S )N)r   rc   )r   r   r   r   r   r   r   r      s    zCmfLock.__exit__)r   N)r!   r"   r#   r$   r   r   r    r   r   r   r   r     s   r   c             K   s  t js
dS | s|sdS ttdd }dt_dt_|r<t  nt }t	j
j|d< ||d< t|d< t|d< t|d< t|d< t|d	< t	j
j|d
< |r|| yh|rt|}|dr|dd  }|dr|dd  }td| d}| } W d Q R X t| |d}	t|	| |d |}
n| rtj| dd}d}x.|jD ]$}t|tjr<|jdkr<d}q<W |s|jtj tj!ddddddd tjd|jtj"g tj#ddddgg d g d g dddg d}|g|_t|dd}	t|	| |d |}
ntddd W nv t$k
rx } zV|t_dt_t|t	j%j&r6|dd l'}|( }t	j
jd| d| d| dd W d d }~X Y nX |t_dt_|
S )NTdisable_raise_lazyloadcmfutilr   r   r   r   r   requestsr\   z/opt/eva-app/   /rE   rC   execrun)modeFr   )r   lineno
col_offset)r   r   )Zposonlyargsry   defaultsZvarargZ
kwonlyargsZkwargZkw_defaults)r   bodyry   r   r   Zdecorator_listz<string>uC   Попытка вызова несуществующего bzPython)r   uT   Произошла ошибка при выполнении bzPython объекта z: 
))configZBZPYTHON_ENABLEDr   r	   r   Zrelaxed_bz_python_modeglobalscopy_build_safe_bzpython_globalsr(   r[   r   r   r   r   r   r   r\   updaters   r   r=   rI   compiler   astrH   r   rv   ZFunctionDefr   r   ZReturnZConstantZ	argumentsarg	ExceptionZ
base_errorZCmfAbortError	traceback
format_exc)r}   r   Zcode_src_objZeval_file_pathZeval_unsaferz   Zprev_disable_raise_lazyloadlrB   Z
code_blockresultZcode_astZhave_fnstfnrg   r   Ztracer   r   r   exec_bzpython  s    






r   c                s    fdd}|S )Nc                 s   t jjd j ddd d S )Nu   Вызов функции "up   " запрещен, требуется разрешить расширенные возможности BzPythonT)r   )r(   r[   r\   r!   )ry   rz   )rB   r   r   wrapper9  s    z_restrict_func.<locals>.wrapperr   )rB   r   r   )rB   r   _restrict_func8  s    r   c             O   sP   t | tst| f||S | dr<d| kr<t| f||S tjjddd d S )Nz/tmpz..u   Неверный путь к файлу или несуществующий файл, требуется разрешить расширенные возможности BzPythonT)r   )rv   rs   r=   r   r(   r[   r\   )rB   ry   rz   r   r   r   _safe_bzpython_openA  s    
r   c                 sV   i } t  }tt|d< tt|d< t|d< t td fdd}||d< || d< | S )	Nevalr   r=   )r   c                s@   |  dd } | tjkr0tjjd|  ddd  | f||S )N.r   u   Пакет "u)   " нельзя импортироватьT)r   )rt   r   ZBZPYTHON_ALLOWED_LIBSr(   r[   r\   )r   ry   rz   )global_importr   r   _bzpython_importY  s    
z6_build_safe_bzpython_globals.<locals>._bzpython_import
__import____builtins__)r   r   r   r   r   r   r   rs   )Zbzpython_globalsbuiltinsr   r   )r   r   r   O  s    	r   c       
      C   s  t | }|jd}|r$|d dkr(d S t|jddgd }|rL|sLd S |d }|s\d S d|krx|jddd	\}}n d
|kr|jd
dd	\}}nd S |dkrtj}n&|dkrtj	}n|dkrtj
}ntj}i }	|r||	d< d
|krtf d|j d| i|	S tf d| i|	S )Nr   rE   )ZshareZdocsr   r2   r   r   rr   )maxsplitr   )docZDOCCmfDocument)ZflZFLZIN_WORK	CmfFolder)CmfListZEPIZsharelink_hashrp   r}   )r   rF   rt   r   Zqueryr7   rsplitrl   r   r   r  r|   rm   r   )
Zurl_strZcheck_sharelinkZurl
path_partsZ
hash_paramZobj_keyZobj_typeZobj_codeZ	obj_modelZ
get_kwargsr   r   r   get_url_path_obji  s8    
r  c              C   s6   t jd\} }}t|}t|}t|tj|S )Nrr   )r   ZEVA_INSTANCE_KEYrt   binasciiZ	unhexlifyr   r   ZMODE_CBC)r   Ziv_hexZkey_hexr   Zivr   r   r   
get_cipher  s    

r  )messager   c             C   s@   t  }| dtjt| tj    }|| }t|dS )Nra   zutf-8)	r  r   Z
block_sizer   Zencryptencodebase64Z	b64encodedecode)r  cipherZpadded_messageZencrypted_messager   r   r   crypt  s    r  c             C   s   t  }|t|   S )N)r  decryptr	  Z	b64decoder
  r   )r  r  r   r   r   r    s    r  )r  r   c          	   C   s^   t j }xN|j D ]@}|r<t jj||d| |p2| dd qt jj|d| |pN| dd qW d S )Nu   ОшибкаrE   )r   r   r   msgr   priority)r   r   r  r   r  )rl   ZCmfPersonGroupZadmin_groupZ
rg_membersZ
all_nestedZ	CmfNotifyZplace_notify)r  r   r   Z	admin_grpr   r   r   r   admin_alert  s     

r  c              C   sl   t  } tjrhd }tj}d|kr.|d\}}d| d| d| _|rh|d\}}t j||| _| S )N@zhttp://)ZhttpZhttpsrr   )r   ZSessionr   ZEGRESS_PROXYrt   ZproxiesZauthZHTTPProxyAuth)requests_sessionZ
auth_proxyZproxy_settingsZusernameZpasswordr   r   r   r    s    r  r>   zlist|CmfModel|SimpleModel)r   c             C   s6   t | tr&x&| D ]}t||d qW nt| d| dS )u   
    Маркируем объект проверенный бизнес логикой политикой по умолчанию.
    Сейчас маркировка проверяется только в api(eva-app)
    )policy_acl_policyN)rv   listacl_set_policysetattr)r   r  r   r   r   r   r    s    

r  c             C   s@   ddl m} t| ddd}t| |jjr2|  n|  |jS )Nr   )baseT)no_metano_acl)	cmf.data_providersr  CmfResultSplitterrv   SimpleMapperSimpleModelsplit_simplert   r   )r   r  Zsplitterr   r   r   
dumps_dict  s    
r   c             C   s8   ddl m} t| }t| |jjr*t|S t|S d S )Nr   )r  )	r  r  r   rv   r  r  	cmf_dumpsujsonr   )r   r  Zres_dictr   r   r   
dumps_json  s
    r#  c             C   s   t | tjrt| S t | tjr(t| S t | tjjrdd | j	
 D }x4| j	
 D ]&\}}t |tjrV|jrV| ||< qVW | j|d< |S t| | jd S )Nc             S   s0   i | ](\}}t |tjjr|jd k	r|j|qS ).)rv   r(   r~   r   _value)r   attrfieldr   r   r   
<dictcomp>  s   z%cmf_dumps_default.<locals>.<dictcomp>rm   )rv   	ipaddressIPv4Addressrs   decimalDecimalr(   rl   	BaseModelr   rf   ZfieldsCmfTypeZvirtualZvirtual_getterrm   	TypeError	__class__)r   r   r%  r&  r   r   r   cmf_dumps_default  s    
r/  c          
   C   sT   yt j| tt jdS  tk
rN } ztd| d|   |W d d }~X Y nX d S )N)defaultoptionzorjson.dumps error e=z, obj=)orjsonr   r/  ZOPT_NON_STR_KEYSr-  r	   rZ   )r   rg   r   r   r   r!    s
    r!  c               @   s   e Zd Zd%ddZdd Zdd Zd&d	d
Zd'ddZd(ddZdd Z	ed)e
jjdddZdd Zdd Zdd Zdd Zdd Zdd  Zd!d" Zd#d$ ZdS )*r  NFc             C   sP   || _ i | _t | _d | _d | _d | _d| _t | _d| _	|| _
i | _|| _d S )Nr   )_objmetar   meta_fields_keepr   startendnumber_of_objectsm2m_id_cachedepthr  3_CmfResultSplitter__project_perm_add_comments_cacher  )r   r   r  r  r   r   r   r   "  s    zCmfResultSplitter.__init__c                s    fdd}|S )Nc                sL   |  j d7  _ | j dkrd S  | f||}|  j d8  _ | j dksHt|S )NrE   r   r   )r:  AssertionError)r   ry   rz   ret)r   r   r   wrap1  s    
z,CmfResultSplitter._check_depth.<locals>.wrapr   )r   r>  r   )r   r   _check_depth0  s    zCmfResultSplitter._check_depthc             C   s"   |j | jkrt|j| j|j < d S )N)rm   r4  r   Zui_meta)r   or   r   r   _add_class_to_meta;  s    z$CmfResultSplitter._add_class_to_metac             C   sZ  | d}| d}| d}| d}| d}| d}| d}	tt|}
i }d}d}|
jsx t|D ]}|d	krzql||= qlW d
|d< |S tjj|||d||||	|d	pt }t|
t	jj
rL|
jrL|rL|dd }tt|d}|j |
j}| d}| d}|r|jsLtjj||||
j|||d|d	}t|t|@ }|dkr|| dr|tj||s|d}t }| ddrt|}d|kr|d t|}d|krd|krdnd
}t| }tjdstjdrd|
jkrd|kr|d x$|D ]}|tjkr6||= q|d	krDq|
j |}|r|jrbq|js|||= d
||< qt|t	jjrt||< d}tjj||||||||	d|d
pt }|dkr|rt }|dkr|r|dr|| jkr(| j| sd|krt|}|d nt|}t|ddgdd }|jjpR|j j}|sd|krvt|}|d d| j|< nH|j!d!|dd"dkrd|krt|}|d d| j|< n
d| j|< |d#kr|d$krt|}|"d d|krd|krdnd
}|r|||< |d
kr||kr||= qW |rH||d%< |rV||d< |S )&u;  
        Чтобы код не дблировать для модели, простой модели и дикта, можно проверять итоговый json
        Правда в таком случае, мы обходим все данные, даже если они не доступны.
        Zperm_effective_acl_idZcmf_owner_idrp   	parent_idrm   perm_parent_id%perm_security_level_allowed_ids_cacheN)rp   rm   ZdenyZ_acl_objF)	initial_acl_keyobject_modelobject_owner_idraise_error	object_idobject_parent_idobject_dictperm_security_level_allowed_idschecked_policyrr   r   perm_inherit_acl_idperm_parent_owner_id)	rE  rF  rG  object_fieldrI  rJ  rK  rH  rM  
CmfCommentZprivateTZcmf_deletedr>   rI   readonlyz.getz.ui_getZcomments)
rE  rF  rP  rG  rI  rK  rJ  rL  rH  rM  zCmfProject:
project_id)r~   Zinclude_deletedzPPP-COM-ADD)r   rH  r}   Z
CmfProjectZ_acl_fields)#r7   r   rl   Z	api_allowr  ZCmfAccessListZcheck_accessr   
issubclassr(   rw   Zacl_parent_field	partitionr~   r  rQ  Zcheck_visibilityremovetuplere   r	   Z
api_methodendswithr   Zapi_hack_fieldsZCmfPassword_PASSWORD_MASKr   r;  r|   rS  r   rB  Zcheck_project_role_accessadd)r   Zobj_dict
acl_policyrE  Zobj_owner_idr^   Zobj_parent_idrm   rC  rD  r{   Z
acl_fieldsZacl_objZcomment_deny
field_nameZaccess_levelsZparent_class_nameZparent_modelZparent_fieldrN  rO  Zparent_access_levelsZ_tmpZobj_keysr&  Z	acl_fieldZfield_access_levelsZproject_Zobj_Zproject_id_r   r   r   _acl_obj_dict?  s    






















zCmfResultSplitter._acl_obj_dictr   c             C   s   |  j d7  _ | js| | d|ji}x4|jddD ]$}|jsBq6| j||d d||j< q6W | js|jdkr| j	|t
|dd d}|S )	NrE   rm   T)
is_defined)r:  Zpublic_readr  )r[  )r8  r  rA  rm   rb   r^  _process_cmf_typer  Zacl_typer]  r   )r   r@  r:  r   vr   r   r   _process_cmf_model  s    

z$CmfResultSplitter._process_cmf_modelc             C   s   |  j d7  _ | js*tt|j}| | i }x<|j D ].\}}| jsZ| j	|j|f | 
|||< q:W | js| j|t|dd d}|S )NrE   r  )r[  )r8  r  r   rl   rm   rA  r   rf   r5  rZ  _processr  r]  )r   r   r:  r{   r   r%  r   r   r   r   _process_simple_model  s    
z'CmfResultSplitter._process_simple_modelc                s    fdd  |S )Nc                s   t | tr fdd| D S t | tr| d} fdd|  D }t |trd|krȈ jd7  _tjj	
|}js| x| D ]}j|j|f qW |j|d< jsȈj|| d	d
}|S | S )Nc                s   g | ]} |qS r   r   )r   el)proccess_jsonr   r   
<listcomp>  s    zSCmfResultSplitter._process_json_as_model.<locals>.proccess_json.<locals>.<listcomp>rp   c                s   i | ]\}} ||qS r   r   )r   kr`  )re  r   r   r'    s    zSCmfResultSplitter._process_json_as_model.<locals>.proccess_json.<locals>.<dictcomp>rr   rE   rm   r  )r[  )rv   r  dictr7   rf   rs   r8  r(   r~   ZCmfTUUIDZget_cls_by_tuuid_strr  rA  r5  rZ  rm   r  r]  )r   r^   datar{   r\  )re  r   r   r   re    s"    





z?CmfResultSplitter._process_json_as_model.<locals>.proccess_jsonr   )r   r@  r   )re  r   r   _process_json_as_model  s    z(CmfResultSplitter._process_json_as_model)r@  c             C   sf  |  j d7  _ | js*| j|jj|jf t|tjj	rD| 
|jS t|tjjr^| |jS t|tjjrx| |jS t|tjjrt|| jkrd S | jt| |jrd S | j|j|d d}| jt| |S t|tjjrJt|| jkrg S | jt| g }x$|jD ]}| |}|| qW | jt| |S |j}|dkr`|jS |jS )NrE   )r:  .)r8  r  r5  rZ  instancerm   rv   r(   r~   ZCmfObjectJsonrj  r   ZCmfObjectList_process_listZ	CmfObjectra  ZCmfRelationBaserp   r9  Zis_nullrV  r   r   Zjsonr0  )r   r@  r:  r   r   ir   r   r   r   r_    s@    

z#CmfResultSplitter._process_cmf_typec                s$     j d7  _  fdd|D }|S )NrE   c                s   g | ]}  |qS r   )rb  )r   rg   )r   r   r   rf  C  s    z3CmfResultSplitter._process_list.<locals>.<listcomp>)r8  )r   r@  r   r   )r   r   rl  A  s    zCmfResultSplitter._process_listc             C   s   |  j d7  _ t }x| D ]z\}}| |||< | js|dkr|rt|trtt|	dd d }|r| 
| x"| D ]}| j|j|f q~W qW |dr|dr| js| j||dd}|S )NrE   rp   rr   r   rm   r  )r[  )r8  r   rf   rb  r  rv   rs   r   rl   rt   rA  re   r5  rZ  rm   r7   r  r]  )r   r@  r   rg  r`  mr\  r   r   r   _process_dictF  s    
zCmfResultSplitter._process_dictc             C   s   |  j d7  _ t|tr"| |S t|tjjr:| |S t|tjj	rR| 
|S t|rd| |S t|trx| |S t|tjtjfrt|S t|ttfr| S |S )NrE   )r8  rv   r  rl  r(   r~   r   r_  rl   r,  ra  r   rc  rh  ro  r(  r)  r*  r+  rs   r   r   Z	isoformat)r   r   r   r   r   rb  V  s     






zCmfResultSplitter._processc             C   sJ   xD| j  D ]6\}}x,t|d D ]}||f| jkr"|d |= q"W qW d S )Nr~   )r4  rf   r  r5  )r   rm   class_valuer\  r   r   r   _process_metai  s    zCmfResultSplitter._process_metac                sd   d fdd	  j  xDj D ]6\}}x,t|d D ]}||fjkr<|d |= q<W q&W d S )Nr   c       	         s   j d7  _ t| rhtt| j}| x6| j D ](\}}j	|j|f  ||d  q8W n.t
| tjjrtt| j}| xL| j D ]>\}}t
|tr|jdk	rj	|j|f  |j|d  qW nt
| trx| D ]} ||d  qW nt
| trd }| d}|rXt
|trXtt|dd d }|rX| x<|  D ]0\}}|rj	|j|f  ||d  qbW d S )NrE   .rp   rr   r   )r8  r   r   rl   rm   rA  r   rf   r5  rZ  rv   r(   r,  r   r$  r  rh  r7   rs   rU  )	r   r:  r{   Zsub_attrZ	sub_valueZ	sub_fieldsubr^   Zsub_key)process_jsonr   r   r   rs  q  s8    




z<CmfResultSplitter._process_simple_meta.<locals>.process_jsonr~   )r   )r3  r4  rf   r  r5  )r   rm   rp  r\  r   )rs  r   r   _process_simple_metap  s    
z&CmfResultSplitter._process_simple_metac             C   s$   | j r| jsd S t| j| j  d S )Ni  )r6  r7  int)r   r   r   r   ms  s    zCmfResultSplitter.msc             C   s4   t   | _| | j| _| js&|   t   | _d S )N)rT   r6  rb  r3  r   r4  rq  r7  )r   r   r   r   rt     s
    
zCmfResultSplitter.splitc             C   s4   t   | _| | j| _| js&|   t   | _d S )N)rT   r6  rb  r3  r   r  rt  r7  )r   r   r   r   r    s
    
zCmfResultSplitter.split_simple)NF)N)r   )r   )r   )r!   r"   r#   r   r?  rA  r]  ra  rc  rj  r(   r~   r   r_  rl  ro  rb  rq  rt  rv  rt   r  r   r   r   r   r     s$   

  

0'r  c             C   s6   dd l }y|| }| S  |jk
r0   dS X d S )Nr   F)psutilZProcessZ
is_runningZNoSuchProcess)pidrw  Zprocessr   r   r   check_process_running  s    
ry  r   avscanc          
   C   sL  |s(t tjjd\}}}|r(t|}|r4t|nd}ddt | dt |dt |d|dg
}|r|d	 |r|d
 || |	r|d |r|d || |r|d |t | n|d |t | tj|ddddd}d|j	krdS d|p| d|  d| d| d|j	 
}t
d| |
rHt||dS )u{  
    !!! import icapclient - похоже ломает gevent
        Либа читает файл только с диска (


    Используем cli клиент:
    /usr/bin/c-icap-client [-V ] [-VV ]
     [-i icap_servername] [-p port] [-s service] [-tls ] [-tls-method tls_method] [-tls-no-verify ]
     [-f filename] [-o filename] [-method method] [-req url] [-resp url]
     [-d level] [-noreshdr ] [-nopreview ] [-no204 ] [-206 ]
     [-x xheader] [-hx xheader] [-rhx xheader] [-w preview] [-v ]

    -V                      : Print version and exits
    -VV                     : Print version and build informations and exits
    -i icap_servername              : The icap server name
    -p port         : The server port
    -s service              : The service name
    -tls                    : Use TLS
    -tls-method tls_method          : Use TLS method
    -tls-no-verify                  : Disable server certificate verify
    -f filename             : Send this file to the icap server.
    Default is to send an options request
    -o filename             : Save output to this file.
    Default is to send to stdout
    -method method          : Use 'method' as method of the request modification
    -req url                : Send a request modification instead of response modification
    -resp url               : Send a responce modification request with request url the 'url'
    -d level                : debug level info to stdout
    -noreshdr                       : Do not send reshdr headers
    -nopreview                      : Do not send preview data
    -no204                  : Do not allow204 outside preview
    -206                    : Support allow206
    -x xheader              : Include xheader in icap request headers
    -hx xheader             : Include xheader in http request headers
    -rhx xheader            : Include xheader in http response headers
    -w preview              : Sets the maximum preview data size
    -v                      : Print response headers

    Valid:
    root@crm:/opt/eva-app# c-icap-client -f uwsgi.ini -req / -i 10.50.17.68 -s avscan -method POST -v >/dev/null
    ICAP server:10.50.17.68, ip:192.168.0.33, port:1344

    No modification needed (Allow 204 response)

    ICAP HEADERS:
            ICAP/1.0 204 Unmodified
            Server: C-ICAP/0.5.6
            Connection: keep-alive
            ISTag: CI0001-B5sZ3i1I3+P4H3at/XZ4mQAA

    REQMOD HEADERS:
            POST / HTTP/1.0
            Last-Modified: Tue Apr  9 10:46:16 2024
            Content-Length: 1426
            Content-Length: 1426
            User-Agent: C-ICAP-Client/x.xx

    Virus:
    root@crm:/opt/eva-app# c-icap-client -f xf-mccs6.exe -req / -i 10.50.17.68 -s avscan -method POST -v >/dev/null
    ICAP server:10.50.17.68, ip:192.168.0.33, port:1344


    ICAP HEADERS:
            ICAP/1.0 200 OK
            Server: C-ICAP/0.5.6
            Connection: keep-alive
            ISTag: CI0001-B5sZ3i1I3+P4H3at/XZ4mQAA
            X-Infection-Found: Type=0; Resolution=2; Threat=Win.Tool.Genkryptik-9939783-0;
            X-Violations-Found: 1
            -
            Win.Tool.Genkryptik-9939783-0
            0
            0
            Encapsulated: res-hdr=0, res-body=108

    RESPMOD HEADERS:
            HTTP/1.0 403 Forbidden
            Server: C-ICAP
            Connection: close
            Content-Type: text/html
            Content-Language: en

    rr   i@  z/usr/bin/c-icap-clientz-fz-iz-pz-sz-vz-tlsz-tls-methodz-tls-no-verifyz-methodz-reqz-respT   )Zcapture_outputrP   Zcheckr   zNo modification neededzicap_check(): failed, url=z, file_path=z
, service=z	, method=z
, result:
z%sF)rs   r	   Zglobal_settingsZicap_serverrU  ru  r   
subprocessr   stderrZloggingZwarningZCmfIcapCheckError)Z	file_pathZresp_urlZreq_urlmethodZserviceZhostZportZtlsZ
tls_methodZtls_no_verifyrH  r   ry   procr  r   r   r   
icap_check  s>    Y$







*
r  )N)N)r   )NN)NNF)F)NN)r>   )
r   NNrz  NNNNNF)Wr   r|  Zurllib.parser   r   r   r(   r   r   r   r   r   r   r   Zflaskr	   Zpathlibr
   Zcmf.cmf_profiler   r   r   r   ZCrypto.Cipherr   r  r	  Zredis.exceptionsrW   rT   Zdataclassesr   r*  r(  r2  r"  collectionsr   r   r   typingr   r   rY  r   r%   r,   r.   r`   rh   rj   rn   rq   ro   r|   r   r   r   r   r   rs   r   r   r   r   r   rX   r   r   r   r   r   r   r  r  r  r  r  r  r  r   r#  r/  r!  r  ry  r  r   r   r   r   <module>   s    /	
&
	

!
'#
b	
5	   
  