CasperSecurity
o
���aT � @ s� d Z ddlZddlZddlZddlmZ ddlmZ ddlmZmZm Z m
Z
G dd� d�Z d d d
�Z e
d kr>e�e � � dS dS )
z@Simple command line interface to get/set password from a keyring� N� )�core)�backend)� set_keyring� get_password� set_password�delete_passwordc @ sl e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z e dd� �Z
edd� �ZdS )�CommandLineToolc C s� t �� | _| jjdddd dd� | jjdddd d d� | jjd
d d d
� | jjdd dd
� | jjdddd� | jjddd� | jjddd� d S )Nz-pz--keyring-path� keyring_pathzPath to the keyring backend)�dest�default�helpz-bz--keyring-backend�keyring_backendzName of the keyring backendz--list-backends�
store_truez List keyring backends and exit)�actionr
z --disablezDisable keyring and exit� operationz get|set|del�?)r
�nargs�service)r �username)�argparse�ArgumentParser�parser� add_argument��self� r �-/usr/lib/python3/dist-packages/keyring/cli.py�__init__ sH
������
�zCommandLineTool.__init__c C s| | j �|�}t| ��t|�� |jr t�� D ]}t|� qd S |jr(t �� d S | �
� | � � t | d| j
� �| j�}|� S )N�do_)r �
parse_args�vars�update�
list_backendsr �get_all_keyring�print�disabler � _check_args�_load_spec_backend�getattrr �
invalid_op)r �argv�args�k�methodr r r �run4 s
zCommandLineTool.runc C s: | j r| jd u s
| jd u r| j�| j � d�� d S d S d S )Nz requires service and username)r r r r �errorr r r r r' F s
�zCommandLineTool._check_argsc C s* t | j| j�}|d u rtd��t|� d S )Nr )r r r �
SystemExitr% �r �passwordr r r �do_getK s zCommandLineTool.do_getc C s0 | � d| j� d| j� d��}t| j| j|� d S )NzPassword for 'z' in 'z': )�input_passwordr r r r2 r r r �do_setQ s �zCommandLineTool.do_setc C s t | j| j� d S �N)r r r r r r r �do_delW s zCommandLineTool.do_delc C s | j �d� d S )Nz)Specify operation 'get', 'del', or 'set'.)r r0 r r r r r* Z s zCommandLineTool.invalid_opc
C sv | j d u rd S z| jrtj�d| j� tt�| j �� W d S tfy: } z| j �
d|� �� W Y d }~d S d }~ww )Nr z"Unable to load specified keyring: ) r r
�sys�path�insertr r � load_keyring� Exceptionr r0 )r �excr r r r( ] s
��z"CommandLineTool._load_spec_backendc C s | � � pt�|�S )z Retrieve password from input.)�pass_from_pipe�getpass)r �promptr r r r5 m s z CommandLineTool.input_passwordc C s t j�� }|o| �t j�� �S )z4Return password from pipe if not on TTY, else False.)r9 �stdin�isatty�strip_last_newline�read)�cls�is_piper r r r? q s z CommandLineTool.pass_from_pipec C s | d| � d� � S )z#Strip one last newline, if present.N�
)�endswith)�strr r r rD w s z"CommandLineTool.strip_last_newlineN)�__name__�
__module__� __qualname__r r/ r' r4 r6 r8 r* r( r5 � classmethodr? � staticmethodrD r r r r r
s &
r c C s& | du r t jdd� } t� }|�| �S )z Main command line interface.Nr )r9 r+ r r/ )r+ �clir r r �main} s
rQ �__main__r7 )�__doc__r@ r r9 � r r r r r r r rQ rK �exitr r r r �<module> s
p
