remove custom json serializer

This commit is contained in:
Martin
2016-04-26 12:18:11 +02:00
parent bb251ec731
commit bbb9ab341b
4 changed files with 10 additions and 55 deletions
-4
View File
@@ -1,13 +1,9 @@
from flask import Flask, jsonify
from flask_sqlalchemy import SQLAlchemy
from api.helpers import JSONEncoder
app = Flask(__name__)
app.config.from_object('config')
# enable this when json encoding works
# app.json_encoder = JSONEncoder
def on_404(e):
return jsonify(dict(error='Not found')), 404
-1
View File
@@ -1 +0,0 @@
from .json import JSONEncoder, JsonSerializer
-46
View File
@@ -1,46 +0,0 @@
from flask.json import JSONEncoder as BaseJSONEncoder
class JSONEncoder(BaseJSONEncoder):
"""Custom :class:`JSONEncoder` which respects objects that include the
:class:`JsonSerializer` mixin.
"""
def default(self, obj):
if isinstance(obj, JsonSerializer):
return obj.to_json()
return super(JSONEncoder, self).default(obj)
class JsonSerializer(object):
"""A mixin that can be used to mark a SQLAlchemy model class which
implements a :func:`to_json` method. The :func:`to_json` method is used
in conjuction with the custom :class:`JSONEncoder` class. By default this
mixin will assume all properties of the SQLAlchemy model are to be visible
in the JSON output. Extend this class to customize which properties are
public, hidden or modified before being being passed to the JSON serializer.
"""
__json_public__ = None
__json_hidden__ = None
__json_modifiers__ = None
def get_field_names(self):
for p in self.__mapper__.iterate_properties:
yield p.key
def to_json(self):
field_names = self.get_field_names()
public = self.__json_public__ or field_names
hidden = self.__json_hidden__ or []
modifiers = self.__json_modifiers__ or dict()
rv = dict()
for key in public:
rv[key] = getattr(self, key)
for key, modifier in modifiers.items():
value = getattr(self, key)
rv[key] = modifier(value, self)
for key in hidden:
rv.pop(key, None)
return rv
+10 -4
View File
@@ -1,9 +1,15 @@
from api import db
from api.helpers import JsonSerializer
class Designer(JsonSerializer, db.Model):
class Designer(db.Model):
__tablename__ = 'designers'
designerid = db.Column('designerid', db.Integer, primary_key=True)
designerid = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(100), nullable=False)
path = db.Column(db.String(100), nullable=False)
path = db.Column(db.String(100), nullable=False)
def to_json(self, nested=False):
return {
'designerid': self.designerid,
'name': self.name,
'path': self.path,
}