sqlalchemy对象转dict的示例

程序员文章站 2022-06-01 21:20:58

...

复制代码代码如下:

def sa_obj_to_dict(obj, filtrate=None, rename=None):
"""
sqlalchemy 对象转为dict
:param filtrate: 过滤的字段
:type filtrate: list or tuple
:param rename: 需要改名的,改名在过滤之后处理, key为原来对象的属性名称，value为需要更改名称
:type rename: dict
:rtype: dict
"""

if isinstance(obj.__class__, DeclarativeMeta):
# an SQLAlchemy class
#该类的相关类型，即直接与间接父类
cla = obj.__class__.__mro__
#过滤不需要的父类
cla = filter(lambda c: hasattr(c, '__table__'), filter(lambda c: isinstance(c, DeclarativeMeta), cla))
columns = []
map(lambda c: columns.extend(c.__table__.columns), cla[::-1])
# columns = obj.__table__.columns
if filtrate and isinstance(filtrate, (list, tuple)):
fields = dict(map(lambda c: (c.name, getattr(obj, c.name)), filter(lambda c: not c.name in filtrate, columns)))
else:
fields = dict(map(lambda c: (c.name, getattr(obj, c.name)), columns))
# fields = dict([(c.name, getattr(obj, c.name)) for c in obj.__table__.columns])
if rename and isinstance(rename, dict):
#先移除key和value相同的项
_rename = dict(filter(lambda (k, v): str(k) != str(v), rename.iteritems()))
#如果原始key不存在，那么新的key对应的值默认为None
#如果新的key已存在于原始key中，那么原始key的值将被新的key的值覆盖
# map(lambda (k, v): fields.setdefault(v, fields.pop(k, None)), _rename.iteritems())
map(lambda (k, v): fields.update({v: fields.pop(k, None)}), _rename.iteritems())
#
return fields
else:
return {}