Chore:Automatically serialize models

Author: krmroland

api/core/Flask.py

@@ -1,9 +1,12 @@
 from flask import Flask as BaseFlask, request
 from .routing.Router import Router
+from .JSONEncoder import JSONEncoder
 
 
 class Flask(BaseFlask):
 
+    json_encoder = JSONEncoder
+
     def dispatch_request(self):
         """triggers an action related  to this request"""
         adapter = self.create_url_adapter(request)

api/core/JSONEncoder.py

@@ -0,0 +1,9 @@
+from flask.json import JSONEncoder as BaseJSONEncoder
+from api.core.storage import Model, ModelCollection
+
+
+class JSONEncoder(BaseJSONEncoder):
+    def default(self, data):
+        if isinstance(data, Model) or isinstance(data, ModelCollection):
+            return data.to_json()
+        return BaseJSONEncoder.default(self, data)

api/core/storage/storage.py

@@ -69,6 +69,9 @@ def count(self):
     def __len__(self):
         return self.count()
 
+    def to_json(self):
+        return [model.to_json() for model in self.models]
+
     def first(self):
         try:
             return self.models[0]
@@ -84,6 +87,7 @@ def __repr__(self):
 
 class Model:
     timestamps = True
+    hidden = []
 
     @classmethod
     def table_name(cls):
@@ -161,6 +165,14 @@ def or_where(cls, *args, **kwargs):
     def hydrate(cls, models):
         return list(map(cls, models))
 
+    def to_json(self):
+
+        return {
+            key: self.attributes[key]
+            for key in self.attributes
+            if (key not in self.hidden)
+        }
+
     @classmethod
     def all(cls):
         models = cls.hydrate(Storage.get_table_data(cls.table_name()).values())