ait d Z ddlZddlZddlZddlmZ ddlmZmZ G d dej Z
G d dej Z d Z
d
Zd Zd Zd
Zd Zd ZddZd ZdZd Zd Zd ZdS )z:
Extensions for validation and output formatting argparse
N) PrettyTable)
parse_date
parse_period2c e Zd ZdZd Zd ZdS )SmartFormattera#
Format argparse help in many line, to format many lines use "R|" at start line
>>>import argparse
>>>parser = argparse.ArgumentParser(formatter_class=SmartFormatter)
>>>parser.add_argument('-a', '--argument', help="R| Descriptions: \n first new line\n second new line")
c | d r |dd S t j | || S )NzR| )
startswith
splitlinesargparse
HelpFormatter _split_lines)selftextwidths /builddir/build/BUILDROOT/lve-stats-4.3.2-1.el9.x86_64/opt/cloudlinux/venv/lib/python3.11/site-packages/lvestats/lib/commons/argparse_utils.pyr zSmartFormatter._split_lines sK ??4 )8&&(( (%224uEEE c t |t r.| || }t |d S t j | || S )N )
isinstance
ParseDatetime_metavar_formatterstrr r
_format_args)r actiondefault_metavar get_metavars r r zSmartFormatter._format_args! sZ fm , , ' 11&/JJK{{1~~&& &%224 QQQr N)__name__
__module__ __qualname____doc__r r r r r r sD F F F
R R R R Rr r c e Zd ZdZddZdS )r z
Interprets a string ([YY]YY-MM-DD[ HH:MM]) as a datetime for argument parsing.
return parsed data as
Nc t |d rd | n|} t | }n+# t $ r }t j | | d |d }~ww xY wt
|| j | d S )N__iter__ z/ is an invalid; use [YY]YY-MM-DD[ HH:MM] format)hasattrjoinr
ValueErrorr
ArgumentErrorsetattrdest)r parser namespacevalues
option_string datetime_strresult_datetimees r __call__zParseDatetime.__call__- s +26:+F+FRsxx'''F ( 6 6OO ( PPP
49o66666s 9
A!A A!N)r r r r! r4 r" r r r r ( s2 7 7 7 7 7 7r r c j t | } | dk s| dk rt j | d | S )z
Check if input argument between 0 and 100
:param str value:
:return int: percent
:except argparse.ArgumentTypeError: generate exception for command line warning message
r d z2 is an invalid; this value can range from 0 to 100)intr ArgumentTypeError)values r check_percentagesr; 9 s?
JJE qyyECKK(E)e)e)efff Lr c t | }n+# t $ r }t j d| d |d}~ww xY wd| v s|dk rt j d| d |S )z
Check if input argument greater than (or equal to) zero and not float
:param str value:
:return int:
:except argparse.ArgumentTypeError: generate exception for command line warning message
"zA" is an invalid; must be a number greater than (or equal to) zeroN.r r8 r) r r9 )r: non_negative_intr3 s r check_non_negative_intrA F s u:: ( X X X X
e||'!++( X X X X
:5:c t | }n+# t $ r }t j d| d |d}~ww xY wd| v s|dk rt j d| d |S )z
Check if input argument is digest, greater than zero and not float
:param str value: line to check
:return int: positive integer
:except argparse.ArgumentTypeError: generate exception for command line warning message
r= z3" is an invalid; must be a number greater than zeroNr> r r? )r: positive_intr3 s r check_positive_intrE Z s 5zz ( J J J J
e|||q((( J J J J
rB c t | }n+# t $ r }t j d| d |d}~ww xY wd|cxk rdk sn t j d| d |S )a=
Check if input argument is a correct timestamp
253370754000.0 == time.mktime(datetime.datetime(year=datetime.MAXYEAR, month=1, day=1).timetuple())
:param str value: line to check
:return float: timestamp
:except argparse.ArgumentTypeError: generate exception for command line warning message
r= zF" is an invalid; timestamp must be a number between 0 and 253370754000Nr l c%| )floatr) r r9 )r: positive_floatr3 s r check_timestamprI n s u ( ] ] ] ]
~ , , , , , , , ,( ] ] ] ]
rB c d} t | }n-# t $ r }t j d| d| |d}~ww xY wd| v s |dk s||k rt j d| d| |S )a
Check if input argument less then maximum integer number
PostgreSQL - 2147483647
MySQL - 2147483647
SQLite - 9223372036854775807
:param str value:
:return int:
:except argparse.ArgumentTypeError: generate exception for command line warning message
ir= zP" is an invalid; must be a number greater than (or equal to) zero and less than Nr> r r? )r: max_intr@ r3 s r check_from_zero_to_max_intrL s G u:: (
'
'
' $
'
'
e||'!++/?'/I/I(
'
'
' $
'
'
s
>9>c t j | j }n+# t $ r }t j d| d |d}~ww xY w|S )z
Check if user present in system and convert user name to user id
:param str value: user name (login)
:return int: user id
:except argparse.ArgumentTypeError: generate exception for command line warning message
zuser "z" not present in systemN)pwdgetpwnampw_uidKeyErrorr r9 )r: user_idr3 s r check_user_namerS sd Y,u%%, Y Y Y()P%)P)P)PQQWXXY Ns
A?Ac v t | }d| v s|dk rt j d| d t | dz } n# t $ r Y nw xY w t | }n4# t $ r'}t j t | |d}~ww xY w|dk rt j d| d |S ) z
Parse multiple period line, and validate input data
:param str value: line to parsing
:return int: period in seconds
:except argparse.ArgumentTypeError: generate exception for command line warning message
r> r r= ze" is an invalid; time period should be greater than (or equal to) 1 minute (60 seconds); not negativemN< zW" is an invalid; time period should be greater than (or equal to) 1 minute (60 seconds))r8 r r9 r r) r )r:
int_convertedsecondsr3 s r period_typerY s
E
%<<=A--,6E 6 6 6
M""S(
8 && 8 8 8(Q00a78||(
$
$
$
$
Ns* AA
AAA' '
B1"BBTc |p t j }| } | dk r2| dddd }|t j d z
}n|| dk r | dddd }|}nZ|r/| dd }| dd }n|}|}|t j t | z }||fS ) aP
Extented period_type function
Added support 'yesterday' and 'today'
:param bool round_to_minutes:
:param str value: line to parsing
:return tuple: period from to (, )
:except argparse.ArgumentTypeError: generate exception for command line warning message
yesterdayr )hourminutesecond microsecondr )daystoday)r^ r_ )rX )datetimeutcnowlowerreplace timedeltarY )r: datetime_nowround_to_minutes datetime_to
datetime_froms r period_type2rk s
=8#4#;#;#=#=L KKMME "**!AST*UU #h&8a&@&@&@@
' $,,!AaUV,WW
" ' ( 0 0q 0 I IM&..aQ.GGKK (M&K+K4F4FGGGG
+ %%r c | d }| dd } t |dd }|D ] }d|j |<
t t |j | | d S ) z
Prepare aliases descriptions to table formatted
:param tuple|list|iter description: massive of data
:return str: ascii table
r r Nr )
padding_width left_paddinglF)border)r alignlistmapadd_row
get_string) descriptionfieldstablefields r format_aliasesrz s} ^F abb/K aa @ @ @E ! ! EU]K ( ())) 5 ) ))r Z c n | dd } t | G fddt j }|S )Nr c e Zd Zd fd ZdS )!alias_action..AliasActionNc | j dv }|rUg }|D ]O}| t t | d Pn;| d d g}t | t }|rRt } d
| }
t j | d|
dd
| fd |D } |s| d
} t || j | d S ) N)+*,r= 'z", "z alias(es) "z" not valid; choice from: z, c g | ]
}| S r" r" ).0_aliases_func_dicts r
z>alias_action..AliasAction.__call__.. s <<.AliasAction.__call__ sw !Z: 5N
A " J JDNN6$
0B0B30G0G#H#HIIIIJ "<<,,22377==??@ $'w<<#:#:3?P;Q;Q#R#R #
#():#;#; &,kk2F&G&G# ,B"5 B B$(II.>$?$?B B =<<>> _alias_dscr_to_dict((('al_1', 'al_2', 'al_m1', 'description1'), ('al_3', 'al_4', 'al_m2', 'description1')))
{'al_2': 'al_m1', 'al_1': 'al_m1', 'al_m1': 'al_m1', 'al_3': 'al_m2', 'al_m2': 'al_m2', 'al_4': 'al_m2'}
:param tuple|list alias_dscr:
:type return: dict
c T g | ]$}| | f%S r" )rd )r r
val_index_s r r z'_alias_dscr_to_dict..& s2 TTTA1Z=..00!J-@TTTr r r c 0 g | ]}| | fS r" r" )r r
key_index_r s r r z'_alias_dscr_to_dict..( s& "V"V"VaAjM1Z=#A"V"V"Vr )rangelenr dict)r aliases_func_pairsr r r s @@r r r s JTTTTTTTC
1
. . 233 X X
!!"V"V"V"V"V:"V"V"VWWWW /00 r c t | d | j }}| j r
| j \ }}t | d| || _ | S )Nfrom)getattrtoperiodr+ )r. period_from period_tos r argparse_process_periodr - sN $Y 7 7 K 2!*!1 Y Iv{+++ IL r )NT)r! r rN rb prettytabler lvestats.lib.commons.dateutilr r r
r r r r; rA rE rI rL rS rY rk rz BY_USAGE_PERCENTr r r r" r r r s
# # # # # # C C C C C C C CR R R R RX+ R R R&7 7 7 7 7HO 7 7 7"
( ( * 4 <& & & &8 * * * F & r