File: //lib/python3.6/site-packages/awscli/customizations/codeartifact/__pycache__/login.cpython-36.pyc
3
�T_K9 � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dlmZ d dl m
Z
d dl m Z d dl
mZmZmZ d dlmZ d dlmZ d d
� ZG d d � d e�ZG d
d� de�ZG dd� de�ZG dd� de�ZG dd� de�Z dS )� N)�datetime)�tzutc)�
relativedelta)�parse_timestamp)�urlparse�RawConfigParser�StringIO)�utils)� BasicCommandc C s~ g }d}xfd D ]^}t | |�}|dkr^|r2|jd� |jt|�� |j|d krX|d d � n|� |rdP |dk}qW d
j|�}|S )
NF�years�months�days�hours�minutesr �and� � )r r r
r r ���)�getattr�append�str�join)� remaining�valuesZprev_non_zero_attr�attr�value�message� r �/usr/lib/python3.6/login.py� get_relative_expiration_time s
r c @ sF e Zd Zddd�Zddd�Zdd� Zd d
� Zdd d �Zed
d� �Z dS )� BaseLoginNc C s" || _ || _|| _|| _|| _d S )N)�
auth_token�
expiration�repository_endpoint�subprocess_utils� namespace)�selfr! r" r# r$ r% r r r �__init__& s
zBaseLogin.__init__Fc C s t d��d S )Nzlogin())�NotImplementedError)r& �dry_runr r r �login. s zBaseLogin.loginc C s@ x:|D ]2}t jjdj|�� t jjtj� t jjtj� qW d S )Nr )�sys�stdout�writer �os�linesep)r& �tool�commands�commandr r r �_dry_run_commands1 s
zBaseLogin._dry_run_commandsc C sr t | jtjt� ��t dd� }t|�}tjjdj || j
�� tjjt j � tjjdj || j�� tjjt j � d S )N� )ZsecondszASuccessfully configured {} to use AWS CodeArtifact repository {} zLogin expires in {} at {})
r r" r Znowr r r+ r, r- �formatr# r. r/ )r& r0 r Zexpiration_messager r r �_write_success_message7 s
z BaseLogin._write_success_messagec C s� |r| j ||� d S xj|D ]b}y | jj|| jj| jjd� W q tk
rz } z"|jtjkrftd| ��|�W Y d d }~X qX qW | j|� d S )N)r, �stderrz-%s was not found. Please verify installation.) r3 r$ Z
check_call�PIPE�OSError�errno�ENOENT�
ValueErrorr6 )r& r0 r1 r) r2 Zexr r r �
_run_commandsG s
zBaseLogin._run_commandsc K s t d��d S )Nzget_commands())r( )�cls�endpointr! �kwargsr r r � get_commands\ s zBaseLogin.get_commands)N)F)F)
�__name__�
__module__� __qualname__r' r* r3 r6 r= � classmethodrA r r r r r $ s
r c @ sF e Zd Zej� j� dkr dndZd dd�Zedd� �Z ed d
� �Z
d S )
�NpmLoginZwindowsznpm.cmd�npmFc C s2 | j | j�}| j| j| j|d�}| jd||� d S )N)�scoperG )� get_scoper% rA r# r! r= )r& r) rH r1 r r r r* g s
zNpmLogin.loginc C sF t jd�}|d kr|S |jd�r&|}n
dj|�}|j|�sBtd��|S )Nz^(@[a-z0-9-~][a-z0-9-._~]*)�@z@{}zZInvalid scope name, scope must contain URL-safe characters, no leading dots or underscores)�re�compile�
startswithr5 �matchr< )r> r% Zvalid_scope_namerH r r r rI p s
zNpmLogin.get_scopec
K s� g }|j d�}|r dj|�nd}|j| jdd||g� tj|�}dj|j|j�}|j| jdd|dg� dj|j|j�} |j| jdd| |g� |S ) NrH z {}:registry�registry�config�setz//{}{}:always-auth�truez//{}{}:_authToken)�getr5 r �NPM_CMDr �urlsplit�netloc�path)
r> r? r! r@ r1 rH rO �repo_uriZalways_auth_configZauth_token_configr r r rA � s
zNpmLogin.get_commandsN)F) rB rC rD �platform�system�lowerrT r* rE rI rA r r r r rF a s
rF c @ s& e Zd ZdZddd�Zedd� �ZdS ) �PipLoginz1{scheme}://aws:{auth_token}@{netloc}{path}simple/Fc C s" | j | j| j�}| jd||� d S )N�pip)rA r# r! r= )r& r) r1 r r r r* � s zPipLogin.loginc K s4 t j|�}| jj|j||j|jd�}dddd|ggS )N)�schemer! rV rW r] rP rQ zglobal.index-url)r rU �PIP_INDEX_URL_FMTr5 r^ rV rW )r> r? r! r@ rX Z
pip_index_urlr r r rA � s
zPipLogin.get_commandsN)F)rB rC rD r_ r* rE rA r r r r r\ � s
r\ c sD e Zd ZdZd � fdd� Zedd� �Zd
dd �Zed
d � �Z� Z S )�
TwineLoginz�[distutils]
index-servers=
pypi
codeartifact
[codeartifact]
repository: {repository_endpoint}
username: aws
password: {auth_token}Nc s0 |d kr| j � }|| _tt| �j||||� d S )N)�get_pypi_rc_path� pypi_rc_path�superr` r' )r& r! r" r# r$ rb )� __class__r r r' � s
zTwineLogin.__init__c K sP d|kst d��|d }| jj||d�}t� }tjj|��r y�|j|� |jdd�}dd� |j d�D �}d |kr�|j
d � |j ddddj |� � d |j
� kr�|jd � |j d d
|� |j d d d � |j d d
|� W nF tk
�r } z(tjjd| � tjjtj� | �W Y d d } ~ X nX n|jt|�� t� }
|j|
� |
j� } |
j� | S )Nrb z pypi_rc_path must be provided.)r# r! Z distutilsz
index-serversc S s g | ]}|j � d kr|j � �qS )� )�strip)�.0Zserverr r r �
<listcomp>� s z+TwineLogin.get_commands.<locals>.<listcomp>�
� codeartifact�
repositoryZusernameZawsZpasswordz%s is in an invalid state.)�AssertionError�DEFAULT_PYPI_RC_FMTr5 r r. rW �exists�readrS �splitr rQ r ZsectionsZ add_section� Exceptionr+ r, r- r/ Zreadfpr �getvalue�close) r> r? r! r@ rb Zdefault_pypi_rcZpypi_rcZ
index_serversZservers�eZpypi_rc_stream� pypi_rc_strr r r rA � s>
zTwineLogin.get_commandsFc C s� | j | j| j| jd�}|r~tjjd� tjjtj� tjjd| j � tjjtj� tjjtj� tjj|� tjjtj� n,t | jd��}|j|� W d Q R X | j
d� d S )N)rb z.Dryrun mode is enabled, not writing to pypirc.z(%s would have been set to the following:zw+�twine) rA r# r! rb r+ r, r- r. r/ �openr6 )r& r) ru �fpr r r r* s
zTwineLogin.loginc C s t jjt jjd�d�S )N�~z.pypirc)r. rW r �
expanduser)r> r r r ra + s zTwineLogin.get_pypi_rc_path)N)F)
rB rC rD rm r' rE rA r* ra �
__classcell__r r )rd r r` � s
8
r` c @ s� e Zd ZdZdedd�dedd�dedd�d�ZdZd Z d
d e
ej � �dd �d
ddd�dddd�dddd�ddddd�dddd�ddd ddd �gZ d d � Z
d d!� Zd"d#� Zd$d%� Zd&S )'�CodeArtifactLoginz>Log in to the idiomatic tool for the requested package format.rG T)�package_format� login_cls�namespace_supportZpypiF)rG r] rv r* z�Sets up the idiomatic tool for your package format to use your CodeArtifact repository. Your login information is valid for up to 12 hours after which you must login again.r0 z1The tool you want to connect with your repository)�name� help_text�choices�required�domainz Your CodeArtifact domain name)r� r� r� z domain-ownerz5The AWS account ID that owns your CodeArtifact domainr% z0Associates a namespace with your repository toolzduration-secondsZintegerz9The time, in seconds, that the login information is valid)r� Z
cli_type_namer� r� rk z!Your CodeArtifact repository namezdry-run�
store_truez�Only print the commands that would be executed to connect your tool with your repository without making any changes to your configuration)r� �actionr� r� �defaultc C s4 | j | d }| r*|jr*tdj|���n|jS d S )Nr z,Argument --namespace is not supported for {})�TOOL_MAPr% r< r5 )r& r0 � parsed_argsZnamespace_compatibler r r �_get_namespace{ s
z CodeArtifactLogin._get_namespacec C s4 |j |j|d�}|jr |j|d< |jf |�}|d S )N)r� rk r5 � domainOwnerZrepositoryEndpoint)r� rk � domain_ownerZget_repository_endpoint)r& �codeartifact_clientr� r} r@ Z get_repository_endpoint_responser r r �_get_repository_endpoint� s
z*CodeArtifactLogin._get_repository_endpointc C s: d|j i}|jr|j|d< |jr*|j|d< |jf |�}|S )Nr� r� ZdurationSeconds)r� r� Zduration_secondsZget_authorization_token)r& r� r� r@ Z get_authorization_token_responser r r �_get_authorization_token� s
z*CodeArtifactLogin._get_authorization_tokenc C s� |j j� }| j| d }tj| jd|�}| j||�}| j|||�}| j||�}|d } t |d �}
| j| d | |
|t
|�} | j |j � dS )Nr} rj ZauthorizationTokenr" r~ r )
r0 r[ r� � cli_utilsZ!create_client_from_parsed_globalsZ_sessionr� r� r� r �
subprocessr* r) ) r& r� Zparsed_globalsr0 r} r� Zauth_token_resr# r% r! r" r* r r r � _run_main� s
zCodeArtifactLogin._run_mainN)rB rC rD �__doc__rF r\ r` r� �NAMEZ DESCRIPTION�list�keysZ ARG_TABLEr� r� r� r� r r r r r| 0 sR
r| ) r: r. rY r+ r� rK r Z dateutil.tzr Zdateutil.relativedeltar Zbotocore.utilsr Z
awscli.compatr r r Zawscli.customizationsr r� Z awscli.customizations.commandsr
r �objectr rF r\ r` r| r r r r �<module> s$ =Gq