Чтобы получить метод 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
person
flazzarini
schedule
20.08.2015
