Your IP : 3.14.255.58
3
�ft`�A � @ s� d dl mZ d dlmZmZ d dlmZ d dlmZ d dl m
Z
d dl Z d dl Z d dl
Z
d dlZd dlZd dlZ d dlZ G dd� dej�ZdS ) � )�print_function)�commands�CliError)�_)�NoModuleException)�loggerNc s* e Zd ZG dd� dej�ZG dd� de�ZG dd� de�ZG dd� de�ZG d d
� d
e�Z G d d � d e�Z
G d
d� de�Z G dd� de�Z G dd� de�Z
G dd� de�ZG dd� de�ZG dd� de�Zeeee e
e e e
eeeh ZehZd%Zed�Z� fdd �Zd d � Zd d � Zd!d"� Zd#d$� Z� ZS )&�
ModuleCommandc s, e Zd Z� fdd�Zdd� Zdd� Z� ZS )zModuleCommand.SubCommandc s( t tj| �j|� tjjj| j�| _d S )N) �superr �
SubCommand�__init__�dnf�module� module_baseZ
ModuleBase�base)�self�cli)� __class__� � /usr/lib/python3.6/module.pyr ( s z!ModuleCommand.SubCommand.__init__c C s� t � }x�| jjD ]�}| jj|�\}}|d kr.q|jr:|jnd}|jrJ|jnd}|jr^|jdksd|jrxt j
t d�j |�� |j
r�|j
nd}| jjj||dd|�}|j|� qW |S )N� � zjOnly module name, stream, architecture or profile is used. Ignoring unneeded information in argument: '{}'���)�set�opts� module_specr � _get_modules�name�stream�version�contextr �infor �format�archr �_moduleContainer�query�update) r �modules_from_specsr �__Znsvcapr r r" �modulesr r r �#_get_modules_from_name_stream_specs, s z<ModuleCommand.SubCommand._get_modules_from_name_stream_specsc C s� t � }t � }x0|D ](}||kr| jjj|�r|j|j� � qW xB|D ]:}tj|�}x*|jtj gd�D ]}|j
rd|j |j
� qdW qDW ||fS )N)Zforms) r r r# ZisModuleActiver% Z getArtifacts�hawkeyZSubjectZget_nevra_possibilitiesZ
FORM_NEVRAr �add) r Z use_modulesZ skip_modulesZ artifactsZ pkg_namesr
ZartifactZsubjZ nevra_objr r r �_get_module_artifact_names> s
z3ModuleCommand.SubCommand._get_module_artifact_names)�__name__�
__module__� __qualname__r r) r, �
__classcell__r r )r r r
& s r
c @ s( e Zd ZdZed�Zdd� Zdd� ZdS ) z ModuleCommand.ListSubCommand�listz,list all module streams, profiles and statesc C s | j j}d|_d|_d S )NT)r �demands�available_repos�sack_activation)r r2 r r r � configureR s z&ModuleCommand.ListSubCommand.configurec C s� | j }| jjr&|j| jjtjjj�}nV| jj rF|j| jjtjjj
�}n6| jj rf|j| jjtjjj �}n|j| jjtjjj
�}|r�t|� d S | jjr�td�}tjj|��d S )NzNo matching Modules to list)r r �enabledZ_get_brief_descriptionr �libdnfr
�ModulePackageContainerZModuleState_ENABLED�disabledZModuleState_DISABLED� installedZModuleState_INSTALLEDZModuleState_UNKNOWN�printr r �
exceptions�Error)r Zmods�output�msgr r r �
run_on_moduleW s( z*ModuleCommand.ListSubCommand.run_on_moduleN)r1 )r- r. r/ �aliasesr �summaryr5 r@ r r r r �ListSubCommandM s rC c @ s( e Zd ZdZed�Zdd� Zdd� ZdS ) z ModuleCommand.InfoSubCommandr z)print detailed information about a modulec C s | j j}d|_d|_d S )NT)r r2 r3 r4 )r r2 r r r r5 t s z&ModuleCommand.InfoSubCommand.configurec C sf | j jr| jj| j j�}n*| j jr4| jj| j j�}n| jj| j j�}|rRt|� nt j
j t d���d S )NzNo matching Modules to list)
r �verboser Z_get_full_infor �profileZ_get_info_profilesZ _get_infor; r r< r= r )r r> r r r r@ y s
z*ModuleCommand.InfoSubCommand.run_on_moduleN)r )r- r. r/ rA r rB r5 r@ r r r r �InfoSubCommando s rF c @ s( e Zd ZdZed�Zdd� Zdd� ZdS ) z ModuleCommand.EnableSubCommand�enablezenable a module streamc C s$ | j j}d|_d|_d|_d|_d S )NT)r r2 r3 r4 � resolving� root_user)r r2 r r r r5 � s
z(ModuleCommand.EnableSubCommand.configurec C s� y| j j| jj� W nl tjjk
r� } zL| jjj rb|j
s@|j rD|�|j rb|j d t
jjjkrb|�tjt|�� W Y d d }~X nX d S )Nr )r rG r r r r< �
MarkingErrorsr �conf�strict�no_match_group_specs�error_group_specs�module_depsolv_errorsr7 r
r8 �!ModuleErrorType_ERROR_IN_DEFAULTSr �error�str)r �er r r r@ � s
z,ModuleCommand.EnableSubCommand.run_on_moduleN)rG )r- r. r/ rA r rB r5 r@ r r r r �EnableSubCommand� s rT c @ s( e Zd ZdZed�Zdd� Zdd� ZdS ) z ModuleCommand.DisableSubCommand�disablez%disable a module with all its streamsc C s$ | j j}d|_d|_d|_d|_d S )NT)r r2 r3 r4 rH rI )r r2 r r r r5 � s
z)ModuleCommand.DisableSubCommand.configurec C s� y| j j| jj� W nl tjjk
r� } zL| jjj rb|j
s@|j rD|�|j rb|j d t
jjjkrb|�tjt|�� W Y d d }~X nX d S )Nr )r rU r r r r< rJ r rK rL rM rN rO r7 r
r8 rP r rQ rR )r rS r r r r@ � s
z-ModuleCommand.DisableSubCommand.run_on_moduleN)rU )r- r. r/ rA r rB r5 r@ r r r r �DisableSubCommand� s rV c @ s( e Zd ZdZed�Zdd� Zdd� ZdS ) z ModuleCommand.ResetSubCommand�resetzreset a modulec C s$ | j j}d|_d|_d|_d|_d S )NT)r r2 r3 r4 rH rI )r r2 r r r r5 � s
z'ModuleCommand.ResetSubCommand.configurec C sb y| j j| jj� W nH tjjk
r\ } z(| jjj r>|j
r>|�t j t
|�� W Y d d }~X nX d S )N)r rW r r r r< rJ r rK rL rM r rQ rR )r rS r r r r@ � s
z+ModuleCommand.ResetSubCommand.run_on_moduleN)rW )r- r. r/ rA r rB r5 r@ r r r r �ResetSubCommand� s rX c @ s( e Zd ZdZed�Zdd� Zdd� ZdS ) z ModuleCommand.InstallSubCommand�installz/install a module profile including its packagesc C s$ | j j}d|_d|_d|_d|_d S )NT)r r2 r3 r4 rH rI )r r2 r r r r5 � s
z)ModuleCommand.InstallSubCommand.configurec C sp y | j j| jj| jjj� W nN tjj k
rj } z.| jjjrL|j
sH|j rL|�t j
t|�� W Y d d }~X nX d S )N)r rY r r r rK rL r r< rJ rM rN r rQ rR )r rS r r r r@ � s
z-ModuleCommand.InstallSubCommand.run_on_moduleN)rY )r- r. r/ rA r rB r5 r@ r r r r �InstallSubCommand� s rZ c @ s( e Zd ZdZed�Zdd� Zdd� ZdS ) z ModuleCommand.UpdateSubCommandr% z0update packages associated with an active streamc C s$ | j j}d|_d|_d|_d|_d S )NT)r r2 r3 r4 rH rI )r r2 r r r r5 � s
z(ModuleCommand.UpdateSubCommand.configurec C s& | j j| jj�}|r"tdj|���d S )Nz, )r Zupgrader r r �join)r Z module_specsr r r r@ � s z,ModuleCommand.UpdateSubCommand.run_on_moduleN)r% )r- r. r/ rA r rB r5 r@ r r r r �UpdateSubCommand� s r\ c @ s( e Zd Zd Zed�Zdd� Zdd� ZdS )
z ModuleCommand.RemoveSubCommand�remove�erasez3remove installed module profiles and their packagesc C s0 | j j}d|_d|_d|_d|_d|_d|_d S )NTF)r r2 Z
allow_erasingr3 Zfresh_metadatarH rI r4 )r r2 r r r r5 � s z(ModuleCommand.RemoveSubCommand.configurec
C s� | j j| jj�}| jjr�| j� }| j|t� �\}}| j| jj j
� |�\}}| jj j � j
� j|d�}| jj j � j
� j|d�}xF|D ]>}||kr�td�j|�} tj| � q�| jjj|| jjjd� q�W |s�d S tjtjj|d�� d S )N)r z0Package {} belongs to multiple modules, skipping)Z
clean_deps)rM )r r] r r �allr) r, r r r# ZgetModulePackages�sackr$ r: �filtermr r! r r Zgoalr^ rK Z clean_requirements_on_removerQ r r< rJ )
r Zskipped_groupsr& Zremove_names_from_specr' Z
keep_namesZ remove_queryZ
keep_query�pkgr? r r r r@ � s&
z,ModuleCommand.RemoveSubCommand.run_on_moduleN)r] r^ )r- r. r/ rA r rB r5 r@ r r r r �RemoveSubCommand� s rc c @ s( e Zd ZdZed�Zdd� Zdd� ZdS ) z ModuleCommand.SwitchToSubCommand� switch-toz7switch a module to a stream and distrosync rpm packagesc C s. | j j}d|_d|_d|_d|_d| jj_d S )NT) r r2 r3 r4 rH rI r rK Zmodule_stream_switch)r r2 r r r r5 s z*ModuleCommand.SwitchToSubCommand.configurec C sr y | j j| jj| jjjd� W nN tjj k
rl } z.| jjjrN|j
sJ|j rN|�t j
t|�� W Y d d }~X nX d S )N)rL )r Z switch_tor r r rK rL r r< rJ rM rN r rQ rR )r rS r r r r@ " s
z.ModuleCommand.SwitchToSubCommand.run_on_moduleN)rd )r- r. r/ rA r rB r5 r@ r r r r �SwitchToSubCommand s re c @ s( e Zd ZdZed�Zdd� Zdd� ZdS ) z ModuleCommand.ProvidesSubCommand�provideszlist modular packagesc C s | j j}d|_d|_d S )NT)r r2 r3 r4 )r r2 r r r r5 0 s z*ModuleCommand.ProvidesSubCommand.configurec C s | j j| jj�}|r t|� d S )N)r Z_what_providesr r r; )r r> r r r r@ 5 s z.ModuleCommand.ProvidesSubCommand.run_on_moduleN)rf )r- r. r/ rA r rB r5 r@ r r r r �ProvidesSubCommand+ s rg c @ s( e Zd ZdZed�Zdd� Zdd� ZdS ) z!ModuleCommand.RepoquerySubCommand� repoqueryz#list packages belonging to a modulec C s | j j}d|_d|_d S )NT)r r2 r3 r4 )r r2 r r r r5 ? s z+ModuleCommand.RepoquerySubCommand.configurec C s� t � }x*| jjD ] }| jj|�\}}|j|� qW | j|t � �\}}t � }| jjs\| jj r�| j j
j � j� j |d�}x|D ]} |j
t| �� qzW | jjr�| j j
j � j� j |d�}x|D ]} |j
t| �� q�W djt|��}
t|
� d S )N)Z nevra_strict)r �
)r r r r r r% r, � availabler: r r` r$ ra r+ rR r[ �sortedr; ) r r& r r( r' Znames_from_specZspec_artifactsZpackage_stringsr$ rb r> r r r r@ D s"
z/ModuleCommand.RepoquerySubCommand.run_on_moduleN)rh )r- r. r/ rA r rB r5 r@ r r r r �RepoquerySubCommand: s rl r
zInteract with Modules.c s> t t| �j� � � fdd�| jD �}d | _dd� |D �| _d S )Nc 3 s | ]}|� �V qd S )Nr )�.0�subcmd)r r r � <genexpr>d s z)ModuleCommand.__init__.<locals>.<genexpr>c S s i | ]}|j D ]
}||�qqS r )rA )rm rn �aliasr r r �
<dictcomp>f s z*ModuleCommand.__init__.<locals>.<dictcomp>)r r r �SUBCMDSrn �_subcmd_name2obj)r r Z subcmd_objs)r )r r r b s
zModuleCommand.__init__c C s |j � }|jdddtd�d� |jdddtd�d� |jd d
dtd �d� |jd d
dtd�d� |jdddtd�d� |jdddtd�d� g }g }xHt| jdd� d�D ]2}|j|jd � |jdj|jd |jp�d�� q�W |jdd |d d j |�d � |jd d!d"td#�d$� d S )%Nz --enabledr6 �
store_truezshow only enabled modules)�dest�action�helpz
--disabledr9 zshow only disabled modulesz --installedr: z'show only installed modules or packagesz --profilerE zshow profile contentz --availablerj z show only available packagesz--allr_ zremove all modular packagesc S s
| j d S )Nr )rA )�xr r r �<lambda>~ s z-ModuleCommand.set_argparser.<locals>.<lambda>)�keyr z{}: {}r rn r z<modular command>ri )�nargs�choices�metavarrw r z module-spec�*zModule specification)r} r{ rw )
Z add_mutually_exclusive_group� add_argumentr rk rr �appendrA r! rB r[ )r �parserZnarrowsZsubcommand_choicesZsubcommand_helprn r r r �
set_argparseri s8
"
zModuleCommand.set_argparserc C sZ y| j | jjd | _W n( ttfk
r@ | jjj� t�Y nX | j| j_| jj� d S )Nr ) rs r rn r �KeyErrorr Z optparserZ print_usager5 )r r r r r5 � s
zModuleCommand.configurec C s | j � | jj� d S )N)�check_required_argumentrn r@ )r r r r �run� s zModuleCommand.runc C sR dd� | j D �}| jjd |krN| jjsNttd�jtjj | jj
| jjd ���d S )Nc S s g | ]}|j D ]}|�qqS r )rA )rm rn rp r r r �
<listcomp>� s z9ModuleCommand.check_required_argument.<locals>.<listcomp>r z{} {} {}: too few arguments) �SUBCMDS_NOT_REQUIRED_ARGr rn r r r r! r �utilZ MAIN_PROGZcommand)r Znot_required_argumentr r r r� � s
z%ModuleCommand.check_required_argument)r
) r- r. r/ r �Commandr
rC rF rT rV rX rZ r\ rc re rg rl rr r� rA r rB r r� r5 r� r� r0 r r )r r r % s. '"% r )Z
__future__r Zdnf.clir r Zdnf.i18nr Zdnf.module.exceptionsr Zdnf.utilr r �sys�osr* r7 Zdnf.module.module_baseZdnf.exceptionsr� r r r r r �<module> s