3
ft` @ s d dl mZ d dl mZ d dl mZ d dlZd dlZd dlZd dl Zd dlm
Z
d dl Zd dl Zd dl
mZ ejdZG dd d e
jZdS )
)absolute_import)print_function)unicode_literalsN)commands)_dnfc @ sl e Zd ZdZedZedd Zdd Zdd Z d d
Z
d d Z d
d Z dd Z
dd Zdd Zdd ZdS ) AliasCommandaliasz List or create command aliasesc C sl | j }|jdddtdd |jdddtdd | jdd d
d d
d gtd
d | jdddtdd d S )Nz--enable-resolvingF
store_truezenable aliases resolving)defaultactionhelpz--disable-resolvingzdisable aliases resolving
subcommand?listadddeletezaction to do with aliases)nargsr choicesr
r *zcommand[=result]zalias definition)r metavarr
)Z add_mutually_exclusive_group add_argumentr )parserZ enable_group r /usr/lib/python3.6/alias.py
set_argparser* s
zAliasCommand.set_argparserc C sH | j j}| jjdkrd|_tj jj | _| jj | jj
| _ | j d S )Nr r T)r r )
clidemandsoptsr Z root_userr aliasesZAliases aliases_baseZ
_load_aliasesenabledZresolving_enabled_update_config_from_options)selfr r r r configure9 s
zAliasCommand.configurec C s d }| j jr d}tjtd | j jr8d}tjtd |d k rtjjt j
j j sft
t j
j j dj t jjjt j
j j dd d|i | jj s|| j_d S )NTzAliases are now enabledFzAliases are now disabledwmainr! )r Zenable_resolvingloggerinfor Zdisable_resolvingospathexistsr r r ZALIASES_CONF_PATHopencloseconfZ
BaseConfigZwrite_raw_configfiler Z_disabled_by_environr! )r# r! r r r r" B s
z(AliasCommand._update_config_from_optionsc C s i }x| j jD ]}|jdd}|d j }t|j dkrLtjtd| q|jdrhtjtd| qt|dkrtjtd| q|d j ||< qW |S )N= r zInvalid alias key: %s-z Alias argument has no value: %s) r r splitstriplenr' warningr
startswith)r# Z new_aliasesr cmdr r r _parse_option_aliasT s
z AliasCommand._parse_option_aliasc C sx t jjtjjjs&ttjjjdj ytjjj tjjj}W n4 tj
j k
rr } zt j
td| d S d }~X nX |S )Nr% zConfig error: %s)r) r* r+ r r r ALIASES_USER_PATHr, r- Z
AliasesConfig
exceptionsZ ConfigErrorr' r5 r )r# r. er r r _load_user_aliasese s z AliasCommand._load_user_aliasesc C sd t tjjjd}d}|dj|7 }|d7 }x*|j D ] \}}|dj|dj|7 }q4W |j| d S )Nr% z[main]
zenabled = {}
z
[aliases]
z{} = {}
) r, r r r r9 formatitemsjoinwrite)r# user_aliasesr! Zfileobjoutputkeyvaluer r r _store_user_aliasesp s z AliasCommand._store_user_aliasesc C sP | j }|j}|d krd S |j| | j||j tjtddj|j d S )NzAliases added: %sz, )
r< r updaterF r! r' r( r r@ keys)r# r r. rB r r r add_aliasesy s
zAliasCommand.add_aliasesc C s | j }|j}|d krd S g }xF|D ]>}y||= |j| W q$ tk
r` tjtd| Y q$X q$W | j||j tjtddj | d S )NzAlias not found: %szAliases deleted: %sz, )
r< r appendKeyErrorr' r( r rF r! r@ )r# cmdsr. rB Z
valid_cmdsr7 r r r remove_aliases s
zAliasCommand.remove_aliasesc C s~ |g}y| j j|}W nH tjjk
r^ } z(tjtd||dj| j j | W Y d d }~X n X t
td|dj|f d S )Nz%s, alias %s="%s"r= z
Alias %s='%s') r Z_resolver r: Errorr' errorr r@ r print)r# r7 argsr; r r r
list_alias s 0zAliasCommand.list_aliasc C s | j jstjtd | jjdkrL| j }|s>tj j
td| j | d S | jjdkr| jj }|g krxtj j
td| j
| d S | jj s| j jstjtd d S xX| j jD ]}| j| qW n s