mirror of
https://github.com/Llloooggg/TextSouls.git
synced 2026-03-06 04:26:23 +03:00
backend: полная реструктуризация
This commit is contained in:
74
backend/textsouls/common/api.py
Normal file
74
backend/textsouls/common/api.py
Normal file
@@ -0,0 +1,74 @@
|
||||
from flask import request
|
||||
from flask.views import MethodView
|
||||
|
||||
from flask_admin.contrib.sqla import ModelView
|
||||
|
||||
from textsouls import db
|
||||
|
||||
|
||||
class CommonAdminView(ModelView):
|
||||
def __init__(self, model, *args, **kwargs):
|
||||
self.column_list = [c.key for c in model.__table__.columns]
|
||||
super(CommonAdminView, self).__init__(model, *args, **kwargs)
|
||||
|
||||
|
||||
class ItemAPI(MethodView):
|
||||
init_every_request = False
|
||||
|
||||
def __init__(self, model, filter_field):
|
||||
self.model = model
|
||||
self.filter_field = filter_field
|
||||
|
||||
def _get_item(self, field_value):
|
||||
return self.model.query.filter(
|
||||
getattr(self.model, self.filter_field) == field_value
|
||||
).first()
|
||||
|
||||
def get(self, field_value):
|
||||
item = self._get_item(field_value)
|
||||
return item.to_dict() if item else []
|
||||
|
||||
def delete(self, field_value):
|
||||
item = self._get_item(field_value)
|
||||
db.session.delete(item)
|
||||
db.session.commit()
|
||||
return "", 200
|
||||
|
||||
|
||||
class ListAPI(MethodView):
|
||||
init_every_request = False
|
||||
|
||||
def __init__(self, model, filter_field):
|
||||
self.model = model
|
||||
self.filter_field = filter_field
|
||||
|
||||
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 [item.to_dict() for item in items]
|
||||
|
||||
def post(self):
|
||||
|
||||
data = request.json
|
||||
|
||||
field_value = data.get("field_value")
|
||||
if field_value:
|
||||
item = self._get_item(field_value)
|
||||
|
||||
if item:
|
||||
return "Already exists!", 400
|
||||
|
||||
db.session.add(self.model(**data))
|
||||
db.session.commit()
|
||||
return "", 200
|
||||
|
||||
|
||||
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)
|
||||
Reference in New Issue
Block a user