KiddlZddlmZddlmZdZdZdZdZdZ d Z d Z e e ze zZ efd e d ed e fdZGdde eZd e de de d e fdZedeeej$efd e dede dzde de deded e fdZed e ded e fdZy)N)Enum) bear_enforce-_abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ 0123456789valuesizereturncjtj|jjd|S)z&Generate a hash fragment of the value.N)hashlibsha256encode hexdigestr r s k/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/key_value/shared/utils/sanitize.pygenerate_hash_fragmentrs) >>%,,. ) 3 3 5et <<ceZdZdZdZdZy)HashFragmentModealwaysneveronly_if_changedN)__name__ __module__ __qualname__ALWAYSNEVERONLY_IF_CHANGEDrrrrs F E'Orrallowed_characters replace_withcjd}d}|D]}||vr||z }d}|r||z }d}t|dk(r|ry|S)aReplace characters in a string. If multiple characters are in a row that are not allowed, only the first one will be replaced. The rest will be removed. If all characters are not allowed, an empty string will be returned. Args: value: The value to replace characters in. to_replace: The characters to replace. replace_with: The characters to replace with. FT)len)r r$r% new_valuelast_char_was_replacedchars rsanitize_characters_in_stringr-"sgI" * % %  I%* "%  %I%) " * 9~5 r max_lengthreplacement_characterhash_fragment_separatorhash_fragment_modehash_fragment_lengthc|tkrdt}t|||dzkDr d}t|t||}t|t|z} |rt |||n|} |t j k(r|| z } | d| } | s|S| |z|zS|t jk(r"| d|} || k(r|S|| z } | d| } | s|S| |z|zS| s d}t|| S)aSanitize the value, replacing characters and optionally adding a fragment a hash of the value if requested. If the entire value is sanitized and hash_fragment_mode is HashFragmentMode.ALWAYS or HashFragmentMode.ONLY_IF_CHANGED, the value returned will be the hash fragment only. If the entire value is sanitized and hash_fragment_mode is HashFragmentMode.NEVER, an error will be raised. Args: value: The value to sanitize. allowed_characters: The allowed characters in the value. max_length: The maximum length of the value (with the hash fragment added). hash_fragment_separator: The separator to add between the value and the hash fragment. hash_fragment_mode: The mode to add the hash fragment. z,max_length must be greater than or equal to zEhash_fragment_length must be less than or equal to half of max_lengthr)r r$r%NzKEntire value was sanitized and hash_fragment_mode is HashFragmentMode.NEVER)MINIMUM_MAX_LENGTH ValueErrorrr)r-rr r") r r.r$r/r0r1r2msg hash_fragmenthash_fragment_size_requiredsanitized_valueactual_max_lengths rsanitize_stringr<@s90&&<=O2<<>M'+ 0 00r)renumr)key_value.shared.type_checking.bear_sprayrr5DEFAULT_HASH_FRAGMENT_SIZEDEFAULT_HASH_FRAGMENT_SEPARATORDEFAULT_REPLACEMENT_CHARACTERLOWERCASE_ALPHABETUPPERCASE_ALPHABETNUMBERSALPHANUMERIC_CHARACTERSstrintrrr-r"r<rAr#rrrMsLB"% #11 ,/AAGK += = = =(sD( #UX]`<&*!>#B+;+K+K :G GGd G G ! G ) GG GGT1c1s1s11r