sqlalchemy模型的已定义列上迭代的方法?

2020/12/07 06:22 · python ·  · 0评论

我一直在尝试找出如何遍历SQLAlchemy模型中定义的列的列表。我想要为一些模型编写一些序列化和复制方法。我不能仅对其进行迭代,obj.__dict__因为它包含很多SA特定项。

有人知道一种从以下内容中获取iddesc名称的方法吗?

class JobStatus(Base):
    __tablename__ = 'jobstatus'

    id = Column(Integer, primary_key=True)
    desc = Column(Unicode(20))

在这种情况下,我可以轻松创建一个:

def logme(self):
    return {'id': self.id, 'desc': self.desc}

但我更喜欢自动生成dict(对于较大的对象)的东西。

您可以使用以下功能:

def __unicode__(self):
    return "[%s(%s)]" % (self.__class__.__name__, ', '.join('%s=%s' % (k, self.__dict__[k]) for k in sorted(self.__dict__) if '_sa_' != k[:4]))

它将排除SA魔术属性,但不排除关系。因此,基本上它可能会加载依赖项,父项,子项等,这绝对是不可取的。

但这实际上要容易得多,因为如果您继承自Base,则拥有一个__table__属性,因此您可以执行以下操作:

for c in JobStatus.__table__.columns:
    print c

for c in JobStatus.__table__.foreign_keys:
    print c

请参阅如何从SQLAlchemy映射的对象中发现表属性-类似的问题。

迈克(Mike)编辑:请参阅Mapper.cMapper.mapped_table之类的函数如果使用0.8或更高版本,还请参见Mapper.attrs和相关函数。

Mapper.attrs的示例

from sqlalchemy import inspect
mapper = inspect(JobStatus)
for column in mapper.attrs:
    print column.key

您可以从映射器获取已定义属性的列表。对于您的情况,您仅对ColumnProperty对象感兴趣。

from sqlalchemy.orm import class_mapper
import sqlalchemy

def attribute_names(cls):
    return [prop.key for prop in class_mapper(cls).iterate_properties
        if isinstance(prop, sqlalchemy.orm.ColumnProperty)]

我意识到这是一个古老的问题,但是我遇到了相同的要求,并希望为将来的读者提供替代解决方案。

如Josh所述,完整的SQL字段名称将由返回JobStatus.__table__.columns,因此您将获得jobstatus.id而不是原始的字段名称 id没有那么有用。

获取最初定义的字段名称列表的解决方案是_data在包含完整数据的列对象上查找属性。如果我们看一下JobStatus.__table__.columns._data,它看起来像这样:

{'desc': Column('desc', Unicode(length=20), table=<jobstatus>),
 'id': Column('id', Integer(), table=<jobstatus>, primary_key=True, nullable=False)}

从这里您可以简单地致电JobStatus.__table__.columns._data.keys()给您一个干净的清单:

['id', 'desc']

self.__table__.columns将“仅”为您提供在该特定类中定义的列,即没有继承的列。如果您需要全部,请使用self.__mapper__.columns在您的示例中,我可能会使用以下内容:

class JobStatus(Base):

    ...

    def __iter__(self):
        values = vars(self)
        for attr in self.__mapper__.columns.keys():
            if attr in values:
                yield attr, values[attr]

    def logme(self):
        return dict(self)

假设您正在使用SQLAlchemy的声明性映射,则可以使用__mapper__attribute来获取类映射器。要获取所有映射的属性(包括关系):

obj.__mapper__.attrs.keys()

如果需要严格的列名,请使用obj.__mapper__.column_attrs.keys()有关其他视图,请参见文档。

https://docs.sqlalchemy.org/zh_CN/latest/orm/mapping_api.html#sqlalchemy.orm.mapper.Mapper.attrs

为了获得as_dict所有班级方法,我使用了一个Mixin使用Ants Aasma描述的技术的班级

class BaseMixin(object):                                                                                                                                                                             
    def as_dict(self):                                                                                                                                                                               
        result = {}                                                                                                                                                                                  
        for prop in class_mapper(self.__class__).iterate_properties:                                                                                                                                 
            if isinstance(prop, ColumnProperty):                                                                                                                                                     
                result[prop.key] = getattr(self, prop.key)                                                                                                                                           
        return result

然后在课堂上像这样使用它

class MyClass(BaseMixin, Base):
    pass

这样,您可以在的实例上调用以下内容MyClass

> myclass = MyClass()
> myclass.as_dict()

希望这可以帮助。


我对此进行了进一步的研究,实际上我需要将实例渲染为HAL对象dict的形式,并带有指向相关对象的链接。因此,我在这里添加了这个小技巧,它将覆盖与上述相同的类的所有属性,不同之处在于,我将更深入地搜索属性并自动生成这些属性Relaionshiplinks

请注意,这仅适用于具有单个主键的关系

from sqlalchemy.orm import class_mapper, ColumnProperty
from functools import reduce


def deepgetattr(obj, attr):
    """Recurses through an attribute chain to get the ultimate value."""
    return reduce(getattr, attr.split('.'), obj)


class BaseMixin(object):
    def as_dict(self):
        IgnoreInstrumented = (
            InstrumentedList, InstrumentedDict, InstrumentedSet
        )
        result = {}
        for prop in class_mapper(self.__class__).iterate_properties:
            if isinstance(getattr(self, prop.key), IgnoreInstrumented):
                # All reverse relations are assigned to each related instances
                # we don't need to link these, so we skip
                continue
            if isinstance(prop, ColumnProperty):
                # Add simple property to the dictionary with its value
                result[prop.key] = getattr(self, prop.key)
            if isinstance(prop, RelationshipProperty):
                # Construct links relaions
                if 'links' not in result:
                    result['links'] = {}

                # Get value using nested class keys
                value = (
                    deepgetattr(
                        self, prop.key + "." + prop.mapper.primary_key[0].key
                    )
                )
                result['links'][prop.key] = {}
                result['links'][prop.key]['href'] = (
                    "/{}/{}".format(prop.key, value)
                )
        return result
self.__dict__

返回一个dict,其中键是属性名称,其值是对象的值。

/!\有一个补充属性:'_sa_instance_state',但是您可以处理它:)

我想动态获取模型的特定实例的数据。我用了这段代码。

def to_json(instance):
    # get columns data
    data = {}
    columns = list(instance.__table__.columns)
    for column in columns:
        data[column.name] = instance.__dict__[column.name]
    return data

我知道这是一个古老的问题,但是:

class JobStatus(Base):

    ...

    def columns(self):
        return [col for col in dir(self) if isinstance(col, db.Column)]

然后,获取列名: jobStatus.columns()

那会回来 ['id', 'desc']

然后,您可以遍历,并对列和值进行处理:

for col in jobStatus.colums():
    doStuff(getattr(jobStatus, col))
本文地址:http://python.askforanswer.com/sqlalchemymoxingdeyidingyilieshangdiedaidefangfa.html
文章标签: ,  
版权声明:本文为原创文章,版权归 admin 所有,欢迎分享本文,转载请保留出处!

文件下载

老薛主机终身7折优惠码boke112

上一篇:
下一篇:

评论已关闭!