a
&c*0 @ s d dl mZ d dl mZ d dlmZmZ d dlmZ d dlZd dl Zd dl
Zd dl Zd dl Zd dl
Zd dlZd dlZd dlZd dlZejjG dd dejjZdS ) )absolute_import)unicode_literals)_logger) OptionParserNc s e Zd ZdgZedZ fddZedd Zdd Z d d
Z
d d Z d
d Z dd Z
dd Zdd Zd ddZd!ddZedd Zd d Zd d Z ZS )"DownloadCommandZdownloadz%Download package to current directoryc s t t| | d | _d | _d S N)superr __init__optsparser)selfcli __class__ 8/usr/lib/python3.9/site-packages/dnf-plugins/download.pyr
+ s zDownloadCommand.__init__c C s | j ddtdd | j ddtdd | j d dtd
d | j d dtd d | j d
ddg tjdtdd | j ddtdd | j ddtdd | j ddddtdd | j d d g d g td d d S )!Npackages+zpackages to download)nargshelpz--source
store_truez download the src.rpm instead)actionr z --debuginfoz'download the -debuginfo package insteadz
--debugsourcez)download the -debugsource package insteadz--archz
--archlistarchesz[arch]z5limit the query to packages of given architectures.)destdefaultr metavarr z --resolvez(resolve and download needed dependenciesz --alldepsz^when running with --resolve, download all dependencies (do not exclude already installed ones)z--urlz--urlsurlzJprint list of urls where the rpms can be downloaded instead of downloading)r r r z--urlprotocolsappend)httphttpsrsyncftpz4when running with --url, limit to specific protocols)r choicesr r ) add_argumentr r Z_SplitCallback)r r r r
set_argparser0 s< z DownloadCommand.set_argparserc C s | j j}d|_d|_| jjr*| jjr*d|_| jjr>| j j
| jj sN| jj
rZ| j j
| jjrr| jj| j j_ntjt | j j_d S )NTF)r demandsZsack_activationZavailable_reposr resolveZalldepsZload_system_reposourcebaseZreposZenable_source_repos debuginfo debugsourceZenable_debug_reposdestdirconfdnfi18nucdosgetcwd)r
r& r r r configureK s zDownloadCommand.configurec C s | j js(| j js(| j js(| | j j}nXg }| j jrH|| | j j | j jrd|| | j j | j jr|| | j j | j j
r|D ]\}|j t j
kr|j| j jd}|rt| qtd|j }| jjjrtj|t| qdS | | dS )z Execute the util action here.)schemesz$Failed to get mirror for package: %sN) r r( r* r+ _get_pkg_objs_rpmsr extend_get_pkg_objs_source_get_pkg_objs_debuginfo_get_pkg_objs_debugsourcer repoidhawkeyCMDLINE_REPO_NAMEZremote_locationZ urlprotocolsprintr namer) r- strictr.
exceptionsErrorr Zwarning
_do_downloads)r
pkgspkgr msgr r r run^ s2
zDownloadCommand.runc C s i }|D ]}| t|g | qg }g }| D ]D}dd |D }|rZ||d q4|jdd d ||d q4|r| j|| jjj |r|D ]V}| }t
j | jj
jt
j |} t
j | rt
j || rqt|| jj
j qtdd || D }
|
S )z=
Perform the download for a list of packages
c S s g | ]}|j tjkr|qS r )r: r; r< .0rD r r r
s z1DownloadCommand._do_downloads..r c S s | j j| j jfS r )ZrepopriorityZcost)xr r r z/DownloadCommand._do_downloads..)keyc S s g | ] }| qS r )localPkgrG r r r rI rM )
setdefaultstrr valuessortr) Zdownload_packagesoutputprogressrO r1 pathjoinr- r, basenameexistssamefileshutilcopysorted) r
rC Zpkg_dictrD Z to_downloadZcmdlineZpkg_listZ pkgs_cmdlinesrcdstZ locationsr r r rB s. z DownloadCommand._do_downloadsc C s" | j jr| |}n
| |}|S )zc
Return a list of dnf.Package objects that represent the rpms
to download.
)r r' _get_packages_with_deps
_get_packages)r
pkg_specsrC r r r r5 s
z"DownloadCommand._get_pkg_objs_rpmsc C s* | |}| |}t| j|dd}|S )zj
Return a list of dnf.Package objects that represent the source
rpms to download.
T)r( )r5 _get_source_packagessetra )r
rb rC source_pkgsr r r r7 s
z$DownloadCommand._get_pkg_objs_sourcec C sz t }| jj }| |D ]T}|j|jfD ]B}|j|t |j
|j |j |j
d}|sZq0|D ]}|| q^ q q0q |S )zm
Return a list of dnf.Package objects that represent the debuginfo
rpms to download.
r> epochversionreleasearch)rd r) sackquery availablera Z
debug_nameZsource_debug_namefilterintrg rh ri rj add)r
rb dbg_pkgsqrD Zdbg_name
dbg_availablepr r r r8 s" z'DownloadCommand._get_pkg_objs_debuginfoc C s` t }| jj }| |D ]:}|j|jt|j |j
|j |j d}|D ]}|
| qJq |S )zo
Return a list of dnf.Package objects that represent the debugsource
rpms to download.
rf )rd r) rk rl rm ra rn Zdebugsource_namero rg rh ri rj rp )r
rb rq rr rD rs rt r r r r9 s z)DownloadCommand._get_pkg_objs_debugsourceFc C s |r
| j n| j}g }|D ]v}z||| W q tjjy } zBttj | | j
j j rxtt
d tj|W Y d}~qd}~0 0 qttj| }|S )z Get packages matching pkg_specs.z Exiting due to strict setting.N)_get_query_source
_get_queryr r. r@ PackageNotFoundErrorr errorr/ r0 r) r- r? r rA list itertoolschain)r
rb r( funcZqueriespkg_specerC r r r ra s
$z DownloadCommand._get_packagesc C s | |}t|}|D ]}t| jj}|| | }|r\|| ||
qt dg}t
d|dd |D t
tj| tj q|S )z-Get packages matching pkg_specs and the deps.z Error in resolve of packages:z
c S s g | ] }t |qS r )rQ rG r r r rI rM z;DownloadCommand._get_packages_with_deps..)ra rd r; ZGoalr) rk installrF updateZ
list_installsZ
list_upgradesr r rx rW r. utilZ_format_resolve_problemsZ
problem_rulesr@ rA ) r
rb r( rC Zpkg_setrD ZgoalrcrE r r r r` s
z'DownloadCommand._get_packages_with_depsc C sv t }| D ]b}|jr6||j tdt||j q
|jdkrX|d|j|jf q
t t
dt| q
t |S )z4Get list of source rpm names for a list of packages.z --> Package : %s Source : %sr^ z
%s-%s.src.rpmz No source rpm defined for %s) rd Z sourcerpmrp r debugrQ rj r> Zevrinfor ry )rC re rD r r r rc s
z$DownloadCommand._get_source_packagesc C s t jj|d }|o |dv }|s8|drdtj|rd| jj|g| jj j
d}| jj j
|dS t j|}|j| jj | jjd}| }|j
dd}| jjr|j| jjd }t| dkrtd
| }t j||S ) z#Return a query to match a pkg_spec.r )r r" filer z.rpm)rU )rD )Zwith_srcT)Zlatest_per_arch_by_priority)rj No package %s available.)r. Zpycompurlparseendswithr1 rV isfiler) Zadd_remote_rpmsrT rU rk rl ZfiltermsubjectSubjectZget_best_queryr r( rm r rn lenrF r r@ rw )r
r} r4 Zis_urlrC subjrr rE r r r rv s zDownloadCommand._get_queryc C sd |dd }t j|}| D ]&}|| jj }|r | S q t d| }t j
|dS )z/Return a query to match a source rpm file name.Nr ) r. r r Zget_nevra_possibilitiesZto_queryr) rk rm Zlatestr r@ rw )r
r} r Z nevra_objZ tmp_queryrE r r r ru , s z!DownloadCommand._get_query_source)F)F)__name__
__module__ __qualname__aliasesr Zsummaryr
staticmethodr% r3 rF rB r5 r7 r8 r9 ra r` rc rv ru
__classcell__r r r r r % s$
#!
r )Z
__future__r r Zdnfpluginscorer r Zdnf.cli.option_parserr r. Zdnf.cliZdnf.exceptionsZdnf.i18nZ dnf.subjectZdnf.utilr; rz r1 r[ ZpluginZregister_commandr ZCommandr r r r r s