98 lines
3.1 KiB
Python
98 lines
3.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
Models for lecture recorder
|
|
"""
|
|
import json
|
|
import logging
|
|
|
|
from sqlalchemy import MetaData, CheckConstraint, UniqueConstraint
|
|
from sqlalchemy.exc import IntegrityError
|
|
|
|
from datetime import datetime, timedelta
|
|
|
|
from backend import db, app, login_manager
|
|
from backend.tools.campus_rooms import get_campus_rooms
|
|
|
|
logger = logging.getLogger("lrc." + __name__)
|
|
|
|
metadata = MetaData()
|
|
|
|
|
|
class Room(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)
|
|
alternate_name = db.Column(db.Unicode(127), unique=False, nullable=True, default=None)
|
|
external_id = db.Column(db.String, nullable=True, unique=True)
|
|
coordinates = db.Column(db.String, nullable=True)
|
|
comment = db.Column(db.Unicode(2047), unique=False, nullable=True, default="")
|
|
number = db.Column(db.Unicode(63), unique=False, nullable=True, default=None)
|
|
building_name = db.Column(db.Unicode(63), unique=False, nullable=True)
|
|
building_number = db.Column(db.Unicode(63), unique=False, nullable=True, default=None)
|
|
|
|
recorder = db.relationship('Recorder', uselist=False, back_populates='room') # one-to-one relation (uselist=False)
|
|
|
|
__table_args__ = (
|
|
CheckConstraint('length(name) > 2',
|
|
name='name_min_length'),
|
|
UniqueConstraint('name', 'number', 'building_number', 'external_id'),
|
|
)
|
|
|
|
def __init__(self, **kwargs):
|
|
super(Room, self).__init__(**kwargs)
|
|
|
|
@staticmethod
|
|
def get_by_name(name):
|
|
"""
|
|
Find group by name
|
|
:param name:
|
|
:return:
|
|
"""
|
|
return Room.query.filter(Room.name == name).first()
|
|
|
|
@staticmethod
|
|
def get_by_building_number(building_number):
|
|
"""
|
|
Find group by name
|
|
:param name:
|
|
:return:
|
|
"""
|
|
return Room.query.filter(Room.building_number == building_number)
|
|
|
|
@staticmethod
|
|
def get_all():
|
|
"""
|
|
Return all rooms
|
|
:return:
|
|
"""
|
|
return Room.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():
|
|
# Room.query.delete() # drop all rooms
|
|
rooms = get_campus_rooms()
|
|
logger.debug("got {} rooms".format(len(rooms)))
|
|
db_rooms = [Room(name=room['name'], number=room['room_number'], alternate_name=room.get('alternate_name', None),
|
|
building_name=room['building_name'], building_number=room['building_number'],
|
|
coordinates=room.get('coordinates', None), external_id=room.get('external_id', None)
|
|
|
|
) for room in
|
|
rooms]
|
|
try:
|
|
db.session.bulk_save_objects(db_rooms)
|
|
db.session.commit()
|
|
logger.debug("rooms commited to DB!")
|
|
except IntegrityError as e:
|
|
logger.error("Could not add rooms! ({})".format(e))
|
|
db.session.rollback()
|