File: //lib/python3/dist-packages/cloudinit/distros/package_management/__pycache__/apt.cpython-310.pyc
o
x[h� � @ 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 d e e d
e e fd d �Z G d
d� 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--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}n | � d�}| � d�}t|t�r|g}n t|t�s$td��t�|�s:t|��� dkrA|rAt� |d �rAt
t t |�S g S )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.
�auto� eatmydata�enabled�commandz,apt_wrapper command must be a string or listr ) �get�
isinstance�str�list� TypeErrorr �is_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�dejdee deee dee f� fd d
�Z e
dejd e d d fd
d��Z d e
fdd�Zdd�dd�Zejdd�dd� �Zdee fdd�Zded efd d �Zd$d d �Zd d!� Zefd"d#�Z� ZS )%�Apt�aptr N��apt_get_wrapper_command�apt_get_command�apt_get_upgrade_subcommand�runnerr% r& r' c sN t � �|� |d u r
t| _|d u rd}t|�t| j� | _|| _ddi| _d S )Nz dist-upgrade�DEBIAN_FRONTEND�noninteractive)�super�__init__�APT_GET_COMMANDr& �tuple� apt_commandr' � environment)�selfr( r% r& r' �� __class__r r r, M s �z Apt.__init__r r c C s& t |t|�d��|�d�|�d�d�S )N�apt_get_wrapperr& r' r$ )r" r! r )�clsr( r r r r � from_configa s ��zApt.from_configc C s t t�| jd ��S )Nr )�boolr
r r& �r1 r r r � availablel s z
Apt.availableF)�forcec C s$ | j jd| jdg|r tntd� d S )Nzupdate-sources�update)�freq)r( �run�run_package_commandr r )r1 r: r r r �update_package_sourceso s
�zApt.update_package_sources� )�maxsizec C s t � ddg�j}t|�� �S )Nz apt-cache�pkgnames)r
�stdout�set�
splitlines)r1 �respr 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�rstriprG )�.0�pkgr8 r r �
<listcomp>� s ��z0Apt.get_unavailable_packages.<locals>.<listcomp>r )r1 rH r r8 r �get_unavailable_packages� s
�z Apt.get_unavailable_packagesc sd | � � t�dt|��}| �dd� |D ��� � r t�d� � � fdd�|D �}|r0| jd|d� � S )N�%s=%sc S s g | ] }|� d �d �qS )�=r )rJ )rL �xr r r rN � 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 | ]}|� vr|�qS r r )rL �p�� unavailabler r rN � s �install)�pkgs)r? r �expand_package_listr rO �LOG�debugr> )r1 rH �
to_installr rT r �install_packages� s ��zApt.install_packagesc C s� |d u rg }t | j�}|rt|t�r|�|� n |r$t|t �r$|�|� |dkr+| j}|�|� t�d|�}|�|� | j || j
dd�d� d S )N�upgraderP F)�args�
update_env�capture)� subp_kwargs) r r/ r r �append�extendr' r rX �_wait_for_apt_commandr0 )r1 r r^ rW � full_commandrH r r r r> � s$
�
�zApt.run_package_commandc
C s� t D ]<}tj�|�s qt|d��&}z t�|tjtjB � W n t y. Y W d � dS w W d � n1 s9w 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�lockf�LOCK_EX�LOCK_NB�OSError)r1 �lock�handler r r �_apt_lock_available� s �� ��zApt._apt_lock_availablec C s� t �� }t�d� t �� | |k rK| �� st �d� q t�d� z tjdi |��W S tjyB | �� r6� t�d� t �d� Y nw t �� | |k std��)zWWait for apt install to complete.
subp_kwargs: kwargs to pass to subp
zWaiting for APT lockr@ zAPT lock availablez*Another process holds APT lock. Waiting...zCould not get APT lockNr ) �time� monotonicrY rZ rs �sleepr
�ProcessExecutionError� TimeoutError)r1 ra �timeout�
start_timer r r rd � s"
��zApt._wait_for_apt_command)NN) �__name__�
__module__� __qualname__�namer �Runnersr r r r, � classmethodr r6 r7 r9 r? � functools� lru_cacherG r rO r
r\ r> rs �APT_LOCK_WAIT_TIMEOUTrd �
__classcell__r r r2 r r" J s4 ���
��
�r" )"rl r� �loggingrh rI rt �typingr r r r r r r � cloudinitr r
r �4cloudinit.distros.package_management.package_managerr r
�cloudinit.settingsr r � getLoggerr{ rY r- rg r� �dictr r! r" r r r r �<module> s $
!