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 ejeZdZd d d
dgZdZe
e ee dddZ G dd deZ dS ) N)AnyIterableListMappingOptionalSequencecast)helperssubputil)PackageManagerUninstalledPackages) PER_INSTANCEapt-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 )cfgreturnc C s | sd}dg}n8| j d}| j d}t|tr6|g}nt|tsHtdtj|stt|j dkr|rtj |d rt
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 eatmydataenabledcommandz,apt_wrapper command must be a string or listr N) get
isinstancestrlist TypeErrorr Zis_truelowerr
Zwhichr r )r r r r /usr/lib/python3.6/apt.pyget_apt_wrapper( s
"r" c s e Zd ZdZf dddejee eee ee d fddZ e
eje d ddd Z d
d Z
ejd d
dd Zee dddZeedddZd ddZd ddZefddZ ZS ) AptZaptN)apt_get_wrapper_commandapt_get_commandapt_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_COMMANDtuple apt_commandr& osenvironcopy environment)selfr' r$ r% r& ) __class__r r! r) L s
z Apt.__init__)r' r r c C s& t |t|jd|jd|jddS )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-sourcesupdate)Zfreq)r' Zrunrun_package_commandr )r1 r r r! update_package_sourcesl s
zApt.update_package_sources )maxsizec C s t j ddgj}t|j S )Nz apt-cacheZpkgnames)r
stdoutset
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!
s z0Apt.get_unavailable_packages..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..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 LOGdebugr6 )r1 r> Z
to_installr )rF r! install_packages s zApt.install_packagesc C s |d kr g }t | j}|r0t|tr0|j| n|rHt|t rH|j| |dkrV| j}|j| tjd|}|j| | j ||| j
ddd d S )NZupgradez%s=%sF)argsenvZcapture) short_cmd subp_kwargs) r r, r r appendextendr& 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|d4}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.
NwFT)
APT_LOCK_FILESr- pathexistsopenfcntllockfZLOCK_EXZLOCK_NBOSError)r1 Z
lock_fileslockZhandler r r! _apt_lock_available s
zApt._apt_lock_availablec C s t j }tjd xt 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 )
zWait 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 ])Zlogfuncmsgfunckwargsz*Another process holds APT lock. Waiting...zCould not get APT lockN) timerI rJ r\ Zsleepjoinr 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! s* $
!