Edit File: 128_move_cleanup_storage_files.cpython-311.opt-1.pyc
� �*vh� � � � d dl Z d dlZd dlZd dlmZmZ d dlmZ d dlm Z ej ddd�� � Z e j e � � Zd� Zd � Zd d �Zd d �ZdS )� N)� CharField�Model)�importer)� FilenameFieldz imav.malwarelib.cleanup.storage�CleanupStorage)�module�name�defaultc �2 � � G � fd�dt � � }|S )zl Model stub for migration because we can't use migrator.orm[] due to custom field FilenameField c � �� e Zd Z G � fd�d� � Z ed�� � Z ed�� � Z ed�� � Z ed�� � Z e de fd�� � Z dS ) � get_model.<locals>.MalwareHitc � �� e Zd ZdZ� ZdS )�"get_model.<locals>.MalwareHit.Meta� malware_hitsN)�__name__� __module__� __qualname__�db_table�database��dbs ��n/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/migrations/128_move_cleanup_storage_files.py�Metar s �� � � � � �%�H��H�H�H� r F)�nullT�returnc � � t j j � | j | j | j g� � S # t $ r Y dS w xY w)zZ Get file name for cleanup storage :return: file name N)�os�path�extsep�join�user�hash�size� TypeError)�selfs r � storage_namez*get_model.<locals>.MalwareHit.storage_name# sN � � ��w�~�*�*�D�I�t�y�$�)�+L�M�M�M�� � � � ��t�t� ���s �58 � A�AN) r r r r r r" r � orig_filer# r$ �property�strr' r s �r � MalwareHitr s� �� � � � � � � � � � � � � � � � �y�e�$�$�$��!�M�u�-�-�-� ��y�d�#�#�#���y�d�#�#�#�� � �#� � � � �� � � r r+ )r )r r+ s ` r � get_modelr, s? �� � � � � � � � �U� � � �* �r c �, � t t j j | |f� � \ } }t t | |f� � \ } } t j | |� � d S # t $ r Y d S t $ r&}t � d|� � Y d }~d S d }~ww xY w)Nz2Failed to move stored file to the new location: %r) �mapr r �joinpathr* �shutil�move�FileNotFoundError� Exception�logger�error)�src�dst�errs r �_mover9 1 s� � ��>�&�/�#�s��<�<�H�C���3��c� �#�#�H�C��P�� �C�� � � � � �� � � � � ��� � P� P� P�� � �I�3�O�O�O�O�O�O�O�O�O�����P���s � A � B�$ B�-B�BFc � � |rd S t |� � }|D ];}|j }|�� t � |j � � }t ||� � �<d S �N)r, r' r r( r9 )�migratorr �fake�kwargsr+ �hitr6 r7 s r �migrater@ < si � � � ����8�$�$�J�� � ��� �� �;� � �)�)�#�-�8�8�� �c�3����� � r c � � d S r; � )r< r r= r> s r �rollbackrC K s � ��Dr )F)�loggingr r0 �peeweer r �defence360agent.utilsr �$defence360agent.model.simplificationr �getr � getLoggerr r4 r, r9 r@ rC rB r r �<module>rJ s� �� ���� � � � � � � � � #� #� #� #� #� #� #� #� *� *� *� *� *� *� >� >� >� >� >� >� �� � ,� � �� � �� �� �8� $� $��� � �<P� P� P� � � � � � � � � � r
Back to File Manager