Python 获取自定义的列名

最新推荐文章于 2024-07-01 20:39:35 发布

小小毛毛虫~

最新推荐文章于 2024-07-01 20:39:35 发布

阅读量678

点赞数

文章标签： python orm

本文链接：https://blog.csdn.net/Lyncai/article/details/117356613

版权

自己在db_model中定义的一些类（Table），想只获取自定义的列名称，试了dir 和 __dict__的方法。

dir -- 会列出所有的属性和方法，利用filter试着去掉下划线，或者过滤去掉builtin，剩下一些还是无法过滤掉的属性，比如metadata, register

__dict__ 只返回已赋值的属性

后来发现inspect的getmember中有个__table__属性会列出自定义的列，因此采用这个方法。如下是代码 --

db_model.py中类的定义


class User(Base):
    __tablename__ = 't_users'
    id = Column(Integer, primary_key=True)
    sys_creation_date = Column(DateTime, default=datetime.datetime.now)
    sys_update_date = Column(DateTime, default=datetime.datetime.now)

    username = Column(String(64), nullable=False, index=True)
    status = Column(Enum('Active', 'Obsolete'), nullable=False, index=True, default='Active')
    password = Column(String(64), nullable=True)
    email = Column(String(64), nullable=True, index=False)
    modules = Column(String(64), nullable=True, index=False)

    face_feature = Column(BLOB, nullable=True, index=False)
    voice_feature = Column(BLOB, nullable=True, index=False)

获取列名称的方法：

mport inspect

key = "__table__"

def get_columns(db_model_object):

    result = {}

    members = inspect.getmembers(db_model_object, not inspect.isbuiltin)
    members_dict = dict(members)
    columns = members_dict.get(key).columns

    for col in columns:
        dict_mem = dict(inspect.getmembers(col, not inspect.isbuiltin))
        result[dict_mem['name']] = str(dict_mem['type'])
    return result


if __name__ == '__main__':
    from models.mysql_db import User
    user = User(username='123')
    result = get_columns(user)
    print(result)

这是输出：

"D:\Program Files\Python\Python39\python.exe" E:/source_code/python/flask_projs/db_opt/basic_module.py
{'id': 'INTEGER', 'sys_creation_date': 'DATETIME', 'sys_update_date': 'DATETIME', 'username': 'VARCHAR(64)', 'status': 'VARCHAR(8)', 'password': 'VARCHAR(64)', 'email': 'VARCHAR(64)', 'modules': 'VARCHAR(64)', 'face_feature': 'BLOB', 'voice_feature': 'BLOB'}

Process finished with exit code 0