File manager

3

�gt`*0�@s�dd
lm
Z

ddlmZ
ddlmZmZ
ddlmZ
ddlZddl	Zddl
ZddlZddlZddl
ZddlZddlZddlZddlZejjGdd�dejj��
ZdS)�)
�absolute_import)
�unicode_literals)�
_�logger)
�OptionParserNcs�eZ
dZd
g
Zed�
Z�f
dd�Zedd��
Zdd�Z	d	d
�Z
dd�Zd
d�Zdd�Z
dd�Zdd�Zd dd�
Zd!dd�
Zedd��
Zdd�Zdd�Z�ZS)"�DownloadCommandZdownloadz%Download package to current directorycs tt
|�j|
�

d|_d|_dS)
N)�superr�__init__�opts�parser)�self�cli)
�	__class__��/usr/lib/python3.6/download.pyr	+s


zDownloadCommand.__init__c

	Cs�|jd
dt
d�
d�
|jddt
d�
d�
|jd	dt
d
�
d�
|jddt
d�
d�
|jd
ddgtjdt
d�
d�
|jddt
d�
d�
|jddt
d�
d�
|jddddt
d�
d�
|jddddd d!ggt
d"�
d#�
dS)$N�packages�
+zpackages to download)�nargs�helpz--source�
store_truezdownload the src.rpm instead)�actionrz--debuginfoz'download the -debuginfo package insteadz
--debugsourcez)download the -debugsource package insteadz--archz
--archlist�archesz[arch]z5limit  the  query to packages of given architectures.)�dest�defaultr�metavarrz	--resolvez(resolve and download needed dependenciesz	--alldepsz^when running with --resolve, download all dependencies (do not exclude already installed ones)z--urlz--urls�urlzJprint list of urls where the rpms can be downloaded instead of downloading)rrrz--urlprotocols�append�http�httpsZrsync�ftpz4when running with --url, limit to specific protocols)r�choicesrr)�add_argumentrrZ_SplitCallback)
rrrr�
set_argparser0s*


















zDownloadCommand.set_argparserc
Cs�|jj
}
d
|
_d
|
_|jjr*|jjr*d|
_|jjr>|j	j
j�
|jjsN|jj
rZ|j	j
j�
|jjrr|jj|j	j_ntjjtj��
|j	j_dS)NTF)r
�demandsZsack_activationZavailable_reposr
�resolveZalldepsZload_system_repo�source�baseZreposZenable_source_repos�	debuginfo�debugsourceZenable_debug_repos�destdir�conf�dnf�i18n�ucd�os�getcwd)rr#rrr�	configureKs






zDownloadCommand.configurec
Cs
|jj
r.|jjr.|jjr.|j|jj�
}
nXg}
|jj
rN|
j|j|jj�
�

|jjrj|
j|j|jj�
�

|jjr�|
j|j	|jj�
�

|jj
r�xd|
D]\}|jtj
kr�|j|jjd
�
}|r�t|�

q�td�
|j}|jjjr�tjj|�
�
tj|�

q�WdS|j|
�

dS)zExecute 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�repoid�hawkey�CMDLINE_REPO_NAMEZremote_locationZurlprotocols�printr�namer&r*�strictr+�
exceptions�ErrorrZwarning�
_do_downloads)r�pkgs�pkgr�msgrrr�run^s.




















zDownloadCommand.runcCs
i}x"|
D]}|jt
|�
g�j|�

q
Wg}g}xP|j�D]D}d
d�|D�
}|r`|j|d�

q:|jdd�d�

|j|d�

q:W|r�|jj||jjj�
|r�x^|D]V}|j	�}t
jj|jj
jt
jj|�
�}	t
jj|	�
r�t
jj||	�r�q�tj||jj
j�
q�Wtdd�||D�
�
}
|
S)z=
        Perform the download for a list of packages
        c
Ssg|]}
|
jt
jkr|
�qSr)r7r8r9)�.0rArrr�
<listcomp>�s
z1DownloadCommand._do_downloads.<locals>.<listcomp>r
c

Ss|jj
|jjfS)
N)ZrepoZpriorityZcost)
�
xrrr�<lambda>�sz/DownloadCommand._do_downloads.<locals>.<lambda>)
�keyc
Ssg|]}
|
j��qSr)
�localPkg)rDrArrrrE�s)�
setdefault�strr�values�sortr&Zdownload_packages�output�progressrIr.�path�joinr*r)�basename�exists�samefile�shutil�copy�sorted)rr@Zpkg_dictrAZto_downloadZcmdlineZpkg_listZpkgs_cmdline�src�dstZ	locationsrrrr?�s.



















zDownloadCommand._do_downloadscCs"|jj
r|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@rrrr2�s


z"DownloadCommand._get_pkg_objs_rpmscCs*|j|
�
}|j
|�
}t|j|d
d��
}|S)zj
        Return a list of dnf.Package objects that represent the source
        rpms to download.
        T)
r%)r2�_get_source_packages�setr[)rr\r@�source_pkgsrrrr4�s




