관리-도구
편집 파일: download.cpython-36.pyc
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