base modification in model for category support

1 files changed, 31 insertions, 0 deletions

diff --git a/pyaggr3g470r/views/api/category.py b/pyaggr3g470r/views/api/category.py
new file mode 100644
index 00000000..31f46751
--- /dev/null
+++ b/pyaggr3g470r/views/api/category.py
@@ -0,0 +1,31 @@
+from flask import g
+
+from pyaggr3g470r.controllers.category import CategoryController
+from pyaggr3g470r.views.api.common import (PyAggResourceNew,
+                                           PyAggResourceExisting,
+                                           PyAggResourceMulti)
+
+
+CAT_ATTRS = {'name': {'type': str},
+             'user_id': {'type': int}}
+
+
+class CategoryNewAPI(PyAggResourceNew):
+    controller_cls = CategoryController
+    attrs = CAT_ATTRS
+
+
+class CategoryAPI(PyAggResourceExisting):
+    controller_cls = CategoryController
+    attrs = CAT_ATTRS
+
+
+class CategoriesAPI(PyAggResourceMulti):
+    controller_cls = CategoryController
+    attrs = CAT_ATTRS
+
+
+g.api.add_resource(CategoryNewAPI, '/category', endpoint='category_new.json')
+g.api.add_resource(CategoryAPI, '/category/<int:obj_id>',
+                   endpoint='category.json')
+g.api.add_resource(CategoriesAPI, '/categories', endpoint='categories.json')