politikorama/app/model/recommendation.py

# 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")
)