o iC@sddlZddlmZdZdZdZdZdZdZd Z eee Z efd e d e d e fd dZ Gddde eZd e de de d e fddZdeeejefd e de de dBde de dede d e fddZd e de d e fddZdS)N)Enum-_abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ 0123456789valuesizereturncCst|d|S)z&Generate a hash fragment of the value.N)hashlibsha256encode hexdigestr r re/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/key_value/shared/utils/sanitize.pygenerate_hash_fragmentsrc@seZdZdZdZdZdS)HashFragmentModealwaysneveronly_if_changedN)__name__ __module__ __qualname__ALWAYSNEVERONLY_IF_CHANGEDrrrrrsrallowed_characters replace_withcCsRd}d}|D]}||vr||7}d}q|rq||7}d}qt|dkr'|r'dS|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 rr new_valuelast_char_was_replacedcharrrrsanitize_characters_in_string s r' max_lengthreplacement_characterhash_fragment_separatorhash_fragment_modehash_fragment_lengthc Cs|tkr dt}t|||dkrd}t|t||d}t|t|} |r0t|||dn|} |tjkrK|| } | d| } | sE|S| ||S|tjkrp| d|} || kr\|S|| } | d| } | sj|S| ||S| sxd}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 rr NzKEntire value was sanitized and hash_fragment_mode is HashFragmentMode.NEVER)MINIMUM_MAX_LENGTH ValueErrorrr#r'rrr) r r(rr)r*r+r,msg hash_fragmenthash_fragment_size_requiredsanitized_valueactual_max_lengthrrrsanitize_string>s>          r5cCs`|tkr dt}t|t||kr|S|d|d}t|}|dd}||S)aHash part of the value if it exceeds the maximum length. This operation will truncate the value to the maximum length minus 8 characters and will swap the last 8 characters with the first 8 characters of the generated hash. Args: value: The value to hash. max_length: The maximum length of the value. Must be greater than 32. Returns: The hashed value if the value exceeds the maximum length, otherwise the original value. z max_length must be greater than Nr)r.r/r#r rrr)r r(r0truncated_value hash_of_valuefirst_eight_of_hashrrrhash_excess_lengths   r9)r enumrr.DEFAULT_HASH_FRAGMENT_SIZEDEFAULT_HASH_FRAGMENT_SEPARATORDEFAULT_REPLACEMENT_CHARACTERLOWERCASE_ALPHABETUPPERCASE_ALPHABETNUMBERSALPHANUMERIC_CHARACTERSstrintrrr'rr5r9rrrrsV    ! J