mirror of
https://github.com/lucaspalomodevelop/eventcally.git
synced 2026-03-13 08:09:37 +00:00
50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
from functools import wraps
|
|
|
|
from flask import request
|
|
from flask_apispec import marshal_with
|
|
from flask_apispec.views import MethodResource
|
|
|
|
from project import db
|
|
from project.api.schemas import ErrorResponseSchema, UnprocessableEntityResponseSchema
|
|
|
|
|
|
def etag_cache(func):
|
|
@wraps(func)
|
|
def wrapper(*args, **kwargs):
|
|
response = func(*args, **kwargs)
|
|
response.add_etag()
|
|
return response.make_conditional(request)
|
|
|
|
return wrapper
|
|
|
|
|
|
@marshal_with(ErrorResponseSchema, 400, "Bad Request")
|
|
@marshal_with(UnprocessableEntityResponseSchema, 422, "Unprocessable Entity")
|
|
class BaseResource(MethodResource):
|
|
decorators = [etag_cache]
|
|
|
|
def create_instance(self, schema_cls, **kwargs):
|
|
instance = schema_cls().load(request.json, session=db.session)
|
|
|
|
for key, value in kwargs.items():
|
|
if hasattr(instance, key):
|
|
setattr(instance, key, value)
|
|
|
|
validate = getattr(instance, "validate", None)
|
|
if callable(validate):
|
|
validate()
|
|
|
|
return instance
|
|
|
|
def update_instance(self, schema_cls, instance):
|
|
with db.session.no_autoflush:
|
|
instance = schema_cls().load(
|
|
request.json, session=db.session, instance=instance
|
|
)
|
|
|
|
validate = getattr(instance, "validate", None)
|
|
if callable(validate):
|
|
validate()
|
|
|
|
return instance
|