Current File : //lib/python3.9/site-packages/cloudinit/distros/package_management/__pycache__/apt.cpython-39.pyc
a
'�Dg� � @ 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�ZdZg d�Z dZ e e e e d �d
d �Z G d d
� d
e�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--quiet)z/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 �L/usr/lib/python3.9/site-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 u rt| _|d u r&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�� �� vr|�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 � � 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 | ]}|� vr|�qS r r )rC �p�Z unavailabler r rE � rJ �install)�pkgs)r8 r �expand_package_listr rF �LOG�debugr7 )r, r? Z
to_installr rL r �install_packages� s ��zApt.install_packagesc C s� |d u r 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 rO �_wait_for_apt_commandr+ )r, r rS rN Z full_commandr? r r r r7 � s$
��zApt.run_package_commandc
C s� t D ]x}tj�|�sqt|d��L}zt�|tjtjB � W n" t y\ Y W d � dS 0 W d � q1 sr0 Y qdS )ztDetermines if another process holds any apt locks.
If all locks are clear, return True else False.
�wNFT)
�APT_LOCK_FILES�os�path�exists�open�fcntl�lockfZLOCK_EXZLOCK_NB�OSError)r, �lockZhandler r r �_apt_lock_available� s 6zApt._apt_lock_availablec C s� t �� }t�d� t �� | |k r�| �� s6t �d� qt�d� ztjf i |��W S tjy� | �� rl� t�d� t �d� Y q0 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� monotonicrP rQ rb �sleepr
ZProcessExecutionError� TimeoutError)r, rT �timeoutZ
start_timer r r rW � 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
rR r7 rb �APT_LOCK_WAIT_TIMEOUTrW �
__classcell__r r r- r r J s, �
�
�r )"r^ rm ZloggingrZ r@ rc �typingr r r r r r r Z cloudinitr r
r Z4cloudinit.distros.package_management.package_managerr r
Zcloudinit.settingsr r Z getLoggerrh rP r( rY ro �dictr r r r r r r �<module> s $
!
Mini Shell