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� dejj��ZdS )� )�absolute_import)�unicode_literals)�_�logger)� OptionParserNc s� e Zd ZdgZed�Z� fdd�Zedd� �Zdd� Z d d
� Z
d d � Z d
d� Z dd� Z
dd� Zdd� Zd dd�Zd!dd�Zedd� �Zd d � Zd d � Z� ZS )"�DownloadCommandZdownloadz%Download package to current directoryc s t t| �j|� d | _d | _d S )N)�superr �__init__�opts�parser)�self�cli)� __class__� � /usr/lib/python3.6/download.pyr + s zDownloadCommand.__init__c C s� | j ddtd�d� | j ddtd�d� | j d dtd
�d� | j d dtd �d� | j d
ddg tjdtd�d� | j ddtd�d� | j ddtd�d� | j ddddtd�d� | j d d d d d d!gg td"�d#� d S )$N�packages�+zpackages to download)�nargs�helpz--source�
store_truez download the src.rpm instead)�actionr z --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 �metavarr z --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)r r r z--urlprotocols�append�http�httpsZrsync�ftpz4when 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_repo�source�baseZreposZenable_source_repos� debuginfo� debugsourceZenable_debug_repos�destdir�conf�dnf�i18n�ucd�os�getcwd)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
r�xd|D ]\}|j t j
kr�|j| j jd�}|r�t|� q�td�|j }| jjjr�tjj|��tj|� q�W 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 �repoid�hawkey�CMDLINE_REPO_NAMEZremote_locationZ urlprotocols�printr �namer& r* �strictr+ �
exceptions�Errorr Zwarning�
_do_downloads)r �pkgs�pkgr �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� |r�x^|D ]V}|j � }t
j j | jj
jt
j j|��} t
j j| �r�t
j j|| �r�q�tj|| jj
j� q�W 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 �
<listcomp>� s z1DownloadCommand._do_downloads.<locals>.<listcomp>r c S s | j j| j jfS )N)ZrepoZpriorityZcost)�xr r r �<lambda>� s z/DownloadCommand._do_downloads.<locals>.<lambda>)�keyc S s g | ] }|j � �qS r )�localPkg)rD rA r r r rE � s )�
setdefault�strr �values�sortr& Zdownload_packages�output�progressrI r. �path�joinr* r) �basename�exists�samefile�shutil�copy�sorted) r r@ Zpkg_dictrA Z to_downloadZcmdlineZpkg_listZ pkgs_cmdline�src�dstZ 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_packages�setr[ )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; �epoch�version�release�arch)r^ r& �sack�query� availabler[ Z
debug_nameZsource_debug_name�filter�intr` ra rb rc �add)r r\ �dbg_pkgs�qrA Zdbg_name�
dbg_available�pr 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<tjtjj |�� | j
j j r|tjt
d�� tjj|��W Y dd}~X qX qW ttj|� �}|S )z Get packages matching pkg_specs.z Exiting due to strict setting.N)�_get_query_source�
_get_queryr r+ r= �PackageNotFoundErrorr �errorr, r- r& r* r< r r> �list� itertools�chain)r r\ r% �funcZqueries�pkg_spec�er@ 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 d�g}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.<locals>.<listcomp>)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
d�t|�� 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; Zevr�infor rr )r@ r_ rA r r r r] s
z$DownloadCommand._get_source_packagesc C s� t jjj|�d }|o |dk}|s8|jd�rdtjj|�rd| jj|g| jj j
d�}| jj j � j
|d�S t jj|�}|j| jj | jjd �}|j� }|j
d
d �}| jjr�|j| jjd �}t|j� �dkr�td
�| }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+ ZpycompZurlparse�endswithr. rP �isfiler& Zadd_remote_rpmsrN rO rd re Zfilterm�subject�SubjectZget_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| jj�j� }|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 �<module> s