관리-도구
편집 파일: revisium.cpython-311.pyc
� ��gj � � � d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z ddl m Z ddl m Z ddlmZ G d� d � � Z G d � d e� � Z G d � d e� � ZdS )u This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with this program. If not, see <https://www.gnu.org/licenses/>. Copyright © 2019 Cloud Linux Software Inc. This software is also available under ImunifyAV commercial license, see <https://www.imunify360.com/legal/eula> � N)�suppress)�Path)�Optional)�Core)� snake_casec �R � e Zd ZdZdd�Zd� Zd� Zed� � � Zed� � � Z dS ) �RevisiumTempFile�txtNc �r � |�t t j � � nt |� � | _ || _ d S �N)r r �TMPDIR�tempdir�mode)�selfr r s �S/opt/imunify360/venv/lib/python3.11/site-packages/imav/malwarelib/utils/revisium.py�__init__zRevisiumTempFile.__init__& s- � �,3�O�t�D�K�(�(�(��g� � �� ��� � � � c �$ � d| j t t j � � dz � � | j fz }| j |z | _ | j rF| j � � � t j t | j � � | j �� � | S )Nz%s_%d.%sg �cA)r ) �name�int�time�extr �_pathr �touch�os�chmod�str)r �filenames r � __enter__zRevisiumTempFile.__enter__* sz � � ���C�� � � �d�0B�,C�,C�T�X� N�N���\�H�,�� � �9� 6� �J� � � � � � �H�S���_�_�4�9� 5� 5� 5� 5�� r c � � t t � � 5 | j � � � d d d � � d S # 1 swxY w Y d S r )r �FileNotFoundErrorr �unlink)r �_s r �__exit__zRevisiumTempFile.__exit__2 s� � � �'� (� (� � � �J� � � � � � � � � � � � � � � � � ���� � � � � � s �<�A �A c �* � t | j � � S r )r r �r s r r zRevisiumTempFile.filename6 s � ��4�:��� r c �4 � t | j j � � S r )r � __class__�__name__r& s r r zRevisiumTempFile.name: s � ��$�.�1�2�2�2r )NN) r) � __module__� __qualname__r r r r$ �propertyr r � r r r r # s� � � � � � � �C�� � � �� � � � � � � � � �X� � �3� 3� �X�3� 3� 3r r c � � e Zd ZdZd� ZdS )�RevisiumJsonFile�jsonc � � | j � � � 5 }t j |� � cd d d � � S # 1 swxY w Y d S r )r �openr0 �load)r �fs r �readzRevisiumJsonFile.readB s� � � �Z�_�_� � � �!��9�Q�<�<� � � � � � � � � � � � ���� � � � � � s �;�?�?N)r) r* r+ r r5 r- r r r/ r/ ? s( � � � � � � �C� � � � � r r/ c �@ � e Zd ZdZedee fd�� � Zd� ZdS )�RevisiumCSVFile�csv�valuec �P � | r | � � � }|dv r|dk S n| �dS | S )N)�true�falser; � )�lower)r9 � low_values r �_convert_booleanz RevisiumCSVFile._convert_booleanJ sB � � � � � � � � �I�� -�-�-� �F�*�*� .� �]��2�� r c �� � � � j � d� � 5 }d� |D � � }t j |g d��� � }� fd�|D � � cddd� � S # 1 swxY w Y dS )z� Reads a CSV file and returns its contents as a list of dictionaries. The file must be read in binary mode and decoded manually to preserve the original line ending and carriage return characters. �rbc � � g | ];}|� t j � � t j � � � � � ��<S ))�errors)�decode�sys�getfilesystemencoding�getfilesystemencodeerrors)�.0�lines r � <listcomp>z(RevisiumCSVFile.read.<locals>.<listcomp>\ sU � � � � � � � � ��-�/�/� �8�:�:� � � �� � r )r4 �e�d�s�t�r)� fieldnamesc �P �� g | ]"}�fd �|� � � D � � ��#S )c �F �� i | ] \ }}|�|�� |� � �� S r )r@ )rI �fieldr9 r s �r � <dictcomp>z3RevisiumCSVFile.read.<locals>.<listcomp>.<dictcomp>g s= �� � � � �$��u� �(� �4�0�0��7�7�(�(�(r )�items)rI �rowr s �r rK z(RevisiumCSVFile.read.<locals>.<listcomp>f sT �� � � � � � � � � �(+� � � � �� � �� � r N)r r2 r8 � DictReader)r r4 �contents�readers ` r r5 zRevisiumCSVFile.readT s� �� � �Z�_�_�T� "� "� �a�� � � � � �H� �^��%C�%C�%C�� � �F�� � � � "� � � � � � � � � � � � � � � ���� � � � � � s �2A�A �"A N) r) r* r+ r � staticmethodr r r@ r5 r- r r r7 r7 G sQ � � � � � � �C����� � � � � �\��� � � � r r7 )�__doc__r8 r0 r rF r � contextlibr �pathlibr �typingr � defence360agent.contracts.configr �defence360agent.utilsr r r/ r7 r- r r �<module>rb s ��� �* � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � 1� 1� 1� 1� 1� 1� ,� ,� ,� ,� ,� ,�3� 3� 3� 3� 3� 3� 3� 3�8 � � � � �'� � � �&� &� &� &� &�&� &� &� &� &� &r