def remove_stylesheet(template_id: int):
template: Template = get_or_404(
db.session.query(Template).filter_by(id=template_id))
data = TemplateSchema.all_request_data()
if isinstance(data, list):
for entry in data:
if not hasattr(entry, "id"):
raise BadRequest("Required property: id")
delete_relation = template_stylesheet_relation.delete().where(
and_(
template_stylesheet_relation.c.template_id == template.id,
template_stylesheet_relation.c.style_id == entry.get("id"),
))
db.session.execute(delete_relation)
elif isinstance(data, dict):
delete_relation = template_stylesheet_relation.delete().where(
and_(
template_stylesheet_relation.c.template_id == template.id,
template_stylesheet_relation.c.style_id == data.get("id"),
))
db.session.execute(delete_relation)
db.session.commit()
template_schema = TemplateSchema(include=("styles", ),
instance=template,
session=db.session)
return jsonify(template_schema.dump(template))
def add_stylesheet(template_id: int):
template: Template = get_or_404(
db.session.query(Template).filter_by(id=template_id))
template_schema = TemplateSchema(include=("styles", ))
data = TemplateSchema.all_request_data()
if isinstance(data, list):
for entry in data:
style: Stylesheet = get_or_404(
db.session.query(Stylesheet).filter_by(**entry))
template.styles.append(style)
else:
style: Stylesheet = get_or_404(
db.session.query(Stylesheet).filter_by(**data))
template.styles.append(style)
db.session.commit()
return jsonify(template_schema.dump(template))