3
�fZv � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dlZd dl m Z m
Z
d dlZd dlmZ d dlmZ d dlZy$d dlmZmZmZmZ dd� ZW nL ek
�r d d
� Zyd d lmZ W n ek
r� d d� ZY nX Y nX d
Z e e d�e d�g�Z e e d�e d�dg�Z e�rdd dl m!Z!m"Z"m#Z# d dl$m%Z%m&Z&m'Z' n(d dl!m!Z!m"Z"m#Z# d dl(m%Z%m&Z&m'Z' ej)j*G dd� dej+j,��Z-ej)j*G dd� de-��Z.dS )� )�print_functionN)�_�logger)�PY3)�ucd)�name�version�codename�os_release_attrc C s t � t� t� fS )N)r r r � r r �/usr/lib/python3.6/copr.py�linux_distribution. s r
c C s dS )N� r )r r r r r
1 s r
)r
c C sr t d��`} i }xF| D ]>}y$|j� jd�\}}|jd�||< W q tk
rP Y qX qW |d |d d fS Q R X d S )Nz/etc/os-release�=�"�NAMEZ
VERSION_ID)�open�rstrip�split�strip�
ValueError)Zos_release_fileZos_release_data�lineZos_release_keyZos_release_valuer r r r
7 s
�copr�yes�y�no�nr )� ConfigParser�
NoOptionError�NoSectionError)�urlopen� HTTPError�URLErrorc @ s e Zd ZdZdZdZdZdZdZed e Z d8Z
e d �Z d
Z
e d �Zed d
� �Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Ze d$d%� �Z d&d'� Z d(d)� Z d*d+� Z d,d-� Z!d.d/� Z"d0d1� Z#e d2d3� �Z$e d4d5� �Z%e d6d7� �Z&dS )9� CoprCommandz Copr plugin for DNF Nzcopr.fedorainfracloud.orgZfedoraZhttpsi� z://r z Interact with Copr repositories.Ta�
enable name/project [chroot]
disable name/project
remove name/project
list --installed/enabled/disabled
list --available-by-user=NAME
search project
Examples:
copr enable rhscl/perl516 epel-6-x86_64
copr enable ignatenkobrain/ocltoys
copr disable rhscl/perl516
copr remove rhscl/perl516
copr list --enabled
copr list --available-by-user=ignatenkobrain
copr search tests
c C s� | j ddddddddgd � | j� }|j d
d td �d
� |j dd td�d
� |j dd td�d
� |j ddtd�d� | j dtd�d� | j ddd� d S ) N�
subcommand� �help�enable�disable�remove�list�search)�nargs�choicesz --installed�
store_truez.List all installed Copr repositories (default))�actionr& z --enabledz List enabled Copr repositoriesz
--disabledz List disabled Copr repositoriesz--available-by-userr z-List available Copr repositories by user NAME)�metavarr& z--hubz(Specify an instance of Copr to work with)r& �arg�*)r, )� add_argumentZ add_mutually_exclusive_groupr )�parserZ list_optionr r r �
set_argparserp s zCoprCommand.set_argparserc C s� | j jjjdkrd S d }t� }g }| jjjd }tjj |t
d �}tjj |�r�|j |� |j
|� |jdd�r�|jdd�r�|jdd�}|jdd�}||g| _n
ddg| _xHtjtjj |t
d ��D ],}|jd�r�tjj |t
d |�} |j | � q�W g }
t| jj��r| jjd jd �}
t|
�d
k�rV| jj�rVtjtd �td � � tj jtd
���nL| jj �r�t|
�d
k�r�| j | _ | j | _ n t|
�d
k�r�|
d }n| jj}|�rH|�rHd | _ |j
t |dd�� | j!||dd �} | �rH| j!||d| j"�} | j!||d| j#�}
| | _ | d | | _ t$|
�| j#k�rH| j d|
7 _ | j d|
7 _ | j �s�d|k�rr|| _ | j"d | | _ n|jdd�d | _ || _ d S )Nr r z.conf�main� distribution�
releaseverFz.d�/� zError: z^specify Copr hub either with `--hub` or using `copr_hub/copr_username/copr_projectname` formatzmultiple hubs specifiedT)�reverse�hostname�protocol�portz://�:r% )%�cliZcommand�optsr �base�confZpluginconfpath�os�path�join� PLUGIN_CONF�isfile�append�readZ
has_option�get�
chroot_config�listdir�endswith�lenr1 r �hubr �criticalr �dnf�CliError�default_hostname�
copr_hostname� default_url�copr_url�sorted�_read_config_item�default_protocol� default_port�int)�selfZcopr_hubZcopr_plugin_configZ config_filesZ config_pathZdefault_config_filer7 r8 �filenameZ config_file�projectr< r= r> r r r � configure� sl
zCoprCommand.configurec C s* y |j ||�S ttfk
r$ |S X d S )N)rK r r )r] �configrP Zsection�defaultr r r rY � s z CoprCommand._read_config_itemc C s t jjdj|j� �� d S )Nz{0}
)�sys�stderr�write�formatr )r] �textr r r �_user_warning_before_prompt� s z'CoprCommand._user_warning_before_promptc
C s� | j jd }|dkr&| jjj| � dS |dkrl| j jrH| j| j j� d S | j| jj j
d | j j | j j � d S y| j j
d }W nL ttfk
r� tjtd�td� � | jjj| � tjjtd���Y nX y\| j j
d }t| j j
�dkr�tjjtd���|jd �| _t| j�d
k �r$tjjtd ���W n* tk
�rP | j� }|jd �| _Y nX |d k�rj| j|� d S |jd
�}t|�dk�r�tjtd�td� � tjjtd���n<t|�dk�r�|d }|d }n |d }|d }|d
| }dj | jj j | j | j |�|�}|dk�rn| j � td�}d
j!| j ||g�}dj |�} | j"|| � | j#||� tj$td�� | j%||� nr|dk�r�| j � | j&||� tj$td�� nD|dk�r�| j � | j'||� tj$td�� ntjjtd�j |���d S )Nr r& r* zError: z>exactly two additional parameters to copr command are requiredr% � zToo many arguments.�-r: zOBad format of optional chroot. The format is distribution-version-architecture.r+ r9 zEuse format `copr_username/copr_projectname` to reference copr projectzbad copr project formatz{0}/_copr:{1}:{2}:{3}.repor' a
Enabling a Copr repository. Please note that this repository is not part
of the main distribution, and quality may vary.
The Fedora Project does not exercise any power over the contents of
this repository beyond the rules outlined in the Copr FAQ at
<https://docs.pagure.org/copr.copr/user_documentation.html#what-i-can-build-in-copr>,
and packages are not held to any quality or security level.
Please do not file bug reports about these packages in Fedora
Bugzilla. In case of problems, contact the owner of this repository.
z!Do you really want to enable {0}?z Repository successfully enabled.r( z!Repository successfully disabled.r) z Repository successfully removed.zUnknown subcommand {}.)ri r: )(rA r$ r@ Z optparserZ
print_helpZavailable_by_user�_list_user_projects� _list_installed_repositoriesrB rC Zreposdir�enabledZdisabledr1 r �
IndexErrorr rQ r rR rS rO �
exceptions�Errorr � chroot_parts�
_guess_chroot�_searchrf � get_reposdirrU �_sanitize_username�
_need_rootrF � _ask_user�_download_repo�info�_runtime_deps_warning�
_disable_repo� _remove_repo)
r] r$ � project_name�chrootr_ �
copr_username�copr_projectname�
repo_filenamery �msgr r r �run� s�
zCoprCommand.runc C s� |j jd�d }tjd| j |�}| j| jko8tjd|�}tjd|�}| jjr`| rh| rhd S n|shd S tjd|�rxd S tjd|�r�d S |j } | r�|s�| r�|r�d S d}
tjd |�r�|jd
d �\} } }
}| d |
d | }n�tjd|��r2|jd d�}|j
d
d
�d jd
d�d } | d |d d |d }n.|jd d�}| j d |d d |d }d}
| �sn|d7 }|
�r||d7 }t |� |
S )Nr9 r% z_copr:Z_copr_z_copr:|^_copr_zcopr:.*:.*:.*:mlz
coprdep:.*Fzcopr:r? r: rj ri r Tz (disabled)z *���)
�repofiler �re�matchrU rW rV rA rP rm �rsplitrT �print)r] �repo_id�repo� enabled_only�
disabled_only� file_nameZ match_newZ match_oldZ match_anyrm �old_repor rU Z
copr_ownerZcopr_dirr� Z copr_namer r r �_list_repo_file8 sB zCoprCommand._list_repo_filec C sF d}x,| j jj� D ] \}}| j||||�rd}qW |rBttd�� d S )NFTz�* These coprs have repo file with an old format that contains no information about Copr hub - the default one was assumed. Re-enable the project to fix this.)rB �repos�itemsr� r� r )r] Z directoryr� r� r� r� r� r r r rl i s z(CoprCommand._list_installed_repositoriesc
C s� dj |�}| j| }| jj|dd�}ytj|j� �}W n* tk
r` tj j
t d�j |���Y nX | j |� t d�j |�}| j
|� xL|d D ]@}dj ||d �}|d p�t d
�} | jjjt|�| �}t|� q�W d S ) Nz!/api_3/project/list?ownername={0}zw+)�modez+Can't parse repositories for username '{}'.zList of {} coprsr� z
{0}/{1} : r � descriptionzNo description given)rf rW rB r �json�loadsrJ r rR ro rp r �_check_json_output�_print_match_section�output�
fmtKeyValFillr r� )
r] Z user_name�api_path�url�res�
json_parse� section_text�itemr� �descr r r rk s s"
z CoprCommand._list_user_projectsc
C s� dj |�}| j| }| jj|dd�}ytj|j� �}W n* tk
r` tj j
t d�j |���Y nX | j |� t d�j |�}| j
|� xJ|d D ]>}dj |d �}|d p�t d
�} | jjjt|�| �}t|� q�W d S ) Nz /api_3/project/search?query={}zw+)r� z Can't parse search for '{}'.z Matched: {}r� z{0} : Z full_namer� zNo description given.)rf rW rB r r� r� rJ r rR ro rp r r� r� r� r� r r� )
r] Zqueryr� r� r� r� r� r� r� r� r r r rs � s
zCoprCommand._searchc C s | j jj|�}t|� d S )N)rB r� Z
fmtSectionr� )r] rg Z formattedr r r r� � s z CoprCommand._print_match_sectionc C sj | j stjjd� d| _ tjjdj|j� �� | jj� rf| jjj sb| jj
j dj|�dj|�d� rfdS dS )N�
Fz{0}
z
{} [y/N]: z
{} [Y/n]: )r� Zdefaultyes_msgT) �
first_warningrc rd re rf r rB Z
_promptWantedrC Zassumenor� Z userconfirm)r] ry r� r r r �_ask_user_no_raise� s
z CoprCommand._ask_user_no_raisec C s | j ||�s tjjtd���d S )Nz Safe and good answer. Exiting.)r� rR ro rp r )r] ry r� r r r rw � s zCoprCommand._ask_userc C s t j� dkr tjjtd���d S )Nr z/This command has to be run under the root user.)rD �geteuidrR ro rp r )�clsr r r rv � s zCoprCommand._need_rootc s | j � � dks&� d dks&� d dkr,t� � | jjjd }t� fdd�dD ��r�d
� krbd | }n&d td
�krxd | }ndj� d |�}n�d� kr�tj d�}d� kr�dj|�}ndj� d |�}nPd� kr�tj d�}d� kr�dj|�}ndj� d |�}nd� d j
dd�d }|S ) z2 Guess which chroot is equivalent to this machine Nr Fr% Zbasearchc s g | ] }|� k�qS r r )�.0r )�distr r �
<listcomp>� s z-CoprCommand._guess_chroot.<locals>.<listcomp>�Fedora� Fedora LinuxZRawhidezfedora-rawhide-ZrawhideZ redhat_support_product_versionzfedora-{0}-{1}ZMageiaz%{distro_arch}ZCauldronzmageia-cauldron-{}zmageia-{0}-{1}ZopenSUSEz%{_target_cpu}Z
Tumbleweedzopensuse-tumbleweed-{}zopensuse-leap-{0}-{1}zepel-%s-x86_64�.)r� r� ) rL r
rB rC �
substitutions�anyr
rf �rpmZ expandMacror )r] Zdistarchr~ r )r� r rr � s,
zCoprCommand._guess_chrootc C s� dj | jd d � �}| jd }dj|||�}y*t| j| �}tjj|�rRtj|� W �n^ t k
�rl } z�|j
dkr�t d�j| j| |j
t |��}t
jj|��t d�}|jjd�} | �r>tj| �jd�}
tj|
�}
|t d �jdj | j�|�7 }|
jd
��r0|t d �d j d
d� |
d
D �� 7 }|t d�j|�7 }t
jj|��n |t d�j|�7 }t
jj|��W Y d d }~X nJ tk
�r� } z,t d�j| j| |jj�}t
jj|��W Y d d }~X nX |j� } | jd�} tj d| ��r�tjj | j j j d| dd � d �}| j| j k�rR|j!ddd�j!| j"d�j!ddd�j!dd�j!dd�}
tjj|
��rRtj|
� t#|d ��.}|j$| � x |j%� D ]} |j$| � �qrW W d Q R X tj&|t'j(t'j)B t'j*B t'j+B � d S ) Nrj r% z%/coprs/{0}/repo/{1}/dnf.repo?arch={2}i� z Request to {0} failed: {1} - {2}z+It wasn't possible to enable this project.
zCopr-Error-Datazutf-8z1Repository '{0}' does not exist in project '{1}'.zavailable chrootsz
Available repositories: z, c s s | ]}d j |�V qdS )z'{}'N)rf )r� �xr r r � <genexpr>� s z-CoprCommand._download_repo.<locals>.<genexpr>z�
If you want to enable a non-default repository, use the following command:
'dnf copr enable {0} <repository>'
But note that the installed repo file will likely need a manual modification.zProject {0} does not exist.z Failed to connect to {0}: {1}z\[copr:r ri z.repoz_copr:�_coprr r? Zgroup_�@�wbr� r� ���),rF rq rf r rW rD rE �existsr) r! �coder �strrR ro rp ZheadersrK �base64Z b64decode�decoder� r� r" �reason�strerror�readliner� r� rB rC rt rV �replacerU r re � readlines�chmod�stat�S_IRUSR�S_IWUSR�S_IRGRP�S_IROTH)r] r} r� Z short_chrootZarchr� Zresponse�eZ error_msgZ
error_dataZerror_data_decodedZ
first_liner Zold_repo_filename�fr r r rx � sX
$
zCoprCommand._download_repoc s� | j jdd� | j j� | j| j|�|��g }x(�jj� D ]}|jd�rJq:|j|� q:W |s`dS t d�}t
j d�� |j dj
� �fdd �|D ���}| j|t d
��s�x,|D ]$}| j jj�j|| j jjd d i� q�W dS )
a,
In addition to the main copr repo (that has repo ID prefixed with
`copr:`), the repofile might contain additional repositories that
serve as runtime dependencies. This method informs the user about
the additional repos and provides an option to disable them.
T)r� zcopr:Na� Maintainer of the enabled Copr repository decided to make
it dependent on other repositories. Such repositories are
usually necessary for successful installation of RPMs from
the main Copr repository (they provide runtime dependencies).
Be aware that the note about quality and bug-reporting
above applies here too, Fedora Project doesn't control the
content. Please review the list:
{0}
These repositories have been enabled automatically.r% z
c s* g | ]"}d j t� �|�jj|d�d��qS )z){num:2}. [{repoid}]
baseurl={baseurl}�baseurl)Znum�repoidr� )rf �next�cfgZgetValue)r� r� )�counterr� r r r� 9 s z5CoprCommand._runtime_deps_warning.<locals>.<listcomp>z!Do you want to keep them enabled?rm �0)rB �resetZread_all_repos�_get_copr_reporu r� �sections�
startswithrI r � itertools�countrf rF r� rC �write_raw_configfiler� r� )r] r r� Z runtime_depsr� ry Zdepr )r� r� r rz s*
z!CoprCommand._runtime_deps_warningc C s� dj | jjdd�d | j|�|�}|| jjkr�dj | j|�|� }}|| jjkr�d| jj| jkr�| jj| jjd�d }y.|jdd�d jdd�d }|| jkr�d S W q� tk
r� Y q�X nd S | jj| S )
Nzcopr:{0}:{1}:{2}r? r% r z{0}-{1}r� r9 ri r� ) rf rU r� ru rB r� r� r rn )r] r r� r� r� rU r r r r� F s
zCoprCommand._get_copr_repoc C st | j ||�}|s,tjjtdj| j||����ytj|j � W n2 t
k
rn } ztjjt |���W Y d d }~X nX d S )Nz&Failed to remove copr repo {0}/{1}/{2}) r� rR ro rp r rf rU rD r) r� �OSErrorr� )r] r r� r� r� r r r r| \ s zCoprCommand._remove_repoc C sd | j ||�}|d kr,tjjtdj||����x2|jj� D ]$}| jj j
|j || jj j ddi� q8W d S )Nz!Failed to disable copr repo {}/{}rm r� )
r� rR ro rp r rf r� r� rB rC r� r� r� )r] r r� r� r� r r r r{ h s zCoprCommand._disable_repoc C s< yt j|j� �}W n$ tk
r6 tjjtd�� dS X |S )z� Wrapper around response from server
check data and print nice error in case of some error (and return None)
otherwise return json object.
z Unknown response from server.N)r� r� rJ r rR r@ rS r )r� r� r� r r r � _get_datat s zCoprCommand._get_datac C s" d|kr t jjdj|d ���d S )N�errorz{})rR ro rp rf )r� Zjson_objr r r r� � s z CoprCommand._check_json_outputc C s& |d dkr dj |dd � �S |S d S )Nr r� zgroup_{}r% )rf )r� r r r r ru � s z CoprCommand._sanitize_username)r )'�__name__�
__module__� __qualname__�__doc__rL rT Z default_hubrZ r[ rV �aliasesr �summaryr� �usage� staticmethodr5 r` rY rh r� r� rl rk rs r� r� rw � classmethodrv rr rx rz r� r| r{ r� r� ru r r r r r# P sD L_1
%82 r# c @ sD e Zd ZdZdZed�ZdZdd� Zdd� Z e
d d
� �Z d d � Z d
S )�PlaygroundCommandz Playground plugin for DNF �
playgroundz$Interact with Playground repository.z [enable|disable|upgrade]c C s0 | j � | jtd�td�� dj| j�}| jj|dd�}| j|�}|j� |d dkrft j
j td���x�|d D ]�}d
j|d |d �}d
j| jj j
|jdd��}yj||d kr�wpdj| j||�}| jj|dd�}| j|�}|j� |o�d|ko�|d dk�r
| j||� W qp t jjk
�r& Y qpX qpW d S )Nz!Enabling a Playground repository.zDo you want to continue?z{0}/api/playground/list/zw+)r� r� �okz Unknown response from server.r� z{0}/{1}ZusernameZcoprnamez{}/_playground_{}.repor9 rj Zchrootsz {0}/api/coprs/{1}/detail/{2}/)rv rw r rf rW rB r r� �closerR r@ rS rC rt r� rx ro rp ) r] r~ Zapi_urlr� r� r� r} r� Zoutput2r r r � _cmd_enable� s8
z PlaygroundCommand._cmd_enablec C s6 | j � x(tjdj| jjj��D ]}| j|� q W d S )Nz{}/_playground_*.repo)rv �globrf rB rC rt r| )r] r� r r r � _cmd_disable� s z PlaygroundCommand._cmd_disablec C s | j dddddgd� d S )Nr$ r% r' r( �upgrade)r, r- )r3 )r4 r r r r5 � s z PlaygroundCommand.set_argparserc C s� t jjd��| jjd }| j� }|dkrB| j|� tjt d�� n`|dkrb| j
� tjt d�� n@|dkr�| j
� | j|� tjt d�� nt jjt d �j |���d S )
Nz%Playground is temporarily unsupportedr r' z-Playground repositories successfully enabled.r( z.Playground repositories successfully disabled.r� z-Playground repositories successfully updated.zUnknown subcommand {}.) rR ro rp rA r$ rr r� r ry r r� rf )r] r$ r~ r r r r� � s
zPlaygroundCommand.runN)r� )
r� r� r� r� r� r r� r� r� r� r� r5 r� r r r r r� � s r� )/Z
__future__r r� r� r� rD r� Zshutilr� rc r� Zdnfpluginscorer r rR Z
dnf.pycompr Zdnf.i18nr r� Zdistror r r r
r
� ImportError�platformrG �setZYESZNOZ configparserr r r Zurllib.requestr r! r" Zurllib2ZpluginZregister_commandr@ ZCommandr# r� r r r r �<module> sP
B