politikorama/app/model/entity.py

38 lines
1.1 KiB
Python

# encoding: utf-8
from slugify import slugify
from app import admin, db
from app.model.model import Model, View
class EntityModel(db.Model, Model):
__tablename__ = "entity"
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(200))
slug = db.Column(db.String(200))
code = db.Column(db.String(20))
picture = db.Column(db.String(2000))
type_id = db.Column(db.Integer, db.ForeignKey("type.id"))
type = db.relationship("TypeModel")
start = db.Column(db.Date)
end = db.Column(db.Date)
country_id = db.Column(db.Integer, db.ForeignKey("country.id"))
country = db.relationship("CountryModel")
parent_id = db.Column(db.Integer, db.ForeignKey("entity.id"))
parent = db.relationship("EntityModel")
def __repr__(self):
return self.name
class AdminView(View):
column_default_sort = "name"
column_filters = ["name", "code", "type.name", "country.name"]
def on_model_change(self, form, model, is_created):
model.slug = slugify(model.name)
admin.add_view(AdminView(EntityModel, db.session, name="Entity", category="CRUD"))