# -*- 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()