o ^i9@s(ddlmZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z ddl mZmZmZmZmZmZddlmZmZmZddlmZmZmZz dd lmZmZWneyne dZGd d d ZYnwd Z!e"e#ej$fZ%e Gd ddZ&e GdddZ'e GdddZ(dS)) annotationsN)Callable) dataclassfield)AnyOptionalUnionget_args get_originget_type_hints) DocstringDocstringParamparse)Color format_text get_underline)NoneType UnionTypec@s eZdZdS)rN)__name__ __module__ __qualname__rrR/var/www/html/stock_analysis/be/venv/lib/python3.10/site-packages/targ/__init__.pyrsrz0.6.0c@s2eZdZUeedZded<eedZded<dS) Argumentsdefault_factory list[str]argszdict[str, Any]kwargsN) rrrrlistr__annotations__dictrrrrrr#s rc@seZdZUdZded<dZded<dZded<eedZ d ed <d&d dZ e ddZ e d'ddZ d(ddZd)ddZd*ddZe d'ddZe d'ddZd d!Zd+d$d%ZdS),Commanda Represents a CLI command. :param command: The Pyton function or coroutine which gets called. :param group_name: Commands can belong to a group. In this situation, the group name must appear before the command name when called from the command line. For example `python my_file.py group_name command_name`. :param command_name: By default the command name is the same as the callable, but it can be overriden here. :param aliases: You can provide aliases, which can be abbreviations or common mispellings. For example, for a `command_name` of ``run``, we could have aliases like ``['start', 'rn']``. rcommandN Optional[str] group_name command_namerraliasesreturnNonecCsJt|jjpd|_t|j|_t|j|_d|_|j s#|jj |_ dSdS)NF) rr$__doc__command_docstringr rinspect signaturesolor'rselfrrr __post_init__Cs zCommand.__post_init__cCs|jr |jd|jS|jS)N )r&r'r1rrr full_nameKs zCommand.full_namestrcCs*t|jjpd}d|jpd|jpdgS)Nr+r4)rr$r,joinshort_descriptionlong_description)r2 docstringrrr descriptionSs zCommand.descriptionOptional[DocstringParam]cCs$|jjD] }|j|kr|SqdSN)r-paramsarg_name)r2r?paramrrr_get_docstring_param]s  zCommand._get_docstring_paramr?cCs||}|r |jp dSdS)Nr+)rAr;)r2r?docstring_paramrrr_get_arg_descriptioncs  zCommand._get_arg_descriptioncCs*|jj|}|r|j}|tjur|SdSr=)r/ parametersgetdefaultr._empty)r2r? parameterrFrrr_get_arg_defaultjs  zCommand._get_arg_defaultcCsg}|jD]<\}}|j|d}|j|d}t|}|dur&d|dnd}|t|tj dd||r>|||dqd |S) zP :returns: A string containing a description for each argument. )r?Nz [default=]r+colorr4 ) ritemsrCrIjsondumpsappendrrcyanr7)r2outputr?_arg_description arg_defaultarg_default_json default_strrrrarguments_descriptionrs       zCommand.arguments_descriptioncCs|jrg}n |jp d}t|tjdg}|jjD]7\}}|jt j ur.| t|tj dq|jdurB| td|dtj dq| td|dtj dqd |S)zb Example: some_command required_arg [--optional_arg=value] [--some_flag] r+rKFz[--rJz=X]r4)r0r'rrgreenr/rDrNrFr.rGrQrRr7)r2rSr'r?rHrrrusages    z Command.usagecCstdt|jttt|jt|jtdtdttdddt|jtdtdttddd|jrEt|jntd|jrgtd ttd ddd |j}tt |t j d tddS) Nr+Usage-) characterArgszNo args Aliasesz, rK) printr'rlenr;r[rYr(r7rrrZ)r2 alias_stringrrr print_helps(      zCommand.print_help arg_classrc Cs|jdr |dSt|j}|j}t|jD]\}}t |jj|}|||<qi}| D];\}}||}|t vrE||}n%t |t tfvrjt|} dd| D} t| dkrj| d}|t vrj||}|||<q3t|jrt|jdi|dS|jdi|dS)z Call the command function with the given arguments. The arguments are all strings at this point, as they're come from the command line. helpNcSsg|]}|tur|qSr)r).0irrr sz%Command.call_with..rrr)rrErgr r$copy enumeraterr.getfullargspecrNCONVERTABLE_TYPESr rrr reiscoroutinefunctionasynciorun) r2rhrrindexvaluekeycleaned_kwargs annotationinner_annotationsfilteredrrr call_withs0         zCommand.call_with)r)r*r)r6)r)r<)r?r6r)r6)r?r6)rhr)rrrr,r!r&r'rr r(r3propertyr5r;rArCrIrYr[rgr{rrrrr#)s(          r#c@seZdZUdZdZded<eeddZded<d.d dZ d/ddZ ddgfd0ddZ d1ddZ d2ddZ d3d4dd Zd5d#d$Zd6d'd(Zed7d)d*Zd8d9d,d-ZdS):CLIz The root class for building the CLI. Example usage: .. code-block:: python cli = CLI() cli.register(some_function) :param description: Customise the title of your CLI tool. zTarg CLIr6r;F)rinitz list[Command]commandsr&r'r)boolcCs*|jD]}|j|kr|j|krdSqdS)z This isn't used by Targ itself, but is useful for third party code which wants to inspect the CLI, to find if a command with the given name exists. TF)rr&r')r2r&r'r$rrrcommand_existss zCLI.command_existsnamecCsd|vrdSdS)z{ Any custom names provided by user should not contain spaces (i.e. command names and group names). r4FTr)r2rrrr_validate_nameszCLI._validate_nameNr$rr%r(rcCsH|r ||s td|r||std|jt||||ddS)a Register a function or coroutine as a CLI command. :param command: The function or coroutine to register as a CLI command. :param group_name: If specified, the CLI command will belong to a group. When calling a command which belongs to a group, it must be prefixed with the ``group_name``. For example ``python my_file.py group_name command_name``. :param command_name: By default, the name of the CLI command will be the same as the function or coroutine which is being called. You can override this here. :param aliases: The command can also be accessed using these aliases. z)The group name should not contain spaces.z+The command name should not contain spaces.)r$r&r'r(N)r ValueErrorrrQr#)r2r$r&r'r(rrrregistersz CLI.registercCsbd|jtt|jdddddg}|jD]}|t|jtjd||j|dqd |S)Nr+z=Enter the name of a command followed by --help to learn more.Commandsz--------rKrM) r;rrerrQrr5rrZr7)r2linesr$rrr get_help_textCs     zCLI.get_help_textcCstjddS)z1 Remove any redundant arguments. rN)sysargvr1rrr_get_cleaned_argsYszCLI._get_cleaned_argsOptional[Command]cCs<|jD]}|j|ks||jvr|r|j|krq|SqdSr=)rr'r(r&)r2r'r&r$rrr _get_command_s zCLI._get_commandrurcCs|dvrdS|dvr dS|S)N)TruetruetT)FalsefalsefFr)r2rurrr_clean_cli_argumentls zCLI._clean_cli_argumentrrcCst}|D]:}|dr4|dddd}t|dkr(|d}||d}n|d}d}||j|<q||}|j|q|S)Nz--=rrT)r startswithsplitrerrrrQ)r2r argumentsarg_str componentsrrurrr_get_arg_classss    zCLI._get_arg_classcCst|jdkS)Nr)rerr1rrr_can_run_in_solo_modeszCLI._can_run_in_solo_moder0c Cs|}d}z|d}Wn tyd}Ynw||d}|r6|js+tddSd}|jd}d|_n t|dkrDt| dS|d}|j |d}|rV|d d}|sut|d kru|d}|d }|j ||d }|ru|d d}|rz | |}| |WdSt y} z*ttd tjd t| |rttntd|td WYd} ~ dSd} ~ wwtd|t| dS)a Run the CLI. :param solo: By default, a command name must be given when running the CLI, for example ``python my_file.py command_name``. In some situations, you may only have a single command registered with the CLI, so passing in the command name is redundant. If ``solo=True``, you can omit the command name i.e. ``python my_file.py``, and Targ will automatically call the single registered command. Nz--traceFTzQError - solo mode is only allowed if a single command is registered with the CLI.r+r)r'rr)r'r&zThe command failed.rKz#For a full stack trace, use --tracezUnrecognised command - )rrtrpoprrdrr0rerrrr{ Exceptionrrred traceback format_excrgrexit) r2r0 cleaned_argsr$rttracer'r&rh exceptionrrrrssd            zCLI.run)r&r6r'r6r)r)rr6r)r)r$rr&r%r'r%r(rr|)r)rr=)r'r6r&r%r)r)rur6r)r)rrr)r)r)r)F)r0r)rrrr,r;r!rr rrrrrrrrrr}rrsrrrrr~s&      (   r~)) __future__rrrdecimalr.rOrrcollections.abcr dataclassesrrtypingrrrr r r docstring_parserr r rformatrrrtypesrr ImportErrortype __VERSION__intfloatDecimalrprr#r~rrrrs6     G