如何使用反射和描述性语法覆盖sqlalchemy中的列名

import inspect
import ast
from sqlalchemy import Column, String, Integer
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True)
    name = Column(String)
    age = Column(Integer)

# 获取User模型的所有列名
column_names = [column.name for column in inspect.getmembers(User, lambda a: isinstance(a, Column))]

# 生成描述性语法树
tree = ast.parse("""
class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True)
    name = Column(String)
    age = Column(Integer)
""")

# 修改描述性语法树，以覆盖列名
for node in ast.walk(tree):
    if isinstance(node, ast.Name):
        if node.id in column_names:
            node.id = node.id.upper()

# 动态执行生成的代码
exec(compile(tree,<string>', 'exec'))