Python实现XML转Oracle SQL语句的问题求助

当前生成的SQL语句 #

CREATE TABLE EMPLOYEE_DETAILS(
  EMP_ID VARCHAR2(70),
  EMP_NAME VARCHAR2(70),
  EMP_SALARY VARCHAR2(70),
  EMP_DEPT VARCHAR2(70),
  EMP_JOB_ID VARCHAR2(70)
);

期望的SQL语句 #

CREATE TABLE BANK_EMPLOYEE(
  EMP_ID VARCHAR2(30),
  EMP_NAME VARCHAR2(70),
  EMP_SALARY NUMBER,
  EMP_DEPT VARCHAR2(20),
  EMP_JOB_ID VARCHAR2(50)
);

现有两种实现代码 #

方法1 #

# Generate the CREATE TABLE statement
create_table_sql = "CREATE TABLE {} (".format(root.tag) #start creating the sql statements to create a table with the name of the root tag
for column in columns: #Iterate through each column in the table
    if data_types[column] == 'str': #Check if the data type of the column is a string
        create_table_sql += "{} VARCHAR2, ".format(column) #Add the column name and data type (TEXT) to the SQL statement
    elif data_types[column] == 'int': #Check if the data type of the column is an integer
        create_table_sql += "{} INT".format(column) # Add the column name and data type (INT) to the SQL statement
    elif data_types[column] == 'float': #Check if the data type of the column is a float
        create_table_sql += "{} DECIMAL(10, 2), ".format(column) #Add the column name and data type (INT) to the SQL statement
create_table_sql = create_table_sql[:-2] + ");" # Remove the trailing comma and space, and add the closing parenthesis to the SQL statement

方法2 #

# Define a dictionary that maps Python data types to SQL data types
sql_data_types = {
    'str': 'VARCHAR2(70)',
    'int': 'NUMBER',
    'float': 'DECIMAL(10, 2)'
}

# Generate the CREATE TABLE statement
create_table_sql = "CREATE TABLE {}(
".format(root.tag)
for column in columns:
    # Add each column and its data type to the CREATE TABLE statement
    create_table_sql += "  {} {},
".format(column, sql_data_types[data_types[column]])

补充XML示例 #

<?xml version="1.0" encoding="UTF-8"?>
<EMPLOYEE_DETAILS>
  <EMPLOYEE>
    <EMP_ID>101</EMP_ID>
    <EMP_NAME>Marshall Walker</EMP_NAME>
    <EMP_SALARY>100000.00</EMP_SALARY>
    <EMP_DEPT>100</EMP_DEPT>
    <EMP_JOB_ID>IT001</EMP_JOB_ID>
  </EMPLOYEE>
</EMPLOYEE_DETAILS>

解决方案 #

问题分析 #

1. 表名固定化：现有代码直接使用XML根节点名称作为表名，无法自定义目标表名。
2. 数据类型映射粗糙：
   • 字符串类型统一映射为VARCHAR2(70)，无法满足不同字段的长度需求；
   • 浮点型字段被映射为DECIMAL(10,2)，但期望是Oracle原生的NUMBER类型；
   • XML解析时可能将数值型字段识别为字符串，导致错误映射。
3. 字符串拼接缺陷：方法2中直接在字符串中插入换行符会引发语法错误，且手动拼接易出现逗号、格式问题。

修正后的代码实现 #

import xml.etree.ElementTree as ET

def xml_to_oracle_create_table(xml_content, target_table_name, column_type_config):
    # 解析XML内容
    root = ET.fromstring(xml_content)
    # 从第一个EMPLOYEE节点提取所有列名
    employee_node = root.find('EMPLOYEE')
    columns = [child.tag for child in employee_node]
    
    # 构建CREATE TABLE语句框架
    create_table_sql = f"CREATE TABLE {target_table_name}(\n"
    # 批量生成列定义
    column_definitions = []
    for col in columns:
        column_definitions.append(f"  {col} {column_type_config[col]}")
    
    # 拼接所有列定义并完成语句
    create_table_sql += ",\n".join(column_definitions) + "\n);"
    return create_table_sql

# 示例调用
xml_sample = """<?xml version="1.0" encoding="UTF-8"?>
<EMPLOYEE_DETAILS>
  <EMPLOYEE>
    <EMP_ID>101</EMP_ID>
    <EMP_NAME>Marshall Walker</EMP_NAME>
    <EMP_SALARY>100000.00</EMP_SALARY>
    <EMP_DEPT>100</EMP_DEPT>
    <EMP_JOB_ID>IT001</EMP_JOB_ID>
  </EMPLOYEE>
</EMPLOYEE_DETAILS>"""

# 自定义每个字段的Oracle数据类型配置
column_config = {
    'EMP_ID': 'VARCHAR2(30)',
    'EMP_NAME': 'VARCHAR2(70)',
    'EMP_SALARY': 'NUMBER',
    'EMP_DEPT': 'VARCHAR2(20)',
    'EMP_JOB_ID': 'VARCHAR2(50)'
}

# 生成并打印目标SQL语句
result_sql = xml_to_oracle_create_table(xml_sample, 'BANK_EMPLOYEE', column_config)
print(result_sql)

代码说明 #

• 支持自定义目标表名，摆脱对XML根节点名称的依赖；
• 为每个字段单独配置数据类型，精准匹配需求；
• 使用join方法拼接列定义，避免手动处理逗号、换行的格式问题；
• 从XML的实际数据节点提取列名，确保列名准确。

内容的提问来源于stack exchange，提问作者sali