3
gt`* @ s d dl mZ d dl mZ d dlmZmZmZ d dlZd dlZd dl Zd dl
Zd dl Z d dl Z d dl
Z
d dlZd dlZejjG dd dejjZdd Zejd Zejd
Zejd Zejd Zd
d ZdS ) )absolute_import)unicode_literals)_loggerP_Nc @ sR e Zd ZdgZedjejjdZ e
dd Z dd Z dd Z
d
d Zd d
ZdS )ConfigManagerCommandzconfig-managerz4manage {prog} configuration options and repositories)progc C s | j dddtdd | j dddtd d
| j d g d d
tdd | j dddtdd
| j dddtdd
| j }|j ddddtdd |j ddddtdd d S )Ncrepo*repozrepo to modify)nargsmetavarhelpz--saveF
store_truez/save the current options (useful with --setopt))defaultactionr z
--add-repoappendZURLz8add (and enable) the repo from the specified file or url)r r r
r z--dumpz,print current configuration values to stdoutz--dump-variablesz print variable values to stdoutz
--set-enabled set_enabledz"enable repos (automatically saves))r destr r z--set-disabled set_disabledz#disable repos (automatically saves)) add_argumentr Z add_mutually_exclusive_group)parserZ enable_group r $/usr/lib/python3.6/config_manager.py
set_argparser) s, z"ConfigManagerCommand.set_argparserc C s | j j}d|_| jjg kp@| jjp@| jjp@| jjp@| jjp@| jj sp| j j
j t dj
djdddddd d
d g | jjg krtjt d | jjs| jj s| jjs| jjrd|_d
d | jjD }dd |D | j_d S )NTz.one of the following arguments is required: {} z--savez
--add-repoz--dumpz--dump-variablesz
--set-enabledz--enablez--set-disabledz --disablez{Warning: --enablerepo/--disablerepo arguments have no meaningwith config manager. Use --set-enabled/--set-disabled instead.c S s g | ]}|d kr|j d qS ),)split).0xr r r
_ s z2ConfigManagerCommand.configure..c S s" g | ]}|D ]}|d kr |q qS ) r )r Zsublistitemr r r r a s )clidemandsZavailable_reposoptsadd_reposavedumpdump_variablesr r Z optparsererrorr formatjoinZrepos_edr ZwarningZ root_userr )selfr$ Z temp_listr r r configureB s* z ConfigManagerCommand.configurec C s | j jr| j n| j dS )z Execute the util action here.N)r% r& modify_repo)r- r r r rund s
zConfigManagerCommand.runc s g t fdd}jjrnxjjD ]|d q.W tjdrxLjjj D ]|d qZW n,tjdrx jjj D ]|d qW rtjjt ddj
j j }i }tjdrjj
rjj
}jjrx*j j jj D ]\}td |f qW jj s0d
jjkrjjr\|r\j j jj j jd
|j| jjrtj jjd
tj j j sd S jjsjjrdj_xt D ]}i }jjrd |d
< njjrd|d
< tjdr*x4jjj D ]$\}}tj|j |r|j | qW jjrT|rTj j j|j |j |j| jjrtj jjd|j t|j qW d S )z< process --set-enabled, --set-disabled and --setopt options c s0 j jj| }|s j n|r, j| d S )N)baseZreposZ get_matchingaddextend)keyZadd_matching_reposZmatching)matching_reposnamenot_matching_repos_idr- r r match_reposq s
z5ConfigManagerCommand.modify_repo..match_reposT repo_setoptsFz No matching repo to modify: %s.z, main_setoptsz%s = %smainN1Zenabled0zrepo: ) setr% r hasattrr9 keysdnf
exceptionsErrorr r, r1 confr: r) Z
substitutionsitemsprintr' Zwrite_raw_configfileZconfig_file_pathr( outputZ
fmtSectionr r sortedfnmatchidupdateZrepofile) r- r8 ZsbcZmodifyvalr Z repo_modifyrepoidZsetoptsr )r5 r6 r7 r- r r/ k s`
z ConfigManagerCommand.modify_repoc
C sN | j jj}d}x | jjD ]}tjjj|jdkrDdt j
j | }t j
td| |jdrt j
j|}t j
j||}y6| j j|dd}tj|j| t j|d |j W n6 tk
r } z|d 7 }t j| wW Y d
d
}~X nX qt|}d jtj j |}t j
j|d | }d
|||f } t || sqqW |rJtj j t!dd|d
S )z process --add-repo option r r! zfile://zAdding repo from: %sz.repozw+)modei Nz$created by {} config-manager from {}z%s.repoz"[%s]
name=%s
baseurl=%s
enabled=1
z Configuration of repo failedz Configuration of repos failed)"r1 rD Z get_reposdirr% r& rA pycompZurlparseschemeospathabspathr infor endswithbasenamer, ZurlopenshutilZcopy2r6 chmodcloseIOErrorr* sanitize_url_to_fsr+ util MAIN_PROG save_to_filerB rC r )
r- Z myrepodirZ errors_counturlZdestnameferM Zreponamecontentr r r r& s8
z ConfigManagerCommand.add_repoN)__name__
__module__ __qualname__aliasesr r+ rA r] r^ Zsummary staticmethodr r. r0 r/ r& r r r r r " s "Br c C sp y4t | d }tjj|| tj| d W d Q R X W n6 ttfk
rj } ztj t
d| | dS d }~X nX dS )Nzw+i z&Could not save repo to repofile %s: %sFT) openrA rP Z
write_to_filerR rY r[ OSErrorr r* r )filenamerc fdrb r r r r_ s
r_ z^\w+:/*(\w+:|www\.)?z[?/:&#|~\*\[\]\(\)\'\\]+z^[,.]*z[,.]*$c C s* ybt j| r`tjjr&| jdjd} n:t| trB| jdjd} n
| jd} t| t r`| jd} W n t
t t t
fk
r~ Y nX t jd| } tjd| } tjd| } tjd| } t| dkr| dd jd}dt|d
}tj }|j| |d jd | d| d |j } d }tj|d| S ) zReturn a filename suitable for the filesystem and for repo id
Strips dangerous and common characters to create a filename we
can use to store the cache in.
Zidnazutf-8r! r N rO zE[^abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789_.:-]) RE_SCHEMEmatchrA rP ZPY3encodedecode
isinstancestrZunicodeUnicodeDecodeErrorUnicodeEncodeError UnicodeError TypeErrorsubRE_SLASHRE_BEGINRE_FINALlenr hashlibZsha256rK Z hexdigestre)r` parts lastindexZcsumZ
allowed_regexr r r r\ s.
r\ ) Z
__future__r r Zdnfpluginscorer r r rA Zdnf.cliZ
dnf.pycompZdnf.utilrI r rR r rX ZpluginZregister_commandr# ZCommandr r_ compilerp r{ r| r} r\ r r r r s( 1