backend: теперь универсальные вью принимают параметр для фильтрации

backend/textsouls/common/views.py

@@ -1,25 +1,27 @@
 from .. import db
 
 from flask import request
-from flask import jsonify
 from flask.views import MethodView
 
 
 class ItemAPI(MethodView):
     init_every_request = False
 
-    def __init__(self, model):
+    def __init__(self, model, filter_field):
         self.model = model
+        self.filter_field = filter_field
 
-    def _get_item(self, id):
-        return self.model.query.get_or_404(id)
+    def _get_item(self, field_value):
+        return self.model.query.filter(
+            getattr(self.model, self.filter_field) == field_value
+        ).first()
 
-    def get(self, id):
-        item = self._get_item(id)
-        return item.to_dict()
+    def get(self, field_value):
+        item = self._get_item(field_value)
+        return item.to_dict() if item else []
 
-    def delete(self, id):
-        item = self._get_item(id)
+    def delete(self, field_value):
+        item = self._get_item(field_value)
         db.session.delete(item)
         db.session.commit()
         return "", 200
@@ -28,22 +30,26 @@ class ItemAPI(MethodView):
 class ListAPI(MethodView):
     init_every_request = False
 
-    def __init__(self, model):
+    def __init__(self, model, filter_field):
         self.model = model
+        self.filter_field = filter_field
 
-    def _get_item(self, id):
-        return self.model.query.filter_by(id=id).first()
+    def _get_item(self, field_value):
+        return self.model.query.filter(
+            getattr(self.model, self.filter_field) == field_value
+        )
 
     def get(self):
         items = self.model.query.all()
-        return jsonify([item.to_dict() for item in items])
+        return [item.to_dict() for item in items]
 
     def post(self):
 
         data = request.json
 
-        if data.get("id"):
-            item = self._get_item(data["id"])
+        field_value = data.get("field_value")
+        if field_value:
+            item = self._get_item(field_value)
 
             if item:
                 return "Already exists!", 400
@@ -53,8 +59,8 @@ class ListAPI(MethodView):
         return "", 200
 
 
-def register_api(app, model, name):
-    item = ItemAPI.as_view(f"{name}-item", model)
-    group = ListAPI.as_view(f"{name}-list", model)
-    app.add_url_rule(f"/{name}/<int:id>", view_func=item)
+def register_api(app, model, name, filter_field="id"):
+    item = ItemAPI.as_view(f"{name}-item", model, filter_field)
+    group = ListAPI.as_view(f"{name}-list", model, filter_field)
+    app.add_url_rule(f"/{name}/<int:field_value>", view_func=item)
     app.add_url_rule(f"/{name}/", view_func=group)