관리-도구
편집 파일: groups_manager.cpython-39.pyc
a �&c�4 � @ 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 dZe�d�e��Ze�d�Zdddd �Zd d � Zd d � Zdd� Ze jjG dd� de jj��Z dS )� )�absolute_import)�unicode_literalsN)�_�loggerz -a-z0-9_.:z^[{}]+$z^[-a-zA-Z0-9_.@]+$T)Zdefault_explicitZuservisible_explicitZ empty_groupsc C s t �| �st�td���| S )zgroup id validatorzInvalid group id)� RE_GROUP_ID�match�argparse�ArgumentTypeErrorr )�value� r �>/usr/lib/python3.9/site-packages/dnf-plugins/groups_manager.py� group_id_type. s r c C sN | � dd�}t|�dkr&t�td���|\}}t�|�sFt�td���||fS )ztranslated texts validator�:� z6Invalid translated data, should be in form 'lang:text'z*Invalid/empty language for translated data)�split�lenr r r �RE_LANGr )r �data�lang�textr r r �translation_type5 s � r c C s: | � � }t�d�t�d|�}|s6tj�td��| ���|S )z#generate group id based on its namez[^{}]� zFCan't generate group id from '{}'. Please specify group id using --id.) �lower�re�sub�format�RE_GROUP_ID_VALID�dnf�cli�CliErrorr )r �group_idr r r � text_to_idA s ��r! c sd e Zd ZdZed�Z� fdd�Zedd� �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__�libcomps�Comps�comps)�selfr �� __class__r r r% Q s z GroupsManagerCommand.__init__c C s� | j ddg dtd�d� | j ddg dtd�d� | j ddtd �d � | j d d d td�d� | j dttd�d� | j ddtd�d� | j dtd�d� | j dttd�d� | 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 td0�d*� | j d1d2d3td4�d5� d S )6Nz--load�appendz COMPS.XMLz load groups metadata from file)�action�default�metavar�helpz--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 packages�packages�*ZPACKAGEzpackage specification)�nargsr/ r0 )� add_argumentr r �intr � add_mutually_exclusive_group)�parserZvisible�sectionr 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�| jjs�tj �td���d S )NTFr z;Can't edit group without specifying it (use --id or --name))r �demands�optsr6 Zsack_activationZavailable_reposZload_system_repo�merge�load�insert�saver, � description� display_order�translated_name�translated_descriptionr3 �id�namer r r )r) r? r r r � configure� s. ��� ���z GroupsManagerCommand.configurec C s8 | j jD �](}t�� }z�|�d�r�t�|��\}tjdd�}z0t � ||� |� � |� |j � W t�|j � nt�|j � 0 W d � q�1 s�0 Y n |� |� W n| tttjf�y" } zZt� }|�� D ]&}||v r�q�t�|�� � |�|� q�tj�t 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' �endswith�gzip�open�tempfileZNamedTemporaryFile�shutil� copyfileobj�closeZ fromxml_frJ �os�unlink�IOError�OSErrorZ ParserError�setZget_last_errorsr �error�strip�addr � exceptions�Errorr r r( )r) � file_nameZ file_compsZgz_fileZ temp_file�err�seenrY 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 )N�Z xml_options���zCan't save file "{}": {})r@ rD r( Zxml_f�COMPS_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 �group�grpr 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 �r�t � }| j j D ]N}t j�|�}|j| jjdddd��� }|s�t�td��|�� q�|�|� q�| j j�r(t � }|D ]}|�|j� q�|�| jj�� j |d�� d d � |D �} | j j �rr| D ]*} |j | t j d �D ]}|j � |� �qX�qDn`| j j!�r�t j"} n| j j#�r�t j$} nt j%} t&| �D ],} |j | | d ��s�|j �'t j(| | d �� �q�dS ) 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<GroupsManagerCommand.edit_group.<locals>.langlist_to_strdictNTF)Z with_nevraZ with_providesZwith_filenameszNo match for argument: {})Zprovidesc S s h | ] }|j �qS r )rJ )�.0�pkgr r r � <setcomp> � z2GroupsManagerCommand.edit_group.<locals>.<setcomp>)rJ r2 ))r@ rJ rE ZdescrF r3 Z uservisiblerG Z name_by_langrH Z desc_by_langr6 rX r ZsubjectZSubjectZget_best_query�baseZsackZlatestr Zwarningr r �updateZ dependenciesZrequires�queryZfilterm�removeZpackages_matchr& ZPACKAGE_TYPE_UNKNOWNZ mandatoryZPACKAGE_TYPE_MANDATORYZoptionalZPACKAGE_TYPE_OPTIONALZPACKAGE_TYPE_DEFAULT�sortedr, ZPackage) r) rh rk r6 Zpkg_specZsubj�qZ requirementsrm Z pkg_names�pkg_nameZpkg_typer r r � edit_group� sX � � z GroupsManagerCommand.edit_groupc C s� | � � | jjs| jjr�| j| jj| jjd�}|d u r�| jjrNtj�t d���t � � }| jjrt| jj|_| jj|_nD| jjr�t | jj�}| j|d d�r�tj �t d��|| jj���||_| jj�|� | �|� | �� | jjs�| jjs�t| 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 �<module> s. �