o uyi@sdZddlZddlmZddlmZmZdedefddZd ed edefd d Z deeefdefd dZ ddedeede fddZ de d edefddZ dededefddZeddedDZeddedDZdedefddZdS) z0Helper methods for working with Python Decimals.N)Decimal)OptionalUnionvaluereturncCs$|\}}}tt||dfS)zGet an unscaled value given a Decimal value. Args: value (Decimal): A Decimal instance. Returns: int: The unscaled value. r)as_tupleintrto_integral_value)rsigndigits_r Z/home/ubuntu/maya3_transcribe/venv/lib/python3.10/site-packages/pyiceberg/utils/decimal.pydecimal_to_unscaleds runscaledscalecCs"t|\}}}t||| fS)zGet a scaled Decimal value given an unscaled value and a scale. Args: unscaled (int): An unscaled value. scale (int): A scale to set for the returned Decimal instance. Returns: Decimal: A scaled Decimal instance. )rr)rrr r r r r runscaled_to_decimal&s rcCsFt|tr |ddSt|trt|ddStd|)zReturn the minimum number of bytes needed to serialize a decimal or unscaled value. Args: value (int | Decimal): a Decimal value or unscaled int value. Returns: int: the minimum number of bytes needed to serialize the value. zUnsupported value: ) isinstancer bit_lengthrr ValueError)rr r rbytes_required4s  r byte_lengthcCs(t|}|dur t|}|j|dddS)zReturn a byte representation of a decimal. Args: value (Decimal): a decimal value. byte_length (int): The number of bytes. Returns: bytes: the unscaled value of the Decimal as bytes. NbigT byteordersigned)rrto_bytes)rrunscaled_valuer r rdecimal_to_bytesEs rcCstj|ddd}t||S)zReturn a decimal from the bytes. Args: value (bytes): the bytes to be converted into a decimal. scale (int): the scale of the decimal. Returns: Decimal: the scaled decimal. rTr)r from_bytesr)rrunscaled_datumr r rbytes_to_decimalTs r"widthcCs4t|}|||||}t|tt|jS)zGet a truncated Decimal value given a decimal value and a width. Args: value (Decimal): a decimal value. width (int): A width for the returned Decimal instance. Returns: Decimal: A truncated Decimal instance. )rrabsrrexponent)rr#r applied_valuer r rtruncate_decimalbs r'c cs:|]}ttttdd|ddVqdS)rN)mathfloorlog10fabspow.0posr r r ps8r2c#s*|]tfddtdDVqdS)c3s |] }t|kr|VqdSN) MAX_PRECISIONr/pr rr2qsz.r3N)nextrange)r0r r6rr2qs(( precisioncCs&|dks|dkrtd|t|S)zCompute the number of bytes required to store a precision. Args: precision: The number of digits to store. Returns: The number of bytes required to store a decimal with a certain precision. rr:z+Unsupported precision, outside of (0, 40]: )rREQUIRED_LENGTH)r;r r rdecimal_required_bytests r=r4)__doc__r*decimalrtypingrrrrrrbytesrr"r'tupler9r5r<r=r r r rs