Когда атрибуты класса ORM sqlalchemy отличаются от столбцов базы данных, как получить список атрибутов класса ORM?

Предположим, вы хотите перебрать атрибуты ORM класса ORM в sqlalchemy. Итак, вам нужен список атрибутов ORM. Как вы получаете этот список?

Если класс ORM не переименовывает атрибуты, и поэтому атрибуты ORM соответствуют столбцам базы данных, вы можете использовать это решение из: qaru.site/info/45400/... (Кстати, в файле исходного кода /lib/sqlalchemy/orm/base.py есть встроенная (закрытая) функция _orm_columns(), которая, как представляется, предоставляет эту функцию)

Но если класс ORM python имеет разные имена, чем столбцы базы данных (например, в этих трех атрибутах ORM):

>>> class User(Base):
...     __tablename__ = 'users'
...
...     id = Column('pkey', Integer, primary_key=True)
...     name = Column('user_name', String)
...     fullname = Column('human_name', String)

то этот метод не работает. Итак, как вы получите версию python атрибутов ORM?

Ответы

Ответ 1

Это уже реализовано с помощью :

from sqlalchemy import *
from sqlalchemy.orm import *
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import inspect

Base = declarative_base()

class A(Base):
    __tablename__ = 'a'
    id = Column(Integer, primary_key=True)
    x = Column(Integer)
    y = Column(Integer)

print inspect(A).c
print inspect(A).c.x
print inspect(A).column_attrs
print inspect(A).column_attrs.x
print inspect(A).column_attrs.x.expression

http://docs.sqlalchemy.org/en/latest/core/inspection.html

http://docs.sqlalchemy.org/en/latest/orm/mapping_api.html#sqlalchemy.orm.mapper.Mapper.all_orm_descriptors

http://docs.sqlalchemy.org/en/latest/orm/mapping_api.html#sqlalchemy.orm.mapper.Mapper.columns

http://docs.sqlalchemy.org/en/latest/orm/mapping_api.html#sqlalchemy.orm.mapper.Mapper.column_attrs

http://docs.sqlalchemy.org/en/latest/orm/mapping_api.html#sqlalchemy.orm.mapper.Mapper.c

Ответ 2

Я определил класс mixin для увеличения

Base = declarative_base()

с классом mixin, как:

import inspect
import sqlalchemy as sqla 

class orm_mixin_class(object):
    """Additional functionality wanted in orm classes (that inherit from Base)."""

    @classmethod
    def orm_class_info(cls):
        """
        Get info about the orm class (as opposed to the database table it is mapped to).

        Returns:
            list of dict where each dict describes a orm class attribute.  
            Keys: s_class_attribute_name
        """
        o_leaf_level_class = cls   # this turns out the be the leaf class instead of this mixin class
        l_orm_attribute_pairs = inspect.getmembers(o_leaf_level_class)

        l_orm_class_info_dicts = []
        for (s_class_attribute_name, o_attr_type) in l_orm_attribute_pairs:
            d_attr_info = {}
            b_orm_attribute = False
            try:
                o_inspect = sqla.inspection.inspect(o_attr_type) 

                if isinstance(o_inspect, sqla.orm.attributes.QueryableAttribute):
                    b_orm_attribute = True
                    d_attr_info['s_class_attribute_name'] = s_class_attribute_name
            except:
                pass            

            if b_orm_attribute:
                # only orm attributes have an entry in the output list
                l_orm_class_info_dicts.append(d_attr_info)

        return(l_orm_class_info_dicts)

поэтому список атрибутов ORM можно легко получить из вызова метода.

Объявление класса ORM теперь:

class User(Base, orm_mixin_class):