Current File : //opt/alt/python37/lib/python3.7/site-packages/lvestats/lib/__pycache__/snapshot.cpython-37.pyc
B
��1b� � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlm Z m
Z
d dl m Z d dl
Z
d dlmZ dZdZee�ZG dd � d e�Ze d
d � �ZdS ) � )�absolute_importN)�Optional�List)�contextmanager)�mapz/var/lve/snapshotsz .snapshotc @ s� e Zd ZdZd$dd�Zdd� Zedd� �Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd � Zed d � �Zed%d d!��Zed"d#� �Zd S )&�Snapshotad
This class is responsible for loading and saving snapshot files for the interval
The files will be saved in a format of: /var/lve/snapshots/[uid]/dump_time.snapshot
dump_time is the timestamp/integer
The directories /var/lve/snaphosts/[uid] and the dump files itself will be owned by user
They will not be readable by other users.
� c C s || _ || _t�d�| _d S )Nz lib-snapshot)�
compresslevel�incident�loggingZ getLogger�log)�selfr
r � r �F/opt/alt/python37/lib/python3.7/site-packages/lvestats/lib/snapshot.py�__init__! s zSnapshot.__init__c C sn |d }|d k st �| �|� t�t�|��� | j�}| �|��"}|� |� | j
� d|j � W d Q R X |j S )NZ dump_timezSnapshot dumped to file %s)
�AssertionError�_replace_unicode_data�zlib�compress�json�dumps�encoder � create_file�writer �debug�name)r
�data� dump_dateZjson_compressed�fr r r �save' s
z
Snapshot.savec C s t | t�r| �dd�S | S )Nzutf-8�replace)�
isinstance�bytes�decode)�objr r r � _to_unicode4 s
zSnapshot._to_unicodec C st g }x*|� dg �D ]}|�tt| j|��� qW ||d< g }x*|� dg �D ]}|�tt| j|��� qJW ||d< d S )NZsnap_sqlZ snap_http)�get�append�listr r% )r
r Z u_queriesZqueryZu_urlsZhttpr r r r : s z Snapshot._replace_unicode_datac C s" | � � }tj�|�r t�|�S g S )N)�get_dir�os�path�isdir�listdir)r
�dir_r r r �_get_file_listF s
zSnapshot._get_file_listc C s | � | �� ||�S )z�
Return ordered list of timestamps when snapshots for this use were created
:param from_ts:
:param to_ts:
:return: list of timesteamps, ordered for that period
)�snapshot_filterr/ )r
�from_ts�to_tsr r r � get_ts_listM s zSnapshot.get_ts_listc C s� g }x�| � ||�D ]�}y\| �| �|��}t�� sVt| jd �� | �|�}W dQ R X n
| �|�}|�t �
|�� W q t t t
fk
r� } z(| j�d| �| �|��t|�f � W dd}~X Y qX qW |S )z�
Get all snapshot for a period
:param from_ts
:param to_ts
:return: list of snapshots, ordered by date
�uidNzCorrupted file: %s (%s))r3 �
get_file_name�
ts_to_namer* �geteuid�drop_privilegesr
�read_file_contentr' r �loads�IOError�
ValueError�OSErrorr Zwarning�str)r
r1 r2 �result�ts�filename�content�ver r r �
get_snapshotsW s
:zSnapshot.get_snapshotsc C sz t |d��}|�� }W d Q R X yt�|�}W nD tjk
rp t�|| j�}t |d��}|�|� W d Q R X Y nX |�� S )N�rb�wb) �open�readr �
decompress�errorr r r r# )r
rA r rB Zcompressed_contentr r r r9 m s zSnapshot.read_file_contentc C s | � | jd | jd �S )zZ
Load all snapshots for given incident
:return: list of snapshots
Zincident_start_timeZincident_end_time)rD r
)r
r r r �get_incident_snapshotsy s z Snapshot.get_incident_snapshotsc C s t j�tt| jd ��S )Nr4 )r* r+ �join�
SNAPSHOT_PATHr> r
)r
r r r r) � s zSnapshot.get_dirc C s t j�| �� |�S )N)r* r+ rL r) )r
r r r r r5 � s zSnapshot.get_file_namec C s | � � }tj�|�s~y.t�|� t�|d� t�|| jd d� W n: tt fk
r| } z| j
� d|t |�� W dd}~X Y nX | �
| �|��}t| jd ��` t|d�}yt�|�� d� W n: tt fk
r� } z| j
� d|t |�� W dd}~X Y nX W dQ R X |S ) z�
create file, change its ownership & permissions if needed. Create directories if needed as well
:param dump_date: int timestamp used as file name
:return: open File object
i� r4 r z Unable to create dir %s (%s)NrF � z&Unable to set file permissions %s (%s))r) r* r+ �exists�makedirs�chmod�chownr
r; r= r rJ r> r5 r6 r8 rG �fchmod�fileno)r
r r. �e� file_name�file_r r r r � s
&
0zSnapshot.create_filec C sj | � � }t�|�}| �|�}| j||d�}||krBtj|dd� n$x"|D ]}t�| �| �|��� qHW dS )z�
Delete old snapshots. If there are no more
:param to_ts: up to which timestamp to remove snapshots
:return: None
)r2 T)�
ignore_errorsN) r) r* r- r0 �shutil�rmtree�remover5 r6 )r
r2 Z_dir�filesZ
all_snapshotsZ ts_to_remover@ r r r �
delete_old� s
zSnapshot.delete_oldc C sB | � t�r>| dt � }|�� r>yt|�S tk
r< Y nX d S )Nr )�endswith� SNAPSHOT_EXT�SNAPSHOT_EXT_LEN�isdigit�intr< )rW r@ r r r �get_ts� s
zSnapshot.get_tsNc C sh |d kr d}|d krt j}g }x@| D ]8}t�|�}|d k r$|| krN|kr$n q$|�|� q$W t|�S )Nr )�sys�maxsizer rc r' �sorted)r\ r1 r2 r? rA r@ r r r r0 � s
zSnapshot.snapshot_filterc C s t | �t S )N)r> r_ )r@ r r r r6 � s zSnapshot.ts_to_name)r )NN)�__name__�
__module__� __qualname__�__doc__r r � staticmethodr% r r/ r3 rD r9 rK r) r5 r r] rc r0 r6 r r r r r s$
r c c sx t �� t �� t �� }}}t�d�d }t �g � t �|� t �| � z
d V W d t �|� t �|� t �|� X d S )NZnobody� ) r* �getuid�getgid� getgroups�pwd�getpwnam� setgroups�setegid�seteuid)r4 Zold_uidZold_gidZ
old_groups�gidr r r r8 � s
r8 )Z
__future__r r r r* rd rY r �typingr r �
contextlibr rp �builtinsr rM r_ �lenr` �objectr r8 r r r r �<module> s ;
Mini Shell