File: //usr/lib/python3/dist-packages/cloudinit/distros/package_management/__pycache__/apt.cpython-38.pyc
U
Ӈg� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZmZm Z m
Z
m Z m Z m
Z
d dlmZmZmZ d dlmZmZ d dlmZmZ e�e�ZdZddd d
gZ d Z e e e e d �d
d�Z G dd� de�Z!dS )� N)�Any�Iterable�List�Mapping�Optional�Sequence�cast)�helpers�subp�util)�PackageManager�UninstalledPackages)�
PER_ALWAYS� PER_INSTANCE)zapt-getz(--option=Dpkg::Options::=--force-confoldz*--option=Dpkg::options::=--force-unsafe-ioz --assume-yesz--quietz/var/lib/dpkg/lock-frontendz/var/lib/dpkg/lockz /var/cache/apt/archives/lockz/var/lib/apt/lists/lock� )�cfg�returnc C s� | sd}dg}n8| � d�}| � d�}t|t�r6|g}nt|t�sHtd��t�|�stt|��� dkr�|r�t� |d �r�t
t t |�S g S dS )a9 Parse the 'apt_get_wrapper' section of cloud-config.
apt_get_wrapper may be defined in cloud-config:
apt_get_wrapper:
enabled: true
command: ["eatmydata"]
The function takes the value of "apt_get_wrapper" and returns the list
of arguments to prefix to the apt-get command.
�autoZ eatmydata�enabled�commandz,apt_wrapper command must be a string or listr N) �get�
isinstance�str�list� TypeErrorr Zis_true�lowerr
�whichr r )r r r � r �J/usr/lib/python3/dist-packages/cloudinit/distros/package_management/apt.py�get_apt_wrapper) s$
�� �r c s� e Zd ZdZdddd�ejee eee ee d�� fdd�Z e
eje d d�d d
��Z e
d �d d
�Zdd�dd�Zejdd�dd� �Zee d�dd�Zeed�dd�Zd"d d �Zd d � Zefd d!�Z� ZS )#�AptZaptr N��apt_get_wrapper_command�apt_get_command�apt_get_upgrade_subcommand)�runnerr" r# r$ c sN t � �|� |d krt| _|d kr&d}t|�t| j� | _|| _ddi| _d S )Nz dist-upgradeZDEBIAN_FRONTENDZnoninteractive)�super�__init__�APT_GET_COMMANDr# �tuple� apt_commandr$ � environment)�selfr% r" r# r$ �� __class__r r r' M s �z Apt.__init__)r% r r c C s& t |t|�d��|�d�|�d�d�S )NZapt_get_wrapperr# r$ r! )r r r )�clsr% r r r r � from_configa s ��zApt.from_config)r c C s t t�| jd ��S )Nr )�boolr
r r# �r, r r r � availablel s z
Apt.availableF)�forcec C s$ | j jd| jdg|rtntd� d S )Nzupdate-sources�update)Zfreq)r% �run�run_package_commandr r )r, r4 r r r �update_package_sourceso s
�zApt.update_package_sources� )�maxsizec C s t � ddg�j}t|�� �S )Nz apt-cacheZpkgnames)r
�stdout�set�
splitlines)r, Zrespr r r �get_all_packagesw s zApt.get_all_packages)�pkglistc s � fdd�|D �S )Nc s. g | ]&}t �d |�d �d�� �� kr|�qS )z/|=r z-^)�re�split�rstripr> )�.0�pkgr2 r r �
<listcomp>� s
��z0Apt.get_unavailable_packages.<locals>.<listcomp>r )r, r? r r2 r �get_unavailable_packages� s
�z Apt.get_unavailable_packages)r? r c sd | � � t�dt|��}| �dd� |D ��� � r<t�d� � � fdd�|D �}|r`| jd|d� � S )N�%s=%sc S s g | ]}|� d �d �qS )�=r )rA )rC �xr r r rE � s z(Apt.install_packages.<locals>.<listcomp>zXThe following packages were not found by APT so APT will not attempt to install them: %sc s g | ]}|� kr|�qS r r )rC �p�Z unavailabler r rE � s �install)�pkgs)r8 r �expand_package_listr rF �LOG�debugr7 )r, r? Z
to_installr rK r �install_packages� s ��zApt.install_packagesc C s� |d kr g }t | j�}|r0t|t�r0|�|� n|rHt|t �rH|�|� |dkrV| j}|�|� t�d|�}|�|� | j || j
dd�d� d S )NZupgraderG F)�argsZ
update_envZcapture)� subp_kwargs) r r* r r �append�extendr$ r rN �_wait_for_apt_commandr+ )r, r rR rM Z full_commandr? r r r r7 � s$
��zApt.run_package_commandc
C sn t D ]d}tj�|�sqt|d��B}zt�|tjtjB � W n" t k
r\ Y W 5 Q R � dS X W 5 Q R X qdS )ztDetermines if another process holds any apt locks.
If all locks are clear, return True else False.
�wFT)
�APT_LOCK_FILES�os�path�exists�open�fcntl�lockfZLOCK_EXZLOCK_NB�OSError)r, �lockZhandler r r �_apt_lock_available� s zApt._apt_lock_availablec C s� t �� }t�d� t �� | |k r�| �� s6t �d� qt�d� ztjf |�W S tjk
r� | �� rj� t�d� t �d� Y qX qtd��dS )zWWait for apt install to complete.
subp_kwargs: kwargs to pass to subp
zWaiting for APT lockr9 zAPT lock availablez*Another process holds APT lock. Waiting...zCould not get APT lockN) �time� monotonicrO rP ra �sleepr
ZProcessExecutionError� TimeoutError)r, rS ZtimeoutZ
start_timer r r rV � s
zApt._wait_for_apt_command)NN) �__name__�
__module__� __qualname__�namer ZRunnersr r r r' � classmethodr r0 r1 r3 r8 � functools� lru_cacher> r rF r
rQ r7 ra �APT_LOCK_WAIT_TIMEOUTrV �
__classcell__r r r- r r J s, �
�
�r )"r] rk ZloggingrY r@ rb �typingr r r r r r r Z cloudinitr r
r Z4cloudinit.distros.package_management.package_managerr r
Zcloudinit.settingsr r Z getLoggerrf rO r( rX rm �dictr r r r r r r �<module> s& $
�!