82 lines
2.1 KiB
Python
82 lines
2.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
Models for lecture recorder
|
|
"""
|
|
import json
|
|
|
|
from sqlalchemy import MetaData, CheckConstraint
|
|
from sqlalchemy.exc import IntegrityError
|
|
|
|
from datetime import datetime, timedelta
|
|
|
|
from backend import db, app, login_manager
|
|
from backend.tools.scrape_rooms import scrape_rooms
|
|
|
|
metadata = MetaData()
|
|
|
|
|
|
class AccessControlEntry(db.Model):
|
|
id = db.Column(db.Integer, autoincrement=True, primary_key=True)
|
|
created_at = db.Column(db.DateTime, nullable=False, default=datetime.utcnow())
|
|
name = db.Column(db.Unicode(127), unique=False, nullable=False)
|
|
url = db.Column(db.Unicode(2047), unique=False, nullable=True, default="")
|
|
|
|
required_permission_id = db.Column(db.Integer, db.ForeignKey("permission.id"))
|
|
required_permission = db.relationship(
|
|
"Permission", back_populates="access_control_entry"
|
|
)
|
|
|
|
__table_args__ = (CheckConstraint("length(name) > 2", name="name_min_length"),)
|
|
|
|
def __init__(self, **kwargs):
|
|
super(AccessControlEntry, self).__init__(**kwargs)
|
|
|
|
@staticmethod
|
|
def get_by_name(name):
|
|
"""
|
|
Find ace by name
|
|
:param name:
|
|
:return:
|
|
"""
|
|
return AccessControlEntry.query.filter(AccessControlEntry.name == name).first()
|
|
|
|
@staticmethod
|
|
def get_all():
|
|
"""
|
|
Return all ace
|
|
:return:
|
|
"""
|
|
return AccessControlEntry.query.all()
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
def to_dict(self):
|
|
return dict(id=self.id, name=self.name)
|
|
|
|
def toJSON(self):
|
|
return json.dumps(
|
|
self.to_dict(), default=lambda o: o.__dict__, sort_keys=True, indent=4
|
|
)
|
|
|
|
|
|
def pre_fill_table():
|
|
a_es = {
|
|
"url": "",
|
|
}
|
|
access_entries = [
|
|
AccessControlEntry(
|
|
name=room["name"],
|
|
number=room["room_number"],
|
|
building_name=room["building_name"],
|
|
building_number=room["building_number"],
|
|
)
|
|
for room in a_es
|
|
]
|
|
|
|
try:
|
|
db.session.bulk_save_objects(access_entries)
|
|
db.session.commit()
|
|
except IntegrityError as e:
|
|
db.session.rollback()
|