dZddlmZddlmZddlmZddlZddlZdZddd d d d d ddddd Zdd ddddd ddddd Z dZ ddZ ddZ ddZ ddZddZddZd dZddZy)!z:Scaled integer ISO/IEC unit prefix parsing and formatting.)absolute_import)division)unicode_literalsNz ^ # Beginning of input marker. (?P\d+) # Amount. ((?P[-/a-zA-Z]+))? # Optional scale and type abbr. $ # End of input marker. ii@Biʚ;lJ)lI5iii@ll ) kMGTPkiMiGiTiPicttj|d}|Dcgc] \}}|s ||zc}}Scc}}w)z:Returns a list of the units in scales sorted by magnitude.c|d|dfSNrrvalues .lib/googlecloudsdk/core/util/scaled_integer.pyz#_UnitsByMagnitude..Bs%(E!H)=key)sortedsix iteritems)units type_abbr scale_itemsr_s r_UnitsByMagnituder$?s<s}}U+=?+(3 ; fc1s# / ;; ;s ==c|s|S|j}t|}t|jD]}|sn||dz |k(s|dz}|d|S)z7Returns suffix with trailing type abbreviation deleted.rN)upperlenreversed)suffixr!sics rDeleteTypeAbbrr-Fs`  M lln! !f! IOO% &a  Qx1}1fa ' rc t||}|s!|}|sd}nw|jdrfd|ddz}n]|dj}|dk(rd}t|dkDr|ddvr |dz }|d d}n|dd}|rt d j ||||xst j|}|s5t d j ||d jt|d|S) aReturns the size per unit for binary suffix string. Args: suffix: str, A case insensitive unit suffix string with optional type abbreviation. type_abbr: str, The optional case insensitive type abbreviation following the suffix. default_unit: The default unit prefix name. units: {str: int} map of unit prefix => size. Raises: ValueError: on unknown units of type suffix. Returns: The binary size per unit for a unit+type_abbr suffix. rKrrNr)r+Ir+z4Invalid type [{}] in [{}], expected [{}] or nothing.z2Invalid suffix [{}] in [{}], expected one of [{}].,) r- startswithr&r' ValueErrorformat_ISO_IEC_UNITSgetjoinr$)r)r! default_unitr prefixunitsizes r GetUnitSizer=Ts" &) ,&  D  d   48^d !9?? D s{ d 6{Q6!9 2 ckdabzfabzf Mvvy9 ;;  !> & &t ,$ IfT6 XX&7r&BCE FF +rc(t|||tS)aReturns the binary size per unit for binary suffix string. Args: suffix: str, A case insensitive unit suffix string with optional type abbreviation. type_abbr: str, The optional case insensitive type abbreviation following the suffix. default_unit: The default unit prefix name. Raises: ValueError for unknown units. Returns: The binary size per unit for a unit+type_abbr suffix. r!r9r )r= _BINARY_UNITS)r)r!r9s rGetBinaryUnitSizerAs Vy|( **rc `tjt|tj}|sA|rd|zdznd}t dj ||dj t|||jdxsd}t||||}t|jd}||zS) aParses and returns a units scaled integer from string. ISO/IEC/SI rules relaxed to ignore case in unit and type names/abbreviations. Args: units: {str: int} map of unit prefix => size. string: The string to parse the integer + units. default_unit: The default unit prefix name. type_abbr: The optional type abbreviation suffix, validated but otherwise ignored. Raises: ValueError: on invalid input. Returns: The scaled integer value. []rzB[{}] must the form INTEGER[UNIT]{} where units may be one of [{}].r2r)r?amount) rematch_INTEGER_SUFFIX_TYPE_PATTERNVERBOSEr4r5r8r$groupr=int) r stringr9r!rGoptional_type_abbrr)r<rEs r_ParseScaledIntegerrNs& ((/ D% 2;y3. L **5)<= ? @@ ;;x &B&  E K$ u{{8$ %& $rc(tt|||S)ajParses and returns an ISO Decimal/Binary scaled integer from string. ISO/IEC prefixes: 1k == 1000, 1ki == 1024. Args: string: The string to parse the integer + units. default_unit: The default unit prefix name. type_abbr: The optional type abbreviation suffix, validated but otherwise ignored. Returns: The scaled integer value. r9r!)rNr6rLr9r!s r ParseIntegerrRs f<9 NNrctttjtdD](\}}||ks ||zrdj ||z||cSdj ||S)zReturns a pretty string representation of an ISO Decimal value. Args: value: A scaled integer value. type_abbr: The optional type abbreviation suffix, validated but otherwise ignored. Returns: The formatted scaled integer value. c|d|dfSrrrs rrzFormatInteger..q588Lrrz{}{}{}{}{})r(rrrr6r5)rr!r)r<s r FormatIntegerrWsfvcmmN&C*L NOlfd u}UT\ __Ud]FI >>O ui ((rctttjtdD]=\}}||ks |dk(r||zr||z }t |d}dj ||||cSdj ||S)aReturns a pretty string of a binary-base number with decimal precision. Args: value (float|int): A number. type_abbr (str): The optional type abbreviation suffix, validated but otherwise ignored. decimal_places (int): Number of decimal places to include of quotient for unit conversion. Does not allow rounding if -1. Will suffer float inaccuracy at high values. Returns: A formatted scaled value string. c|d|dfSrrrs rrz$FormatBinaryNumber..rUrrrz{:.{precision}f}{}{}) precisionrV)r(rrrr@maxr5)rr!decimal_placesr)r< scaled_valuer[s rFormatBinaryNumberr_svcmmM&B*L NOlfd u} 2 %$,T\lna(i # * *  Y+@@O ui ((rc(tt|||S)aParses and returns a Binary scaled integer from string. All ISO/IEC prefixes are powers of 2: 1k == 1ki == 1024. This is a concession to the inconsistent mix of binary/decimal unit measures for memory capacity, disk capacity, cpu speed. Ideally ParseInteger should be used. Args: string: The string to parse the integer + units. default_unit: The default unit prefix name. type_abbr: The optional type abbreviation suffix, validated but otherwise ignored. Returns: The scaled integer value. rP)rNr@rQs rParseBinaryIntegerras" V,) MMr)B)rbrN)rbr)rrb)rbrZ)__doc__ __future__rrrrFrrHr6r@r$r-r=rArNrRrWr_rarrrres A&'                         < *Z*(BN$)$)8Mr