OwlCyberSecurity - MANAGER
Edit File: apt_btrfs_snapshot.cpython-38.pyc
U ٪zc�& � @ s� d dl mZmZ d dlZd dlZd dlZd dlZd dlZd dlZd dl m Z d dl m Z G dd� de �ZG dd� de�ZG d d � d e�ZG d d � d e�ZG d d� de�ZG dd� de�ZG dd� de�ZdS )� )�print_function�unicode_literalsN)�find_executable)�gettextc @ s e Zd ZdS )�AptBtrfsSnapshotErrorN��__name__� __module__� __qualname__� r r �@/usr/lib/python3/dist-packages/DistUpgrade/apt_btrfs_snapshot.pyr ! s r c @ s e Zd ZdS )�AptBtrfsNotSupportedErrorNr r r r r r % s r c @ s e Zd ZdS )� AptBtrfsRootWithNoatimeErrorNr r r r r r ) s r c @ s. e Zd ZdZedd� �Zd dd�Zdd� Zd S ) � FstabEntryz a single fstab entry line c C s" |� d�d �� }t|dd� � S )N�#r � )� partition�splitr )�cls�line�argsr r r � from_line/ s zFstabEntry.from_liner c C s( || _ || _|| _|| _|| _|| _d S �N��fs_spec� mountpoint�fstype�options�dump�passno)�selfr r r r r r r r r �__init__7 s zFstabEntry.__init__c C s d| j | j| j| j| j| jf S )Nz*<FstabEntry '%s' '%s' '%s' '%s' '%s' '%s'>r �r r r r �__repr__@ s �zFstabEntry.__repr__N)r r )r r r �__doc__� classmethodr r! r# r r r r r - s r c s" e Zd ZdZd� fdd� Z� ZS )�Fstabz a list of FstabEntry items � /etc/fstabc s� t t| ��� t|��`}dd� |D �D ]J}|dks&|�d�r>q&zt�|�}W n tk rd Y q&Y nX | �|� q&W 5 Q R X d S )Nc s s | ]}|� � V qd S r )�strip)�.0�lr r r � <genexpr>L s z!Fstab.__init__.<locals>.<genexpr>� r ) �superr& r! �open� startswithr r � ValueError�append)r �fstabZ fstab_filer �entry�� __class__r r r! H s zFstab.__init__)r' )r r r r$ r! � __classcell__r r r4 r r& F s r& c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �LowLevelCommandszi lowlevel commands invoked to perform various tasks like interact with mount and btrfs tools c C s t �d||g�}|dkS )N�mountr �� subprocessZcall)r r r �retr r r r8 Z s zLowLevelCommands.mountc C s t �d|g�}|dkS )N�umountr r9 )r r r; r r r r< ^ s zLowLevelCommands.umountc C s t �ddd||g�}|dkS )N�btrfs� subvolume�snapshotr r9 )r �source�destr; r r r �btrfs_subvolume_snapshotb s �z)LowLevelCommands.btrfs_subvolume_snapshotc C s t �ddd|g�}|dkS )Nr= r> �deleter r9 )r r? r; r r r �btrfs_delete_snapshotg s z&LowLevelCommands.btrfs_delete_snapshotN)r r r r$ r8 r< rB rD r r r r r7 V s r7 c @ s� e Zd ZdZdZed Zd)dd�Zdd� Zd d � Zd*d d �Z d d� Z dd� Z dd� Z d+dd�Z d,dd�Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd-d$d%�Zd&d'� Zd(S ).�AptBtrfsSnapshotz? the high level object that interacts with the snapshot system z@apt-snapshot-z old-root-r' c C s t |�| _t� | _d | _d S r )r& r2 r7 �commands�_btrfs_root_mountpoint)r r2 r r r r! t s zAptBtrfsSnapshot.__init__c C s t d�s dS | �� }|dk S )zx verify that the system supports apt btrfs snapshots by checking if the right fs layout is used etc r= FN)r �%_get_supported_btrfs_root_fstab_entry�r r3 r r r �snapshots_supportedy s z$AptBtrfsSnapshot.snapshots_supportedc C s6 | j D ]*}|jdkr|jdkrd|jkr| S qdS )z4 return the supported btrfs root FstabEntry or None �/r= zsubvol=@N)r2 r r r rI r r r rH � s ��� z6AptBtrfsSnapshot._get_supported_btrfs_root_fstab_entryc C s$ | j D ]}|j|kr|j S qdS )z4 return the device or UUID for the given mountpoint N)r2 r r )r r r2 r3 r r r �_uuid_for_mountpoint� s z%AptBtrfsSnapshot._uuid_for_mountpointc C s4 | � d�}tjdd�}| j�||�s(d S || _| jS )NrK zapt-btrfs-snapshot-mp-)�prefix)rL �tempfileZmkdtemprF r8 rG )r Zuuidr r r r �mount_btrfs_root_volume� s z(AptBtrfsSnapshot.mount_btrfs_root_volumec C s$ | j �| j�}t�| j� d | _|S r )rF r< rG �os�rmdir)r �resr r r �umount_btrfs_root_volume� s z)AptBtrfsSnapshot.umount_btrfs_root_volumec C s t j �� jdd��td��S )Nr )Z microsecond�_)�datetime�now�replaceZ isoformat�strr" r r r � _get_now_str� s �z AptBtrfsSnapshot._get_now_strr, c C sv | � � }| �� }tj�|d�}tj�|| j| | �}tj�|�rXttd�|f � dS | j � ||�}| � � |S d S )N�@zDINFO: snapshot directory '%s' already exists, not creating duplicateT) rO rY rP �path�join� SNAP_PREFIX�exists�printrT rF rB rS )r Zadditional_prefix�mpZsnap_idr@ �targetrR r r r �create_btrfs_root_snapshot� s � �z+AptBtrfsSnapshot.create_btrfs_root_snapshotr c C s� g }|dkr.| � � }|s t� �d|jkr.t� �|dkr>t�� }| �� }t�|�D ]<}|�| j �rPtj � tj � ||dd��}||k rP|� |� qP| �� |S )z� get the list of available snapshot If "older_then" is given (in unixtime format) it will only include snapshots that are older then the given date) r Znoatime�etcr2 )rH r r r �timerO rP �listdirr/ r] r[ �getatimer\ r1 rS )r � older_than�resultr3 r` �eZatimer r r � get_btrfs_root_snapshots_list� s" z.AptBtrfsSnapshot.get_btrfs_root_snapshots_listc C s t d� t d�| �� �� dS )NzAvailable snapshots:� T)r_ r\ rj r" r r r �print_btrfs_root_snapshots� s z+AptBtrfsSnapshot.print_btrfs_root_snapshotsc C s> t � � }|�d�std��t|d d� �}||d d d S )N�dz&Please specify time in days (e.g. 10d)���� �< )rd �endswith� Exception�int)r �timefmtrV Zdaysr r r � _parse_older_than_to_unixtime� s z.AptBtrfsSnapshot._parse_older_than_to_unixtimec C sX | � |�}z&td| � td�| j|d��� W n" tk rR tj�d� Y dS X dS )Nz$Available snapshots older than '%s':rk �rg �6Error: fstab option 'noatime' incompatible with optionFT)ru r_ r\ rj r �sys�stderr�write)r rt �older_than_unixtimer r r �%print_btrfs_root_snapshots_older_than� s � z6AptBtrfsSnapshot.print_btrfs_root_snapshots_older_thanc C sZ d}| � |�}z$| j|d�D ]}|| �|�M }q W n" tk rT tj�d� Y dS X |S )NTrv rw F)ru rj �delete_snapshotr rx ry rz )r rt rR r{ Zsnapr r r �%clean_btrfs_root_snapshots_older_than� s � z6AptBtrfsSnapshot.clean_btrfs_root_snapshots_older_thanc C s | � |�}|S r )� set_default)r � snapshot_namerR r r r �command_set_default� s z$AptBtrfsSnapshot.command_set_defaultTc C s� | � � }tj�||�}tj�|�r�|�d�r�|dkr�tj�|d�}tj�|| j| �� �}t�||� t�||� t d| � nt d� | � � dS )z set new default rZ z@Default changed to %s, please reboot for changes to take effect.z_You have selected an invalid snapshot. Please make sure that it exists, and that it is not "@".T) rO rP r[ r\ �isdirr/ � BACKUP_PREFIXrY �renamer_ rS )r r� Zbackupr` Znew_rootZ default_rootr r r r � s$ ��� �z AptBtrfsSnapshot.set_defaultc C s* | � � }| j�tj�||��}| �� |S r )rO rF rD rP r[ r\ rS )r r� r` rR r r r r} s �z AptBtrfsSnapshot.delete_snapshotN)r' )r' )r, )r )T)r r r r$ r] r� r! rJ rH rL rO rS rY rb rj rl ru r| r~ r� r r} r r r r rE l s& rE )Z __future__r r rU rP r: rx rd rN Zdistutils.spawnr r rT rr r r r �objectr �listr& r7 rE r r r r �<module> s