代碼如下:
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 {}
新聞熱點
疑難解答