Edit File: 187_fix_scan_unserialization.cpython-311.opt-1.pyc
� �*vh � � � d Z ddlZddlZddlZddlmZ ej e� � Z ed� � Z dZ dZ G d� dej � � Z d � Zd d �d �Zd d �d �ZdS )zZ Used to fix issue with inability to unserialize stored scans. See DEF-23121 for details. � N)�Pathz$/var/imunify360/aibolit/scans.picklezim360.malwarelibzimav.malwarelibc � � � e Zd Z� fd�Z� xZS )� AVUnpicklerc �$ �� t � � � ||� � S # t $ r` |� t � � rDt j |� t t � � � � }t ||� � cY S � w xY w�N) �super� find_class�ModuleNotFoundError� startswith�IM360_MALWARELIB� importlib� import_module�replace� AV_MALWARELIB�getattr)�self�module�name� av_module� __class__s ��l/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/migrations/187_fix_scan_unserialization.pyr zAVUnpickler.find_class s� �� � ��7�7�%�%�f�d�3�3� 3��"� � � �� � �!1�2�2� 0� %� 3��N�N�#3�]�C�C� � � � �y�$�/�/�/�/�/� � ���s �!% �A&B� B)�__name__� __module__� __qualname__r � __classcell__)r s @r r r s8 �� � � � � � � � � � � � � � � r c �� � |� |j dz � � }|� d� � 5 }t j | |� � d d d � � n # 1 swxY w Y |� |� � d S )Nz.temp�wb)� with_namer �open�pickle�dumpr )�obj�path� temp_path�fs r r" r" s� � ����t�y�7� 2�3�3�I� � � �� � � ��� �C����� � � � � � � � � � � ���� � � � � ���d�����s �A�A� AF)�fakec �� � |st � � � sd S t � � � t � � � v r� t � d� � 5 }t |� � � � � }d d d � � n # 1 swxY w Y t |t � � d S # t $ r,}t � dt |� � Y d }~d S d }~ww xY wd S )N�rbz"Failed to load pickle scans %s: %s) � SCANS_PATH�existsr �encode� read_bytesr r �loadr" � Exception�logger� exception)�migratorr' �_�__r& r# �excs r �migrater6 ' sC � � � �:�$�$�&�&� ���� � � � �J�$9�$9�$;�$;�;�;� "�� � ��&�&� ,�!�!�!�n�n�)�)�+�+�� ,� ,� ,� ,� ,� ,� ,� ,� ,� ,� ,���� ,� ,� ,� ,� ��j� !� !� !� !� !�� � � � � � � �4�j�#� � � � � � � � � ����� ���� <�;s< �B; �+"B� B; �B � B; � B �!B; �; C1�!C,�,C1c � � d S r � )r2 r' r3 r4 s r �rollbackr9 7 s � ��Dr )�__doc__r �loggingr! �pathlibr � getLoggerr r0 r* r r � Unpicklerr r" r6 r9 r8 r r �<module>r? s� ��� � � � � � ���� � � � � � � � � � � �� �8� $� $�� �T�8� 9� 9� �%� �!� � � � � � �&�"� � � � � � � %� "� "� "� "� "� !&� � � � � � � r
Back to File Manager