Edit File: obj.cpython-36.pyc
3 گa~Z � @ sp d Z ddlZddlZddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z dd lm Z dd lm Z ddl Z ddlZ dd lmZ dd lmZ dd lmZ ddlmZ ddlmZ ddlmZ eje�ZdZ dZ dd d Z G dd� d�Z e � Z e d�Z!e j"j#ej$�G dd� d��Z%e j"j#ej$�G dd � d ��Z&e e%e&f d �d d �Z'e e%e&f dd �d!d"�Z(dS )#zFThis modules define the actual display implementations used in Certbot� N)�Any)�Iterable)�List)�Optional)�TextIO)�Tuple)�TypeVar)�Union)�errors)� interfaces)� constants)� completer)�util)�os�okZcancelz- �' �-c @ s e Zd Zdd�dd�ZdS )�_DisplayServiceN)�returnc C s d | _ d S )N)�display)�self� r �/usr/lib/python3.6/obj.py�__init__, s z_DisplayService.__init__)�__name__� __module__� __qualname__r r r r r r + s r �Tc s� e Zd ZdZeedd�� fdd� Zd*eeeeedd�d d �Zd+ee e e eef e e f e e e e e e e e e e eee ee f d � d d �Zd,ee e e e eee eef d�dd�Zd-eeee e e e eeed�dd�Zd.ee e e e e e e eee ee e f d�dd�Zee e e e ee e d�dd�Zeed �d d �Zd/ee e e e eee eef d�d d �Zee ee f e e e e d!�d"d#�Zee e e eef e e f dd$�d%d&�Ze e ee f d'�d(d)�Z� ZS )0� FileDisplayzFile-based display.N)�outfile�force_interactiver c s t � j� || _|| _d| _d S )NF)�superr r r �skipped_interaction)r r r )� __class__r r r < s zFileDisplay.__init__TF)�message�pause�wrapr �decorater c C s~ |rt j|�}tjd|� | jj|r(dndd |r6dnd jtjt |d�� | jj � |rz| j |�rpt j d� n tjd� d S ) a Displays a notification and waits for user acceptance. :param str message: Message to display :param bool pause: Whether or not the program should pause for the user's confirmation :param bool wrap: Whether or not the application should wrap text :param bool force_interactive: True if it's safe to prompt the user because it won't cause any workflow regressions :param bool decorate: Whether to surround the message with a decorated frame zNotifying user: %sz{line}{frame}{line}� z {msg}{line}z {frame}{line})�line�frame�msgzPress Enter to Continuez!Not pausing for user confirmationN) r � wrap_lines�logger�debugr �write�formatr �linesep� SIDE_FRAME�flush� _can_interact�input_with_timeout)r r$ r% r& r r' r r r � notificationB s zFileDisplay.notification) r$ �choices�ok_label� cancel_label� help_label�default�cli_flagr � unused_kwargsr c K sJ | j ||||�} | dk r t| fS | j||� | jt|��\} } | | d fS )aC Display a menu. .. todo:: This doesn't enable the help label/button (I wasn't sold on any interface I came up with for this). It would be a nice feature :param str message: title of menu :param choices: Menu lines, len must be > 0 :type choices: list of tuples (tag, item) or list of descriptions (tags will be enumerated) :param default: default value to return (if one exists) :param str cli_flag: option used to set this value with the CLI :param bool force_interactive: True if it's safe to prompt the user because it won't cause any workflow regressions :returns: tuple of (`code`, `index`) where `code` - str display exit code `index` - int index of the user's selection :rtype: tuple N� )�_return_default�OK� _print_menu�_get_valid_int_ans�len) r r$ r7 r8 r9 r: r; r< r r= �return_default�code� selectionr r r �menuc s zFileDisplay.menu)r$ r; r< r r= r c K sT | j ||||�}|dk r t|fS tjd| �d }tj|�}|dkrLtdfS t|fS )a Accept input from the user. :param str message: message to display to the user :param default: default value to return (if one exists) :param str cli_flag: option used to set this value with the CLI :param bool force_interactive: True if it's safe to prompt the user because it won't cause any workflow regressions :returns: tuple of (`code`, `input`) where `code` - str display exit code `input` - str of the user's input :rtype: tuple Nz%s (Enter 'c' to cancel):� �c�Cz-1)rI rJ )r? r@ r r, r5 �CANCEL)r r$ r; r< r r= rD �ansr r r �input� s zFileDisplay.input�Yes�No)r$ � yes_label�no_labelr; r< r r= r c K s� | j ||||�}|dk r |S tj|�}| jjdjtjttj |d�� | jj � xttj djtj |�tj |�d��} | j |d j � �s�| j |d j� �r�dS | j |d j � �s�| j |d j� �rRdS qRW dS ) a� Query the user with a yes/no question. Yes and No label must begin with different letters, and must contain at least one letter each. :param str message: question for the user :param str yes_label: Label of the "Yes" parameter :param str no_label: Label of the "No" parameter :param default: default value to return (if one exists) :param str cli_flag: option used to set this value with the CLI :param bool force_interactive: True if it's safe to prompt the user because it won't cause any workflow regressions :returns: True for "Yes", False for "No" :rtype: bool Nz{0}{frame}{msg}{0}{frame})r* r+ z {yes}/{no}: )�yes�nor TF)r? r r, r r/ r0 r r1 r2 r3 r5 Zparens_around_char� startswith�lower�upper) r r$ rP rQ r; r< r r= rD rL r r r �yesno� s" zFileDisplay.yesno)r$ �tagsr; r< r r= r c K s� | j ||||�}|dk r t|fS x�| j||� | jddd�\}} |tkr�| j� srdjdd� tdt|�d �D ��} tj | �} | j | |�} | r�|| fS | j j d t j � | j j� q"|g fS q"W dS ) aC Display a checklist. :param str message: Message to display to user :param list tags: `str` tags to select, len(tags) > 0 :param default: default value to return (if one exists) :param str cli_flag: option used to set this value with the CLI :param bool force_interactive: True if it's safe to prompt the user because it won't cause any workflow regressions :returns: tuple of (`code`, `tags`) where `code` - str display exit code `tags` - list of selected tags :rtype: tuple NzrSelect the appropriate numbers separated by commas and/or spaces, or leave input blank to select all options shownT)r rH c s s | ]}t |�V qd S )N)�str)�.0�xr r r � <genexpr>� s z(FileDisplay.checklist.<locals>.<genexpr>r> z!** Error - Invalid selection **%s)r? r@ rA rM �strip�join�rangerC r Zseparate_list_input�_scrub_checklist_inputr r/ r r1 r3 ) r r$ rX r; r< r r= rD rE rL �indicesZ selected_tagsr r r � checklist� s$ " zFileDisplay.checklist)�promptr; r< r r c C sN | j |�rdS |dkr<dj|�}|r2|dj|�7 }tj|��tjd||� |S )a� Should we return the default instead of prompting the user? :param str prompt: prompt for the user :param T default: default answer to prompt :param str cli_flag: command line option for setting an answer to this question :param bool force_interactive: if interactivity is forced :returns: The default value if we should return it else `None` :rtype: T or `None` Nz-Unable to get an answer for the question: {0}zA You can provide an answer on the command line with the {0} flag.z-Falling back to default %s for the prompt: %s)r4 r0 r �Errorr- r. )r rc r; r< r r+ r r r r? � s zFileDisplay._return_default)r r c C s@ | j s |s tjj� r"| jj� r"dS | js<tjdtj � d| _dS )z�Can we safely interact with the user? :param bool force_interactive: if interactivity is forced :returns: True if the display can interact with the user :rtype: bool Tz�Skipped user interaction because Certbot doesn't appear to be running in a terminal. You should probably include --non-interactive or %s on the command line.F) r �sys�stdin�isattyr r" r- Zwarningr ZFORCE_INTERACTIVE_FLAG)r r r r r r4 s zFileDisplay._can_interactc K s$ t j� � | j||||�S Q R X dS )a Display a directory selection screen. :param str message: prompt to give the user :param default: default value to return (if one exists) :param str cli_flag: option used to set this value with the CLI :param bool force_interactive: True if it's safe to prompt the user because it won't cause any workflow regressions :returns: tuple of the form (`code`, `string`) where `code` - display exit code `string` - input entered by the user N)r Z CompleterrM )r r$ r; r< r r= r r r �directory_select- s z FileDisplay.directory_select)ra rX r c sl ydd� |D �}W n t k r& g S X tt|��}x$|D ] }|dk sR|t� �kr:g S q:W � fdd�|D �S )z�Validate input and transform indices to appropriate tags. :param list indices: input :param list tags: Original tags of the checklist :returns: valid tags the user selected :rtype: :class:`list` of :class:`str` c S s g | ] }t |��qS r )�int)rZ �indexr r r � <listcomp>M s z6FileDisplay._scrub_checklist_input.<locals>.<listcomp>r> c s g | ]}� |d �qS )r> r )rZ rj )rX r r rk Y s )� ValueError�list�setrC )r ra rX Z indices_intrj r )rX r r` @ s z"FileDisplay._scrub_checklist_input)r$ r7 r c C s� |r t |d t�r dd� |D �}| jjdjtj|d�� | jjttj � xDt|d�D ]6\}}dj||d�}| jjt j |�� | jjtj� qVW | jjttj � | jj � d S ) z�Print a menu on the screen. :param str message: title of menu :param choices: Menu lines :type choices: list of tuples (tag, item) or list of descriptions (tags will be enumerated) r c S s g | ]}d |d |d f �qS )z%s - %sr r> r )rZ rI r r r rk g s z+FileDisplay._print_menu.<locals>.<listcomp>z{new}{msg}{new})�newr+ r> z {num}: {desc})Znum�descN) � isinstance�tupler r/ r0 r r1 r2 � enumerater r, r3 )r r$ r7 �irp r+ r r r rA [ s zFileDisplay._print_menu)�max_r c C s� d}|dkrdj |d�}nd}x�|dk r�tj|�}|jd�sF|jd�rNtd fS y$t|�}|dk sh||krpd }t�W q tk r� | jjdj t j �� | jj � Y q X q W t |fS ) a5 Get a numerical selection. :param int max: The maximum entry (len of choices), must be positive :returns: tuple of the form (`code`, `selection`) where `code` - str display exit code ('ok' or cancel') `selection` - int user's selection :rtype: tuple r> zMSelect the appropriate number [1-{max_}] then [enter] (press 'c' to cancel): )ru z@Press 1 [enter] to confirm the selection (press 'c' to cancel): rI rJ z{0}** Invalid input **{0}���rv rv ) r0 r r5 rT rK ri rl r r/ r r1 r3 r@ )r ru rF Z input_msgrL r r r rB y s& z FileDisplay._get_valid_int_ans)TTFT)NNNNNF)NNF)rN rO NNF)NNF)NNF) r r r �__doc__r �boolr rY r6 r r r r ri r rG rM rW rb r r? r4 rh r r` rA rB � __classcell__r r )r# r r 7 s6 ! > * &* ) ( r c sh e Zd ZdZeeedd�� fdd� Zd eee ee j d�dd �Z d ee e e edd �d d�Z d eeeeeef ee f ee ee ee ee ee eeeef d� dd�Zd eee ee eeeef d�dd�Zd!eee ee ee ee ee d�dd�Zd"eee eee ee eeeee f d�dd�Zd#eee ee eeeef d�dd �Z� ZS )$�NoninteractiveDisplayzKA display utility implementation that never asks for interactive user inputN)r � unused_argsr= r c s t � j� || _d S )N)r! r r )r r r{ r= )r# r r r � s z NoninteractiveDisplay.__init__r( )r$ r< �extrar c C s8 d}||7 }|r |d| 7 }|r.|dj |�7 }tj|�S )zNReturn error to raise in case of an attempt to interact in noninteractive modez<Missing command line flag or config entry for this setting: � z& (You can set this with the {0} flag))r0 r �MissingCommandlineFlag)r r$ r< r| r+ r r r �_interaction_fail� s z'NoninteractiveDisplay._interaction_failFT)r$ r% r& r' r= r c K sZ |rt j|�}tjd|� | jj|r(dndd |r6dnd jtjt |d�� | jj � dS )ae Displays a notification without waiting for user acceptance. :param str message: Message to display to stdout :param bool pause: The NoninteractiveDisplay waits for no keyboard :param bool wrap: Whether or not the application should wrap text :param bool decorate: Whether to apply a decorated frame to the message zNotifying user: %sz{line}{frame}{line}r( z {msg}{line}z {frame}{line})r) r* r+ N) r r, r- r. r r/ r0 r r1 r2 r3 )r r$ r% r& r' r= r r r r6 � s z"NoninteractiveDisplay.notification) r$ r7 r8 r9 r: r; r< r= r c K s& |dkr | j ||dt|� ��t|fS )a_ Avoid displaying a menu. :param str message: title of menu :param choices: Menu lines, len must be > 0 :type choices: list of tuples (tag, item) or list of descriptions (tags will be enumerated) :param int default: the default choice :param dict kwargs: absorbs various irrelevant labelling arguments :returns: tuple of (`code`, `index`) where `code` - str display exit code `index` - int index of the user's selection :rtype: tuple :raises errors.MissingCommandlineFlag: if there was no default Nz Choices: )r �reprr@ ) r r$ r7 r8 r9 r: r; r< r= r r r rG � s zNoninteractiveDisplay.menu)r$ r; r< r= r c K s |dkr| j ||��t|fS )aK Accept input from the user. :param str message: message to display to the user :returns: tuple of (`code`, `input`) where `code` - str display exit code `input` - str of the user's input :rtype: tuple :raises errors.MissingCommandlineFlag: if there was no default N)r r@ )r r$ r; r< r= r r r rM � s zNoninteractiveDisplay.input)r$ rP rQ r; r< r= r c K s |dkr| j ||��|S )a+ Decide Yes or No, without asking anybody :param str message: question for the user :param dict kwargs: absorbs yes_label, no_label :raises errors.MissingCommandlineFlag: if there was no default :returns: True for "Yes", False for "No" :rtype: bool N)r )r r$ rP rQ r; r< r= r r r rW � s zNoninteractiveDisplay.yesno)r$ rX r; r< r= r c K s( |dkr | j ||dj|�d ��t|fS )aj Display a checklist. :param str message: Message to display to user :param list tags: `str` tags to select, len(tags) > 0 :param dict kwargs: absorbs default_status arg :returns: tuple of (`code`, `tags`) where `code` - str display exit code `tags` - list of selected tags :rtype: tuple Nz? �?)r r^ r@ )r r$ rX r; r< r= r r r rb s z NoninteractiveDisplay.checklistc K s | j |||�S )a� Simulate prompting the user for a directory. This function returns default if it is not ``None``, otherwise, an exception is raised explaining the problem. If cli_flag is not ``None``, the error message will include the flag that can be used to set this value with the CLI. :param str message: prompt to give the user :param default: default value to return (if one exists) :param str cli_flag: option used to set this value with the CLI :returns: tuple of the form (`code`, `string`) where `code` - int display exit code `string` - input entered by the user )rM )r r$ r; r< r= r r r rh s z&NoninteractiveDisplay.directory_select)r( )FTT)NNNNN)NN)NNNN)NN)NN)r r r rw r r r rY r r r~ r rx r6 r r r ri rG rM rW r rb rh ry r r )r# r rz � s" T 6 rz )r c C s t jstd��t jS )z�Get the display utility. :return: the display utility :rtype: Union[FileDisplay, NoninteractiveDisplay] :raise: ValueError if the display utility is not configured yet. zlThis function was called too early in Certbot's execution as the display utility hasn't been configured yet.)�_SERVICEr rl r r r r � get_display- s r� )r r c C s t jj| tj� | t_dS )zqSet the display service. :param Union[FileDisplay, NoninteractiveDisplay] display: the display service N)�zopeZ componentZprovideUtilityr �IDisplayr� r )r r r r � set_display; s r� ))rw Zloggingre Ztypingr r r r r r r r Zzope.componentr� Zzope.interfaceZcertbotr r Zcertbot._internalr Zcertbot._internal.displayr r Zcertbot.compatr Z getLoggerr r- r@ rK r2 r r� r Z interfaceZ implementerr� r rz r� r� r r r r �<module> s@ k
Back to File Manager