File: //lib/python3.6/site-packages/cloudinit/distros/package_management/__pycache__/apt.cpython-36.pyc
3
\�me� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZmZmZm Z m
Z
m Z m Z d dl
mZmZmZ d dlmZmZ d dlmZ eje�ZdZd d d
dgZdZe
e ee d�dd�Z G dd� de�Z dS )� N)�Any�Iterable�List�Mapping�Optional�Sequence�cast)�helpers�subp�util)�PackageManager�UninstalledPackages)� PER_INSTANCE�apt-get�(--option=Dpkg::Options::=--force-confold�*--option=Dpkg::options::=--force-unsafe-io� --assume-yes�--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| j d�}| j d�}t|t�r6|g}nt|t�sHtd��tj|�stt|�j� dkr�|r�tj |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
Zwhichr r )r r r � r �/usr/lib/python3.6/apt.py�get_apt_wrapper( s
"r" c s� e Zd ZdZf ddd�ejee eee ee d�� fdd�Z e
eje d d�dd ��Z d
d � Z
ejd d
�dd� �Zee d�dd�Zeed�dd�Zd dd�Zd dd�Zefdd�Z� ZS ) �AptZaptN)�apt_get_wrapper_command�apt_get_command�apt_get_upgrade_subcommand)�runnerr$ r% r& c sV t � j|� |d krt}|d kr$d}t|�t|� | _|| _tjj� | _ d| j d< d S )Nz dist-upgradeZnoninteractiveZDEBIAN_FRONTEND)
�super�__init__�APT_GET_COMMAND�tuple� apt_commandr& �os�environ�copy� environment)�selfr' r$ r% r& )� __class__r r! r) L s
z Apt.__init__)r' r r c C s& t |t|jd��|jd�|jd�d�S )NZapt_get_wrapperr% r& )r$ r% r& )r# r" r )�clsr' r r r r! � from_configa s
zApt.from_configc C s | j jd| jdgtd� d S )Nzupdate-sources�update)Zfreq)r' Zrun�run_package_commandr )r1 r r r! �update_package_sourcesl s
zApt.update_package_sources� )�maxsizec C s t j ddg�j}t|j� �S )Nz apt-cacheZpkgnames)r
�stdout�set�
splitlines)r1 Zrespr r r! �get_all_packagest s zApt.get_all_packages)�pkglistc s � fdd�|D �S )Nc s g | ]}|� j � kr|�qS r )r= )�.0Zpkg)r1 r r! �
<listcomp> s z0Apt.get_unavailable_packages.<locals>.<listcomp>r )r1 r> r )r1 r! �get_unavailable_packages~ s z Apt.get_unavailable_packages)r> r c s` | j � tjdt|��}| jdd� |D ��� tjd� � � fdd�|D �}|r\| jd|d� � S )Nz%s=%sc S s g | ]}|j d �d �qS )�=r )�split)r? �xr r r! r@ � 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 )r? �p)� unavailabler r! r@ � s Zinstall)�pkgs)r7 r �expand_package_listr rA �LOG�debugr6 )r1 r> Z
to_installr )rF r! �install_packages� s zApt.install_packagesc C s� |d kr g }t | j�}|r0t|t�r0|j|� n|rHt|t �rH|j|� |dkrV| j}|j|� tjd|�}|j|� | j ||| j
dd�d� d S )NZupgradez%s=%sF)�args�envZcapture)� short_cmd� subp_kwargs) r r, r r �append�extendr& r rH �_wait_for_apt_commandr0 )r1 r rL rG Z full_commandr> r r r! r6 � s"
zApt.run_package_commandc C sp |dkr t }x^|D ]V}tjj|�s$qt|d��4}ytj|tjtjB � W n t k
r\ dS X W dQ R X qW dS )ztDetermines if another process holds any apt locks.
If all locks are clear, return True else False.
N�wFT)
�APT_LOCK_FILESr- �path�exists�open�fcntl�lockfZLOCK_EXZLOCK_NB�OSError)r1 Z
lock_files�lockZhandler r r! �_apt_lock_available� s
zApt._apt_lock_availablec C s� t j � }tjd� x�t j � | |k r�| j� s8t jd� qtjd� y2d|� ddj|d �� d�}tjtj|tj|d �S tj k
r� | j� r�� tjd
� t jd� Y qX qW t
d ��d S )
z�Wait for apt install to complete.
short_cmd: Name of command like "upgrade" or "install"
subp_kwargs: kwargs to pass to subp
zWaiting for APT lockr8 zAPT lock availablezapt-z [� rL �])Zlogfunc�msg�func�kwargsz*Another process holds APT lock. Waiting...zCould not get APT lockN) �timerI rJ r\ Zsleep�joinr Zlog_timer
ZProcessExecutionError� TimeoutError)r1 rN rO ZtimeoutZ
start_timeZlog_msgr r r! rR � s(
zApt._wait_for_apt_command)NN)N)�__name__�
__module__� __qualname__�namer ZRunnersr r r r) � classmethodr r4 r7 � functools� lru_cacher= r rA r
rK r6 r\ �APT_LOCK_WAIT_TIMEOUTrR �
__classcell__r r )r2 r! r# I s 0
r# )r r r r r ) rX rj Zloggingr- rb Ztypingr r r r r r r Z cloudinitr r
r Z4cloudinit.distros.package_management.package_managerr r
Zcloudinit.settingsr Z getLoggerre rI r* rT rl �dictr r" r# r r r r! �<module> s* $
!