Edit File: groups_manager.cpython-36.pyc
3 �*�d�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 dZejdje��Zejd�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 j| �stjtd���| S )zgroup id validatorzInvalid group id)� RE_GROUP_ID�match�argparse�ArgumentTypeErrorr )�value� r �$/usr/lib/python3.6/groups_manager.py� group_id_type. s r c C sN | j dd�}t|�dkr&tjtd���|\}}tj|�sFtjtd���||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: | j � }tjdjt�d|�}|s6tjjtd�j| ���|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�groups-managerz$create and edit groups metadata filec s t t| �j|� tj� | _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� }|j d!d"d d td#�d$� |j d%d"d&d td'�d$� | j� }|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. r/ z--print� store_trueFz#print the result metadata to stdout)r, r- r/ z--idzgroup id)�typer/ z-nz--namez group name)r/ z --descriptionzgroup descriptionz--display-orderzgroup display orderz--translated-namez LANG:TEXTz translated name for the group)r, r- r. r1 r/ z--translated-descriptionz$translated description for the groupz--user-visible� user_visiblez%make the group user visible (default))�destr, r- r/ z--not-user-visibleZ store_falsez make the group user invisiblez --mandatoryz%add packages to the mandatory section)r, r/ 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. r/ )� add_argumentr r �intr Z add_mutually_exclusive_group)�parserZvisibleZsectionr r r � set_argparserU sR z"GroupsManagerCommand.set_argparserc C s� | j j}| jjr"d|_d|_d|_| jjrP| jjj d| jj� | jj j | jj� | jj s�| jj s�| jjs�| jjs�| jjd k s�| jjr�| jj r�| jj r�tj jtd���d S )NTFr z;Can't edit group without specifying it (use --id or --name))r �demands�optsr4 Zsack_activationZavailable_reposZload_system_repo�merge�load�insert�saver+ � description� display_order�translated_name�translated_descriptionr2 �id�namer r r )r) r; r r r � configure� s" z GroupsManagerCommand.configurec C s �x| j jD �] }tj� }yp|jd�r~tj|��F}tjdd�}z$t j ||� |j � |j |j � W dtj|j � X W dQ R X n |j |� W n~ tttjfk �r } zXt� }x2|j� D ]&}||kr�q�tj|j� � |j|� q�W tjjt d�j ||���W Y dd}~X q X | j |7 _ q W dS )zm Loads all input xml files. Returns True if at least one file was successfuly loaded z.gzF)�deleteNzCan't load file "{}": {}) r< r> r& r' �endswith�gzip�open�tempfileZNamedTemporaryFile�shutilZ copyfileobj�closeZ fromxml_frF �os�unlink�IOError�OSErrorZ ParserError�setZget_last_errorsr �error�strip�addr � exceptions�Errorr r r( )r) � file_nameZ file_compsZgz_fileZ temp_file�err�seenrT r r r �load_input_files� s, $z%GroupsManagerCommand.load_input_filesc C s� x�| j jD ]�}y| jj|td�}W n* tjk rL } z |g}W Y d d }~X nX |r x"|d d� D ]}tj|j � � q`W t j j t d�j||d j � ���q W d S )N)� xml_options� zCan't save file "{}": {}���r_ )r< r@ r( Zxml_f�COMPS_XML_OPTIONSr& Z XMLGenErrorr rT rU r rW rX r r )r) rY �errorsrZ r r r �save_output_files� s z&GroupsManagerCommand.save_output_filesc C s\ d}|r*x | j jD ]}|j|kr|}P qW |dkrX|rXx | j jD ]}|j|kr@|}P q@W |S )zl Try to find group according to command line parameters - first by id then by name. N)r( �groupsrE rF )r) r rF �groupZgrpr 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 jdk rT| j j|_| j jrj|| j j�|_| j j r�|| j j �|_ | j j �r�t � }xZ| j j D ]N}t jj|�}|j| jjdddd�j� }|s�tjtd�j|�� q�|j|� q�W | j j�r2t � }x|D ]}|j|j� �qW |j| jjj� j |d�� d d � |D �} | j j �r�x�| D ].} x&|j | t j d �D ]}|j j |� �qfW �qPW nd| j j!�r�t j"} n| j j#�r�t j$} nt j%} x8t&| �D ],} |j | | d ��s�|j j't j(| | d �� �q�W dS ) zE Set attributes and package lists for selected group c S s& t j� }x| D ]\}}|||< qW |S )N)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 )rF )�.0�pkgr r r � <setcomp> s z2GroupsManagerCommand.edit_group.<locals>.<setcomp>)rF r1 ))r< rF rA ZdescrB r2 Z uservisiblerC Z name_by_langrD Z desc_by_langr4 rS r ZsubjectZSubjectZget_best_query�baseZsackZlatestr Zwarningr r �updateZ dependenciesZrequiresZqueryZfilterm�removeZpackages_matchr& ZPACKAGE_TYPE_UNKNOWNZ mandatoryZPACKAGE_TYPE_MANDATORYZoptionalZPACKAGE_TYPE_OPTIONALZPACKAGE_TYPE_DEFAULT�sortedr+ ZPackage) r) rd rf r4 Zpkg_specZsubj�qZ requirementsrh Z pkg_namesZpkg_nameZpkg_typer r r � edit_group� sT z GroupsManagerCommand.edit_groupc C s | j � | jjs| jjr�| j| jj| jjd�}|d kr�| jjrNtjjt d���t j � }| jjrt| jj|_| jj|_nD| jjr�t | jj�}| j|d d�r�tj jt d�j|| jj���||_| jjj|� | j|� | j� | jjs�| jj r�t| jjtd�� d S )N)r rF z-Can't remove packages from non-existent groupzRGroup id '{}' generated from '{}' is duplicit. Please specify group id using --id.)r] )r\ r< rE rF re rl r rW rX r r& ZGroupr! r r r r( rc r+ ro rb �printr@ Zxml_strr` )r) rd r r r r �run! s, zGroupsManagerCommand.run)r# )�__name__� __module__� __qualname__�aliasesr Zsummaryr% � staticmethodr: rG r\ rb re ro rq � __classcell__r r )r* r r" L s 1$=r" ) Z __future__r r r rJ r& rO r rM rL Zdnfpluginscorer r r Zdnf.clir �compiler r r r` r r r! ZpluginZregister_commandr ZCommandr" r r r r �<module> s,