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