序列化 SqlAlchemy 的结果为 json 字符串

http://my.oschina.net/gongshang/blog/395431?p=1

在设计 RESTful 的网站时,我们总是希望 ORM 框架返回的结果是可以直接给 View 层使用的 JSON 字符串。但是 Sqlalchemy 的返回结果直接使用 json.dumps(res) 会报 TypeError 的错误。

解决方案

其实就像遇到 datetime 无法使用 json.dumps() 序列化的问题一样,我们需要继承一个json.JSONEncoder

  1.  
     
  2.  
    from sqlalchemy.ext.declarative import DeclarativeMeta
  3.  
    class AlchemyEncoder(json.JSONEncoder):
  4.  
    def default(self, obj):
  5.  
    if isinstance(obj.__class__, DeclarativeMeta):
  6.  
    # an SQLAlchemy class
  7.  
    fields = {}
  8.  
    for field in [x for x in dir(obj) if not x.startswith('_') and x != 'metadata']:
  9.  
    data = obj.__getattribute__(field)
  10.  
    try:
  11.  
    json.dumps(data) # this will fail on non-encodable values, like other classes
  12.  
    fields[field] = data
  13.  
    except TypeError: # 添加了对datetime的处理
  14.  
    if isinstance(data, datetime.datetime):
  15.  
    fields[field] = data.isoformat()
  16.  
    elif isinstance(data, datetime.date):
  17.  
    fields[field] = data.isoformat()
  18.  
    elif isinstance(data, datetime.timedelta):
  19.  
    fields[field] = (datetime.datetime.min + data).time().isoformat()
  20.  
    else:
  21.  
    fields[field] = None
  22.  
    # a json-encodable dict
  23.  
    return fields
  24.  
     
  25.  
    return json.JSONEncoder.default(self, obj)

参考地址: stackoverflow ,在其基础上添加了对datatime的一个简单处理。

调用的时候如下:

  1.  
    c = YourAlchemyClass()
  2.  
    print json.dumps(c, cls=AlchemyEncoder)

猜你喜欢

转载自www.cnblogs.com/ExMan/p/10313230.html