# encoding: utf-8

from slugify import slugify

from app import admin, db
from app.model.model import Model, View


class TypeModel(db.Model, Model):
    __tablename__ = "type"
    id = db.Column(db.Integer, primary_key=True)
    code = db.Column(db.String(200))
    name = db.Column(db.String(500))
    slug = db.Column(db.String(500))

    def __repr__(self):
        return self.name


class AdminView(View):
    form_excluded_columns = ["slug"]

    def on_model_change(self, form, model, is_created):
        model.slug = slugify(model.name)


admin.add_view(AdminView(TypeModel, db.session, name="Type", category="CRUD"))