3
gt`*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| j| d | _d | _d S )N)superr __init__optsparser)selfcli) __class__ /usr/lib/python3.6/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 d d d d!gg 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--urlprotocolsappendhttphttpsZrsyncftpz4when 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
j | jj sN| jj
rZ| j j
j | jjrr| jj| j j_ntjjtj | 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 j r.| j j r.| j j r.| j| j j}nXg }| j jrN|j| j| j j | j jrj|j| j| j j | j jr|j| j | j j | j j
rxd|D ]\}|j t j
kr|j| j jd}|rt| qtd|j }| jjjrtjj|tj| qW dS | j| 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^ s.
zDownloadCommand.runc C s i }x"|D ]}|j t|g j| q
W g }g }xP|j D ]D}dd |D }|r`|j|d q:|jdd d |j|d q:W |r| jj|| jjj |rx^|D ]V}|j }t
j j | jj
jt
j j|} t
j j| rt
j j|| rqtj|| jj
j qW tdd || D }
|
S )z=
Perform the download for a list of packages
c S s g | ]}|j tjkr|qS r )r7 r8 r9 ).0rA r r r
s z1DownloadCommand._do_downloads..r c S s | j j| j jfS )N)ZrepoZpriorityZcost)xr r r s z/DownloadCommand._do_downloads..)keyc S s g | ] }|j qS r )localPkg)rD rA r r r rE s )
setdefaultstrr valuessortr& Zdownload_packagesoutputprogressrI r. pathjoinr* r) basenameexistssamefileshutilcopysorted) r r@ Zpkg_dictrA Z to_downloadZcmdlineZpkg_listZ pkgs_cmdlinesrcdstZ locationsr r r r? s.
z DownloadCommand._do_downloadsc C s" | j jr| j|}n
| j|}|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_specsr@ r r r r2 s
z"DownloadCommand._get_pkg_objs_rpmsc C s* | j |}| j|}t| j|dd}|S )zj
Return a list of dnf.Package objects that represent the source
rpms to download.
T)r% )r2 _get_source_packagessetr[ )r r\ r@ source_pkgsr r r r4 s
z$DownloadCommand._get_pkg_objs_sourcec C s t }| jjj j }xh| j|D ]Z}xT|j|jgD ]D}|j|t |j
|j |j |j
d}|s^q4x|D ]}|j| qdW P q4W q"W |S )zm
Return a list of dnf.Package objects that represent the debuginfo
rpms to download.
)r; epochversionreleasearch)r^ r& sackquery availabler[ Z
debug_nameZsource_debug_namefilterintr` ra rb rc add)r r\ dbg_pkgsqrA Zdbg_name
dbg_availablepr r r r5 s
z'DownloadCommand._get_pkg_objs_debuginfoc C sh t }| jjj j }xL| j|D ]>}|j|jt|j |j
|j |j d}x|D ]}|j
| qNW q"W |S )zo
Return a list of dnf.Package objects that represent the debugsource
rpms to download.
)r; r` ra rb rc )r^ r& rd re rf r[ rg Zdebugsource_namerh r` ra rb rc ri )r r\ rj rk rA rl rm r r r r6 s
z)DownloadCommand._get_pkg_objs_debugsourceFc C s |r
| j n| j}g }x||D ]t}y|j|| W q tjjk
r } z list itertoolschain)r r\ r% funcZqueriespkg_specer@ r r r r[ s
"z DownloadCommand._get_packagesc C s | j |}t|}x|D ]}tj| jj}|j| |j }|r^|j|j |j|j
qt dg}t j
dj|dd |D t j
tjj|j tjj qW |S )z-Get packages matching pkg_specs and the deps.z Error in resolve of packages:z
c S s g | ] }t |qS r )rK )rD rA r r r rE s z;DownloadCommand._get_packages_with_deps..)r[ r^ r8 ZGoalr& rd ZinstallrC updateZ
list_installsZ
list_upgradesr r rq rQ r+ utilZ_format_resolve_problemsZ
problem_rulesr= r> ) r r\ r% r@ Zpkg_setrA ZgoalZrcrB r r r rZ s
z'DownloadCommand._get_packages_with_depsc C sz t }xj| D ]b}|jr8|j|j tjdt||j q |jdkrZ|jd|j|jf q tj t
dt| q W t |S )z4Get list of source rpm names for a list of packages.z --> Package : %s Source : %srX z
%s-%s.src.rpmz No source rpm defined for %s) r^ Z sourcerpmri r debugrK rc r; Zevrinfor rr )r@ r_ rA r r r r] s
z$DownloadCommand._get_source_packagesc C s t jjj|d }|o |dk}|s8|jdrdtjj|rd| jj|g| jj j
d}| jj j j
|dS t jj|}|j| jj | jjd }|j }|j
d
d }| jjr|j| jjd }t|j dkrtd
| }t jj||S )z#Return a query to match a pkg_spec.r r r filer z.rpm)rO )rA )Zwith_srcT)Zlatest_per_arch_by_priority)rc zNo package %s available.)r r r| r )r+ ZpycompZurlparseendswithr. rP isfiler& Zadd_remote_rpmsrN rO rd re ZfiltermsubjectSubjectZget_best_queryr
r% rf r rg lenrC r r= rp )r rv r1 Zis_urlr@ subjrk rB r r r ro s zDownloadCommand._get_queryc C sd |dd }t jj|}x.|j D ]"}|j| jjj }|r"|j S q"W t d| }t j
j |dS )z/Return a query to match a source rpm file name.N zNo package %s available.) r+ r r Zget_nevra_possibilitiesZto_queryr& rd rf Zlatestr r= rp )r rv r Z nevra_objZ tmp_queryrB r r r rn , s z!DownloadCommand._get_query_source)F)F)__name__
__module__ __qualname__aliasesr Zsummaryr staticmethodr" r0 rC r? r2 r4 r5 r6 r[ rZ r] ro rn
__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.utilr8 rs r. rU ZpluginZregister_commandr
ZCommandr r r r r s