mirror of
https://github.com/tencentmusic/supersonic.git
synced 2025-12-10 19:51:00 +00:00
60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
# -*- coding:utf-8 -*-
|
|
import re
|
|
|
|
import os
|
|
import sys
|
|
|
|
sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
|
|
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
sys.path.append(os.path.dirname(os.path.abspath(__file__)))
|
|
|
|
from instances.logging_instance import logger
|
|
|
|
|
|
def schema_link_parse(schema_link_output: str):
|
|
try:
|
|
schema_link_output = schema_link_output.strip()
|
|
pattern = r'Schema_links:(.*)'
|
|
schema_link_output = re.findall(pattern, schema_link_output, re.DOTALL)[0].strip()
|
|
except Exception as e:
|
|
logger.exception(e)
|
|
schema_link_output = None
|
|
|
|
return schema_link_output
|
|
|
|
def combo_schema_link_parse(schema_linking_sql_combo_output: str):
|
|
try:
|
|
schema_linking_sql_combo_output = schema_linking_sql_combo_output.strip()
|
|
pattern = r'Schema_links:(\[.*?\])|Schema_links: (\[.*?\])'
|
|
schema_links_match = re.search(pattern, schema_linking_sql_combo_output)
|
|
|
|
if schema_links_match.group(1):
|
|
schema_links = schema_links_match.group(1)
|
|
elif schema_links_match.group(2):
|
|
schema_links = schema_links_match.group(2)
|
|
else:
|
|
schema_links = None
|
|
|
|
except Exception as e:
|
|
logger.exception(e)
|
|
schema_links = None
|
|
|
|
return schema_links
|
|
|
|
def combo_sql_parse(schema_linking_sql_combo_output: str):
|
|
try:
|
|
schema_linking_sql_combo_output = schema_linking_sql_combo_output.strip()
|
|
pattern = r'SQL:(.*)'
|
|
sql_match = re.search(pattern, schema_linking_sql_combo_output)
|
|
|
|
if sql_match:
|
|
sql = sql_match.group(1)
|
|
else:
|
|
sql = None
|
|
except Exception as e:
|
|
logger.exception(e)
|
|
sql = None
|
|
|
|
return sql
|
|
|