politikorama/app/model/membership.py

39 lines
1.2 KiB
Python

# encoding: utf-8
from app import admin, db
from app.model.model import Model, View
class MembershipModel(db.Model, Model):
__tablename__ = "membership"
id = db.Column(db.Integer, primary_key=True)
representative_id = db.Column(db.Integer, db.ForeignKey("representative.id"))
representative = db.relationship(
"RepresentativeModel", backref=db.backref("memberships", lazy="dynamic")
)
role_id = db.Column(db.Integer, db.ForeignKey("role.id"))
role = db.relationship(
"RoleModel", backref=db.backref("memberships", lazy="dynamic")
)
start = db.Column(db.Date)
end = db.Column(db.Date)
entity_id = db.Column(db.Integer, db.ForeignKey("entity.id"))
entity = db.relationship(
"EntityModel", backref=db.backref("memberships", lazy="dynamic")
)
@property
def timestamp(self):
if self.end is None:
return 9999999999 + self.start.timestamp()
else:
return self.end.timestamp()
class AdminView(View):
column_default_sort = [("representative.name", False), ("entity.name", False)]
column_filters = ["representative.name", "entity.name", "role.name"]
admin.add_view(AdminView(MembershipModel, db.session, name="Membership", category="CRUD"))