使用xml文件更改sql中的列not null和默认值

<columns>
  <column>
    <name>column1</name>
    <notNull>true</notNull>
    <defaultValue>0</defaultValue>
  </column>
  <column>
    <name>column2</name>
    <notNull>false</notNull>
    <defaultValue>null</defaultValue>
  </column>
</columns>

import xml.etree.ElementTree as ET

def change_column_constraints(xml_file, sql_file):
    tree = ET.parse(xml_file)
    root = tree.getroot()

    sql_statements = []

    for column in root.findall('column'):
        column_name = column.find('name').text
        not_null = column.find('notNull').text
        default_value = column.find('defaultValue').text

        sql = f"ALTER TABLE table_name MODIFY {column_name}"

        if not_null.lower() == 'true':
            sql += " NOT NULL"
        else:
            sql += " NULL"

        if default_value.lower() != 'null':
            sql += f" DEFAULT {default_value}"

        sql_statements.append(sql)

    with open(sql_file, 'w') as f:
        f.write('\n'.join(sql_statements))

# 使用示例
change_column_constraints('columns.xml', 'change_column_constraints.sql')