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