Source code for pm4py.algo.querying.llm.utils.sql_utils

import re


ref_stri_1 = "ordabcchr"
ref_stri_2 = "ordabc"

re1 = re.compile(r"([^a-zA-Z0-9]+)")
re2 = re.compile(ref_stri_1)




def mask_non_alphanumeric(stri):
    stri_split = re1.split(stri)
    ret = []
    for el in stri_split:
        for char in el:
            if (
                char.isalnum()
                or char == " "
                or char
                in [
                    "(",
                    ")",
                    "*",
                    ".",
                    ",",
                    "'",
                    '"',
                    "=",
                    "<",
                    ">",
                    "_",
                    "+",
                    "-",
                    "!",
                ]
            ):
                ret.append(char)
            else:
                ret.append(
                    ref_stri_1 + ref_stri_2 + str(ord(char)) + ref_stri_1
                )
    return "".join(ret)





def restore_non_alphanumeric(stri):
    stri_split = re2.split(stri)
    ret = []
    for el in stri_split:
        if el.startswith(ref_stri_2):
            ret.append(chr(int(el[len(ref_stri_2):])))
        else:
            ret.append(el)
    return "".join(ret)