o  @sdZddlmZddlmZddlmZddlZddlZdZddd d d d d ddddd Zdd ddddd ddddd Z ddZ d(ddZ d)ddZ d*ddZ d+ddZd+dd Zd(d!d"Zd,d$d%Zd+d&d'ZdS)-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 ) kMGTPkiMiGiTiPics(tt|ddd}fdd|DS)z:Returns a list of the units in scales sorted by magnitude.cS|d|dfSNrrvaluerrD/tmp/google-cloud-sdk/lib/googlecloudsdk/core/util/scaled_integer.pyBz#_UnitsByMagnitude..keycsg|] \}}|r|qSrr).0r_ type_abbrrr Csz%_UnitsByMagnitude..)sortedsix iteritems)unitsr scale_itemsrrr_UnitsByMagnitude?s r%BcCsV|s|S|}t|}t|D]}|sn ||d|kr$|d8}q|d|S)z7Returns suffix with trailing type abbreviation deleted.rN)upperlenreversed)suffixrsicrrrDeleteTypeAbbrFs r.rc Cst||}|s|}|sd}nB|drd|dd}n4|d}|dkr(d}t|dkr?|ddvr?|d7}|d d}n|dd}|rPtd ||||pSt|}|shtd ||d t |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#prefixunitsizerrr GetUnitSizeTs6     r=cCst|||tdS)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. rr9r#)r= _BINARY_UNITS)r*rr9rrrGetBinaryUnitSizesr@c Csztt|tj}|s#|rd|dnd}td||dt|||dp)d}t ||||d}t |d}||S) 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#stringr9rrEoptional_type_abbrr*r<rCrrr_ParseScaledIntegersrLcCtt|||dS)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)rLr6rJr9rrrr ParseIntegersrPcCsVttttdddD]\}}||kr$||s$d||||Sq d||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. cSrrrrrrrrrzFormatInteger..rz{}{}{}{}{})r)r r!r"r6r5)rrr*r<rrr FormatIntegers  rRcCsrttttdddD]%\}}||kr2|dkr||rq ||}t|d}dj||||dSq d||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. cSrrrrrrrrrz$FormatBinaryNumber..rrSrz{:.{precision}f}{}{}) precisionrQ)r)r r!r"r?maxr5)rrdecimal_placesr*r< scaled_valuerTrrrFormatBinaryNumbers    rXcCrM)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. rN)rLr?rOrrrParseBinaryIntegersrY)r&)r&rN)r&r)rr&)r&rS)__doc__ __future__rrrrDr!rFr6r?r%r.r=r@rLrPrRrXrYrrrrsP      -  !