o
j6hy @ s d Z ddlmZ ddlZddlZddlmZ ddlmZ G dd de ej
dZ G d d
d
e Z G d d d e Z
G d
d dejZdS )a Customized ArgumentParser.
The default behvaior of argparse.ArgumentParser's parse_args() method is to
exit with a SystemExit exception in the following cases:
1. When the user requests a help message (with the --help or -h flags), or
2. When there's a parsing error (e.g. missing required flags or mistyped flags)
To make the errors more user-friendly, this class customizes
argparse.ArgumentParser and raises either ParserNormalExit for (1) or
ParserError for (2); this way the caller has control over how to display them
to the user.
) annotationsN)Sequence) ipython_envc sN e Zd ZdZd fdd Zdd
d Zd d
ZdddZej dddZ
Z S )_ParserBaseExceptionz7Base class for parser exceptions including normal exit.msgs
Sequence[str]c s0 t jd|g|R i | || _d | _d S )N )super__init__join_msgs _ipython_env)selfr argskwargs __class__ {/var/www/html/chefvision.cloud.itp360.com/venv/lib/python3.10/site-packages/google/generativeai/notebook/argument_parser.pyr
' s
z _ParserBaseException.__init__envipython_env.IPythonEnvreturnNonec C s
|| _ d S N)r
r r r r r set_ipython_env, s
z$_ParserBaseException.set_ipython_envc C s | | j d S r )displayr
r r r r _ipython_display_/ s z&_ParserBaseException._ipython_display_c C s | j S r )r r r r r r 2 s z_ParserBaseException.msgs ipython_env.IPythonEnv | Nonec C s dS )z-Display this exception on an IPython console.Nr r r r r r 5 s z _ParserBaseException.display)r r )r r r r )r r r r r r ) __name__
__module__ __qualname____doc__r
r r r abcabstractmethodr
__classcell__r r r r r $ s
r ) metaclassc @ e Zd ZdZd ddZdS )
ParserNormalExitzzException thrown when the parser exits normally.
This is usually thrown when the user requests the help message.
r r r r c C s | j D ]}t| qd S r )r printr r msgr r r r D s
zParserNormalExit.displayNr r! r" r# r$ r r r r r r* > s r* c @ r) )
ParserErrorz(Exception thrown when there is an error.r r r r c C s. | j D ]}t| q|d ur|d d S d S )Nz,ERROR)r r+ display_htmlr, r r r r L s
zParserError.displayNr r. r r r r r/ I s r/ c s4 e Zd ZdZ fddZd
ddZd dd Z ZS ) ArgumentParsera! Customized ArgumentParser for LLM Magics.
This class overrides the parent argparse.ArgumentParser's error-handling
methods to avoid side-effects like printing to stderr. The messages are
accumulated and passed into the raised exceptions for the caller to
handle them.
c s t j|i | g | _d S r )r r
_messages)r r r r r r r
] s
zArgumentParser.__init__Nc C s ~| j | dS )z2Override ArgumentParser's _print_message() method.N)r2 append)r messagefiler r r _print_messagea s z ArgumentParser._print_messager c C s6 |r| | | j}g | _|dkrt|dt|d)z(Override ArgumentParser's exit() method.r )r )r6 r2 r* r/ )r statusr4 r r r r exitf s
zArgumentParser.exitr )r N)r! r" r# r$ r
r6 r8 r' r r r r r1 T s
r1 )r$
__future__r r% argparsetypingr google.generativeai.notebookr RuntimeErrorABCMetar r* r/ r1 r r r r s