politikorama/app/model/recommendation.py

44 lines
1.3 KiB
Python

# encoding: utf-8
from slugify import slugify
from app import admin, db
from app.model.model import Model, View
class RecommendationModel(db.Model, Model):
__tablename__ = "recommendation"
id = db.Column(db.Integer, primary_key=True)
matter_id = db.Column(db.Integer, db.ForeignKey("matter.id"))
matter = db.relationship(
"MatterModel", backref=db.backref("recommendations", lazy="dynamic")
)
entity_id = db.Column(db.Integer, db.ForeignKey("entity.id"))
entity = db.relationship(
"EntityModel", backref=db.backref("recommendations", lazy="dynamic")
)
name = db.Column(db.String(200))
slug = db.Column(db.String(200))
code = db.Column(db.String(20))
date = db.Column(db.Date)
description = db.Column(db.Text)
value = db.Column(db.String(200))
weight = db.Column(db.Integer, default=1)
def __repr__(self):
return f"{self.name} - {self.value} ({self.weight})"
class AdminView(View):
column_default_sort = "name"
column_filters = ["name", "matter.name"]
form_columns = ["matter", "entity", "name", "code", "date", "description", "value", "weight"]
def on_model_change(self, form, model, is_created):
model.slug = slugify(model.name)
admin.add_view(
AdminView(RecommendationModel, db.session, name="Recommendation", category="CRUD")
)