File: //lib/python3/dist-packages/DistUpgrade/__pycache__/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 ��
subprocess�call)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= rC rE 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 rJ r r r rI � 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 )NrL zapt-btrfs-snapshot-mp-)�prefix)rM �tempfileZmkdtemprG r8 rH )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 )rG r= rH �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) rP rZ rQ �path�join� SNAP_PREFIX�exists�printrU rG rC rT )r Zadditional_prefix�mpZsnap_idrA �targetrS 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 )rI r
r r �timerP rQ �listdirr/ r^ r\ �getatimer] r1 rT )r �
older_than�resultr3 ra �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] rk 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)���� �< )re �endswith� Exception�int)r �timefmtrW 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':rl �rh �6Error: fstab option 'noatime' incompatible with optionFT)rv r` r] rk r �sys�stderr�write)r ru �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 )NTrw rx F)rv rk �delete_snapshotr ry rz r{ )r ru rS 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_namerS 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 r[ 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) rP rQ r\ r] �isdirr/ �
BACKUP_PREFIXrZ �renamer` rT )r r� Zbackupra Znew_rootZ default_rootr r r r� � s$
��� �z AptBtrfsSnapshot.set_defaultc C s* | � � }| j�tj�||��}| �� |S r )rP rG rE rQ r\ r] rT )r r� ra rS r r r r~ s �z AptBtrfsSnapshot.delete_snapshotN)r' )r' )r, )r )T)r r r
r$ r^ r� r! rK rI rM rP rT rZ rc rk rm rv r} r r� r� r~ r r r r rF l s&
rF )Z
__future__r r rV rQ r: ry re rO Zdistutils.spawnr r rU rs r r
r �objectr �listr& r7 rF r r r r �<module> s