模型序列化

app/__init__.py

@@ -2,4 +2,28 @@
 create by gaowenfeng on 
 """
 
+from .app import Flask
+
 __author__ = "gaowenfeng"
+
+
+def register_blueprint(app):
+    from app.api.v1 import create_blueprint_v1
+    app.register_blueprint(create_blueprint_v1(), url_prefix='/v1')
+
+
+def register_plug(app):
+    from app.models.base import db
+    db.init_app(app)
+    with app.app_context():
+        db.create_all()
+
+
+def create_app():
+    app = Flask(__name__)
+    app.config.from_object('app.config.secure')
+    app.config.from_object('app.config.setting')
+
+    register_blueprint(app)
+    register_plug(app)
+    return app

app/api/v1/user.py

@@ -17,8 +17,4 @@
 @auth.login_required
 def get_user(uid):
     user = User.query.get_or_404(uid)
-    r = {
-        'nickname': user.nickname,
-        'email': user.email
-    }
-    return jsonify(r), 200
+    return jsonify(user), 200

app/app.py

@@ -1,29 +1,30 @@
 """
 create by gaowenfeng on 
 """
-from flask import Flask
+from flask import Flask as _Flask
+from flask.json import JSONEncoder as _JSONEncoder
+
+from datetime import date
+
+from app.libs.error_code import ServerError
 
 __author__ = "gaowenfeng"
 
 
-def register_blueprint(app):
-    from app.api.v1 import create_blueprint_v1
-    app.register_blueprint(create_blueprint_v1(), url_prefix='/v1')
+class JSONEncoder(_JSONEncoder):
+
+    def default(self, o):
+        if hasattr(o, 'keys') and hasattr(o, '__getitem__'):
+            return dict(o)
+        # 兼容其他的序列化
+        if isinstance(o, date):
+            return o.strftime('%Y-%m-%d')
+        raise ServerError()
 
 
-def register_plug(app):
-    from app.models.base import db
-    db.init_app(app)
-    with app.app_context():
-        db.create_all()
+class Flask(_Flask):
+    json_encoder = JSONEncoder
 
 
-def create_app():
-    app = Flask(__name__)
-    app.config.from_object('app.config.secure')
-    app.config.from_object('app.config.setting')
 
-    register_blueprint(app)
-    register_plug(app)
-    return app
 

app/models/base.py

@@ -54,6 +54,9 @@ class Base(db.Model):
     def __init__(self):
         self.create_time = int(datetime.now().timestamp())
 
+    def __getitem__(self, item):
+        return getattr(self, item)
+
     def set_attrs(self, attrs_dict):
         for key, value in attrs_dict.items():
             if hasattr(self, key) and key != 'id':

app/models/user.py

@@ -17,6 +17,9 @@ class User(Base):
     nickname = Column(String(24), nullable=False)
     _password = Column('password', String(128))
 
+    def keys(self):
+        return ('id', 'email', 'nickname')
+
     @property
     def password(self):
         return self._password

app/test/test_dict.py

@@ -0,0 +1,23 @@
+"""
+create by gaowenfeng on 
+"""
+
+__author__ = "gaowenfeng"
+
+
+class Person:
+    name = 'gwf'
+    age = 18
+
+    def __init__(self):
+        self.gender = 'male'
+
+    def keys(self):
+        return ('name', 'age', 'gender')
+
+    def __getitem__(self, item):
+        return getattr(self, item)
+
+
+o = Person()
+print(dict(o))

ginger.py

@@ -3,7 +3,7 @@
 """
 from werkzeug.exceptions import HTTPException
 
-from app.app import create_app
+from app import create_app
 from app.libs.error_code import ServerError
 from app.libs.errors import APIException