Чтобы получить метод 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()
Надеюсь, это поможет.
Я немного поиграл с этим, мне действительно нужно было представить мои экземпляры как dict как форму HAL-объекта с его ссылками на связанные объекты. Итак, я добавил сюда маленькую магию, которая будет сканировать все свойства класса, такие же, как и выше, с той разницей, что я углублюсь в свойства Relaionship и сгенерирую links для них автоматически.
Обратите внимание, что это будет работать только для отношений, имеющих один первичный ключ
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