a
&c4 @ s d dl mZ d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
m Z m Z d dl
Z
d dlZ
dZedeZedZdddd Zd
d Zd d
Zdd Ze
jjG dd de
jjZ dS ) )absolute_import)unicode_literalsN)_loggerz
-a-z0-9_.:z^[{}]+$z^[-a-zA-Z0-9_.@]+$T)Zdefault_explicitZuservisible_explicitZ empty_groupsc C s t | sttd| S )zgroup id validatorzInvalid group id) RE_GROUP_IDmatchargparseArgumentTypeErrorr )value r >/usr/lib/python3.9/site-packages/dnf-plugins/groups_manager.py
group_id_type. s
r
c C sN | dd}t|dkr&ttd|\}}t|sFttd||fS )ztranslated texts validator: z6Invalid translated data, should be in form 'lang:text'z*Invalid/empty language for translated data)splitlenr r r RE_LANGr )r
datalangtextr r r translation_type5 s
r c C s: | }tdtd|}|s6tjtd| |S )z#generate group id based on its namez[^{}] zFCan't generate group id from '{}'. Please specify group id using --id.) lowerresubformatRE_GROUP_ID_VALIDdnfcliCliErrorr )r group_idr r r
text_to_idA s r! c sd e Zd ZdZedZ fddZedd Zdd Z d d
Z
d d Z d
d Z dd Z
dd Z ZS )GroupsManagerCommand)zgroups-managerz$create and edit groups metadata filec s t t| | t | _d S N)superr" __init__libcompsCompscomps)selfr __class__r r r% Q s z GroupsManagerCommand.__init__c C s | j ddg dtdd | j ddg dtdd | j ddtd d
| j d d d
tdd | j dttdd | j ddtdd | j dtdd | j dttdd | j ddg d ttd d | j d dg d ttd d | }|j d!d"d d td#d$ |j d%d"d&d td'd$ | }|j d(d td)d* |j d+d td,d* |j d-d d
td.d | j d/d td0d* | j d1d2d3td4d5 d S )6Nz--loadappendz COMPS.XMLz load groups metadata from file)actiondefaultmetavarhelpz--savez save groups metadata to filez--mergez%load and save groups metadata to file)r/ r0 z--print
store_trueFz#print the result metadata to stdout)r- r. r0 z--idzgroup id)typer0 z-nz--namez
group name)r0 z
--descriptionzgroup descriptionz--display-orderzgroup display orderz--translated-namez LANG:TEXTz translated name for the group)r- r. r/ r2 r0 z--translated-descriptionz$translated description for the groupz--user-visible user_visiblez%make the group user visible (default))destr- r. r0 z--not-user-visible store_falsez make the group user invisiblez --mandatoryz%add packages to the mandatory section)r- r0 z
--optionalz$add packages to the optional sectionz--removez5remove packages from the group instead of adding themz--dependenciesz-include also direct dependencies for packagespackages*ZPACKAGEzpackage specification)nargsr/ r0 ) add_argumentr r
intr add_mutually_exclusive_group)parserZvisiblesectionr r r
set_argparserU sr
z"GroupsManagerCommand.set_argparserc C s | j j}| jjr"d|_d|_d|_| jjrP| jj d| jj | jj
| jj | jj s| jj
s| jjs| jjs| jjd us| jjr| jjs| jjstj tdd S )NTFr z;Can't edit group without specifying it (use --id or --name))r demandsoptsr6 Zsack_activationZavailable_reposZload_system_repomergeloadinsertsaver, description
display_ordertranslated_nametranslated_descriptionr3 idnamer r r )r) r? r r r configure s.
z GroupsManagerCommand.configurec C s8 | j jD ](}t }z|drt|\}tjdd}z0t
|| | | |j
W t|j
nt|j
0 W d q1 s0 Y n
| | W n| tttjfy" } zZt }| D ]&}||v rqt| || qtjt d ||W Y d}~qd}~0 0 | j |7 _ qdS )zm
Loads all input xml files.
Returns True if at least one file was successfuly loaded
z.gzF)deleteNzCan't load file "{}": {}) r@ rB r& r' endswithgzipopentempfileZNamedTemporaryFileshutil copyfileobjcloseZ fromxml_frJ osunlinkIOErrorOSErrorZ ParserErrorsetZget_last_errorsr errorstripaddr
exceptionsErrorr r r( )r) file_nameZ
file_compsZgz_fileZ temp_fileerrseenrY r r r load_input_files s.
< z%GroupsManagerCommand.load_input_filesc C s | j jD ]}z| jj|td}W n, tjyL } z|g}W Y d }~n
d }~0 0 |r|d d D ]}t| q^t
j t
d||d qd S )NZ xml_optionszCan't save file "{}": {})r@ rD r( Zxml_fCOMPS_XML_OPTIONSr& Z XMLGenErrorr rY rZ r r\ r] r r )r) r^ errorsr_ r r r save_output_files s z&GroupsManagerCommand.save_output_filesc C sX d}|r(| j jD ]}|j|kr|} q(q|du rT|rT| j jD ]}|j|kr<|} qTq<|S )zl
Try to find group according to command line parameters - first by id
then by name.
N)r( groupsrI rJ )r) r rJ groupgrpr r r
find_group s
z GroupsManagerCommand.find_groupc C s dd }| j jr| j j|_| j jr,| j j|_| j jr>| j j|_| j jdurT| j j|_| j jrj|| j j|_| j j r|| j j |_
| j j rt }| j j D ]N}t
j|}|j| jjdddd }|sttd| q|| q| j jr(t }|D ]}||j q|| jj j |d d d
|D } | j j rr| D ]*}
|j |
t j d D ]}|j | qXqDn`| j j!rt j"} n| j j#rt j$} nt j%} t&| D ],}
|j |
| d s|j 't j(|
| d qdS ) zE
Set attributes and package lists for selected group
c S s" t }| D ]\}}|||< q |S r# )r& ZStrDict)ZlstZstr_dictr r r r r langlist_to_strdict s
z.langlist_to_strdictNTF)Z
with_nevraZ
with_providesZwith_filenameszNo match for argument: {})Zprovidesc S s h | ]
}|j qS r )rJ ).0pkgr r r z2GroupsManagerCommand.edit_group..)rJ r2 ))r@ rJ rE ZdescrF r3 Z uservisiblerG Z name_by_langrH Z desc_by_langr6 rX r ZsubjectZSubjectZget_best_querybaseZsackZlatestr Zwarningr r updateZ dependenciesZrequiresqueryZfiltermremoveZpackages_matchr& ZPACKAGE_TYPE_UNKNOWNZ mandatoryZPACKAGE_TYPE_MANDATORYZoptionalZPACKAGE_TYPE_OPTIONALZPACKAGE_TYPE_DEFAULTsortedr, ZPackage) r) rh rk r6 Zpkg_specZsubjqZ requirementsrm Z pkg_namespkg_nameZpkg_typer r r
edit_group sX
z GroupsManagerCommand.edit_groupc C s | | jjs| jjr| j| jj| jjd}|d u r| jjrNtjt dt
}| jjrt| jj|_| jj|_nD| jjrt | jj}| j|d drtj
t d|| jj||_| jj| | | | | jjs| jjst| jjtd d S )N)r rJ z-Can't remove packages from non-existent groupzRGroup id '{}' generated from '{}' is duplicit. Please specify group id using --id.rb )ra r@ rI rJ rj rs r r\ r] r r& ZGroupr! r r r r( rg r, rw rf printrD Zxml_strrd )r) rh r r r r run! s0
zGroupsManagerCommand.run)__name__
__module__ __qualname__aliasesr Zsummaryr% staticmethodr> rK ra rf rj rw ry
__classcell__r r r* r r" L s
0$=r" ) Z
__future__r r r rN r& rT r rQ rP Zdnfpluginscorer r r Zdnf.clir compiler r r rd r
r r! ZpluginZregister_commandr ZCommandr" r r r r s.