如何从sqlalchemy中的关系获取列表的字典？

Question

我发现你可以在关系中使用集合来改变返回值的类型，特别是我对字典感兴趣。 文档举例：

class Item(Base): 
    __tablename__ = 'item' 
    id = Column(Integer, primary_key=True) 
    notes = relationship("Note", 
         collection_class=attribute_mapped_collection('keyword'), 
         cascade="all, delete-orphan") 

class Note(Base): 
    __tablename__ = 'note' 
    id = Column(Integer, primary_key=True) 
    item_id = Column(Integer, ForeignKey('item.id'), nullable=False) 
    keyword = Column(String) 
    text = Column(String) 

它的工作原理。不过，我希望如果不止有一个同名的关键字，它会创建列表值。但它只将最后一个值放在唯一键名下。

下面是一个例子：

|    Note table    | 
|---------------------|------------------| 
|   id   |  keyword  | 
|---------------------|------------------| 
|   1   |  foo  | 
|---------------------|------------------| 
|   2   |  foo  | 
|---------------------|------------------| 
|   3   |  bar  | 
|---------------------|------------------| 
|   4   |  bar  | 
|---------------------|------------------| 

item.notes将返回这样的事情：

{'foo': <project.models.note.Note at 0x7fc6840fadd2>, 
'bar': <project.models.note.Note at 0x7fc6840fadd4>} 

其中foo和酒吧对象的ID分别为2和4。

什么我要找的是让这样的事情：

{'foo': [<project.models.note.Note at 0x7fc6840fadd1, 
      <project.models.note.Note at 0x7fc6840fadd2>], 
'bar': [<project.models.note.Note at 0x7fc6840fadd3>, 
     <project.models.note.Note at 0x7fc6840fadd4>]} 

是否有可能从SQLAlchemy的关系得到清单字典？

Answer 1

因此，事实证明，您可以简单地继承MappedCollection并在setitem处执行您喜欢的任何操作。

from sqlalchemy.orm.collections import (MappedCollection, 
             _SerializableAttrGetter, 
             collection, 
             _instrument_class) 

#This will ensure that the MappedCollection has been properly 
#initialized with custom __setitem__() and __delitem__() methods 
#before used in a custom subclass 
_instrument_class(MappedCollection) 


class DictOfListsCollection(MappedCollection): 

    @collection.internally_instrumented 
    def __setitem__(self, key, value, _sa_initiator=None): 
     if not super(DictOfListsCollection, self).get(key): 
      super(DictOfListsCollection, self).__setitem__(key, [], _sa_initiator) 
     super(DictOfListsCollection, self).__getitem__(key).append(value)