z$DownloadCommand._get_pkg_objs_sourcec	Cs�t�}|j
jj�j�}xh|j|
�
D]Z}xT|j|jgD]D}|j|t	|j
�
|j|j|j
d
�}|s^q4x|D]}|j|�

qdWPq4Wq"W|S)zm
        Return a list of dnf.Package objects that represent the debuginfo
        rpms to download.
        )r;�epoch�version�release�arch)r^r&�sack�query�	availabler[Z
debug_nameZsource_debug_name�filter�intr`rarbrc�add)rr\�dbg_pkgs�
qrAZdbg_name�
dbg_available�
prrrr5�s 












z'DownloadCommand._get_pkg_objs_debuginfocCsht�}|j
jj�j�}xL|j|
�
D]>}|j|jt|j	�
|j
|j|jd
�}x|D]}|j
|�

qNWq"W|S)zo
        Return a list of dnf.Package objects that represent the debugsource
        rpms to download.
        )r;r`rarbrc)r^r&rdrerfr[rgZdebugsource_namerhr`rarbrcri)rr\rjrkrArlrmrrrr6�s









z)DownloadCommand._get_pkg_objs_debugsourceFcCs�|r
|jn|j
}g}x||
D]t}y|j||�
�

Wqtjjk
r�
}
z<tjtjj	|�
�

|j
jjr|tjt
d
�
�

tjj|�
�
WYdd}~XqXqWttj|��
}|S)z Get packages matching pkg_specs.zExiting due to strict setting.N)�_get_query_source�
_get_queryrr+r=�PackageNotFoundErrorr�errorr,r-r&r*r<rr>�list�	itertools�chain)rr\r%�funcZqueries�pkg_spec�
er@rrrr[�s










"
zDownloadCommand._get_packagesc	Cs�|j|
�
}t
|�
}x�|D]�}tj|jj�
}|j|�

|j�}|r^|j|j	��

|j|j
��

qtd
�
g
}tj
dj|dd�|D�
�
�

tj
tjj|j��
�

tjj��
qW|S)z-Get packages matching pkg_specs and the deps.zError in resolve of packages:z
    c
Ssg|]}
t|
�
�qSr)
rK)rDrArrrrE
sz;DownloadCommand._get_packages_with_deps.<locals>.<listcomp>)r[r^r8ZGoalr&rdZinstallrC�updateZ
list_installsZ
list_upgradesrrrqrQr+�utilZ_format_resolve_problemsZ
problem_rulesr=r>)	rr\r%r@Zpkg_setrAZgoalZrcrBrrrrZ�s















z'DownloadCommand._get_packages_with_depsc
Cszt�}
xj|D]b}|j
r8|
j|j
�

tjd
t|�
|j
�
q|jdkrZ|
jd|j|jf�

qtj	t
d�
t|�
�
qWt|
�
S)z4Get list of source rpm names for a list of packages.z  --> Package : %s Source : %srXz
%s-%s.src.rpmzNo source rpm defined for %s)r^Z	sourcerpmrir�debugrKrcr;Zevr�inforrr)r@r_rArrrr]
s









z$DownloadCommand._get_source_packagescCs�tj
jj|
�
d
}|o|dk}|s8|
jd�
rdtjj|
�
rd|jj|
g
|jj	j
d�}|jjj�j
|d�
Stjj|
�
}|j|jj|jjd	�}|j�}|j
d
d�
}|jjr�|j|jjd�
}t|j��
d
kr�td
�
|
}tjj|�
�
|S)z#Return a query to match a pkg_spec.r
rr�filerz.rpm)
rO)
rA)
Zwith_srcT)
Zlatest_per_arch_by_priority)
rczNo package %s available.)rrr|r)r+ZpycompZurlparse�endswithr.rP�isfiler&Zadd_remote_rpmsrNrOrdreZfilterm�subject�SubjectZget_best_queryr
r%rfrrg�lenrCrr=rp)rrvr1Zis_urlr@�subjrkrBrrrro
s













zDownloadCommand._get_querycCsd|
d
d�}
tj
j|
�
}x.|j�D]"}|j|jj�
j�}|r"|j�Sq"Wt	d�
|
}tj
j|�
�
d
S)z/Return a query to match a source rpm file name.N�zNo package %s available.���)r+rr�Zget_nevra_possibilitiesZto_queryr&rdrfZlatestrr=rp)rrvr�Z	nevra_objZ	tmp_queryrBrrrrn,
s





z!DownloadCommand._get_query_source)
F)
F)�__name__�
__module__�__qualname__�aliasesrZsummaryr	�staticmethodr"r0rCr?r2r4r5r6r[rZr]rorn�
__classcell__rr)
rrr%s 
#!


r)Z
__future__rrZdnfpluginscorerrZdnf.cli.option_parserrr+Zdnf.cliZdnf.exceptionsZdnf.i18nZdnf.subjectZdnf.utilr8rsr.rUZpluginZregister_commandr
ZCommandrrrrr�<module>s