1 files changed, 20 insertions, 0 deletions
diff --git a/cli/app/controllers/collection_controller.py b/cli/app/controllers/collection_controller.py
new file mode 100644
index 0000000..7206bd8
--- /dev/null
+++ b/cli/app/controllers/collection_controller.py
@@ -0,0 +1,20 @@
+from flask import request, jsonify, redirect
+from flask_classful import route
+from werkzeug.datastructures import MultiDict
+
+from app.sql.common import db, Session
+from app.sql.models.collection import Collection, CollectionForm, getMediaCounts
+from app.controllers.crud_controller import CrudView
+
+from app.data.export import export_zip
+
+class CollectionView(CrudView):
+  model = Collection
+  form = CollectionForm
+
+  def on_index(self, session, data):
+    data['counts'] = getMediaCounts(session)
+    return data
+
+  def on_destroy(self, session, item):
+    session.query(CollectionMedia).filter(CollectionMedia.collection_id == item.id).delete(synchronize_session=False)