3
��f& � @ s` d dl mZ d dlmZ d dlmZ d dlZd dlZ d dl
Z
d dl Z ej j
Z
G dd� de�ZdS )� )�unicode_literals)� ConfigError)�_Nc s� e Zd Zdd� Zdd� Zdd� Zdd� Zf fd d
�Zd d � Zd
d� Z dd� Z
dd� Z � fdd�Z dd� Z
dd� Zdd� Z� ZS )�RepoDictc C sj |j }|| kr d}t|| ��y|jj� W n0 tk
r\ } ztdj|���W Y d d }~X nX || |< d S )Nz;Repository %s is listed more than once in the configurationz{0})�idr Z_repoZverify� RuntimeError�format)�self�repoZid_�msg�e� r
� /usr/lib/python3.6/repodict.py�add# s z RepoDict.addc C s t jj| j� �S )N)�dnf�util�
MultiCallList�values)r r
r
r �all/ s z RepoDict.allc C s t jj| j� � S )N)r r �empty� iter_enabled)r r
r
r � _any_enabled3 s zRepoDict._any_enabledc C sP xJ| j � D ]>}x8| j||j��D ]$}|js tjtd�|j� |j� q W q
W d S )Nzenabling %s repository)r � get_matchingr �enabled�logger�infor �enable)r Z sub_name_fnr
�foundr
r
r �_enable_sub_repos6 s
zRepoDict._enable_sub_reposc
s� � fdd�}t jj|� �}x:|D ]2}d|kr>djtjj|��}| j||�g7 _q W x$|j� D ]\}} t |||| �� q`W | j
|� t j t
d�|dj|�� |S )a�
Creates new repo object and add it into RepoDict. Variables in provided values will be
automatically substituted using conf.substitutions (like $releasever, ...)
@param repoid: Repo ID - string
@param conf: dnf Base().conf object
@param baseurl: List of strings
@param kwargs: keys and values that will be used to setattr on dnf.repo.Repo() object
@return: dnf.repo.Repo() object
c sp t | t�r tjjj| � j�S t | t�s0t | t�rlg }x.| D ]&}t |t�r:|j tjjj|� j�� q:W |rl|S | S )N)
�
isinstance�str�libdnf�confZ ConfigParser�
substituteZ
substitutions�list�tuple�append)r Z substituted�value)r" r
r r# I s
z)RepoDict.add_new_repo.<locals>.substitutez://z file://{}zAdded %s repo from %sz, )r r
ZRepor �os�path�abspath�baseurl�items�setattrr r r r �join)
r Zrepoidr" r+ �kwargsr# r
r) �keyr' r
)r" r � add_new_repo= s
zRepoDict.add_new_repoc C s dd� }| j |� dS )z@enable debug repos corresponding to already enabled binary reposc S s& | j d�r dj| d d� �S dj| �S )Nz-rpmsz
{}-debug-rpms� z {}-debuginfo���)�endswithr )�namer
r
r �
debug_namee s z/RepoDict.enable_debug_repos.<locals>.debug_nameN)r )r r6 r
r
r �enable_debug_reposa s zRepoDict.enable_debug_reposc C s dd� }| j |� dS )zAenable source repos corresponding to already enabled binary reposc S s& | j d�r dj| d d� �S dj| �S )Nz-rpmsz{}-source-rpmsr2 z {}-sourcer3 )r4 r )r5 r
r
r � source_nameo s z1RepoDict.enable_source_repos.<locals>.source_nameN)r )r r8 r
r
r �enable_source_reposk s z RepoDict.enable_source_reposc sZ t jj� �r,� �fdd��D �}t jj|�S �j� d �}|d krLt jjg �S t jj|g�S )Nc s g | ]}t j |� �r�| �qS r
)�fnmatch)�.0�k)r0 r r
r �
<listcomp>x s z)RepoDict.get_matching.<locals>.<listcomp>)r r Zis_glob_patternr �get)r r0 �lr
r
)r0 r r r u s zRepoDict.get_matchingc C s dd� | j � D �S )Nc s s | ]}|j r|V qd S )N)r )r; �rr
r
r � <genexpr>� s z(RepoDict.iter_enabled.<locals>.<genexpr>)r )r r
r
r r s zRepoDict.iter_enabledc s$ dd� t tt| �j� dd� d�D �S )z return repos sorted by priorityc s s | ]
}|V qd S )Nr
)r; �itemr
r
r rA � s z!RepoDict.items.<locals>.<genexpr>c S s | d j | d jfS )N� )ZpriorityZcost)�xr
r
r �<lambda>� s z RepoDict.items.<locals>.<lambda>)r0 )�sorted�superr r, )r )� __class__r
r r, � s zRepoDict.itemsc C s | j � S )N)�keys)r r
r
r �__iter__� s zRepoDict.__iter__c C s dd� | j � D �S )Nc s s | ]\}}|V qd S )Nr
)r; r< �vr
r
r rA � s z RepoDict.keys.<locals>.<genexpr>)r, )r r
r
r rI � s z
RepoDict.keysc C s dd� | j � D �S )Nc s s | ]\}}|V qd S )Nr
)r; r< rK r
r
r rA � s z"RepoDict.values.<locals>.<genexpr>)r, )r r
r
r r � s zRepoDict.values)�__name__�
__module__� __qualname__r r r r r1 r7 r9 r r r, rJ rI r �
__classcell__r
r
)rH r r ! s $
r )Z
__future__r Zdnf.exceptionsr Zdnf.i18nr Zdnf.utilr Z libdnf.confr! r: r( r r �dictr r
r
r
r �<module> s