# -*- coding: utf-8 -*- """ Models for lecture recorder """ import importlib import json import pkgutil import os from sqlalchemy import MetaData from sqlalchemy.exc import IntegrityError from sqlalchemy.ext.hybrid import hybrid_property from sqlalchemy.orm import validates from backend import db, app, login_manager import re from sqlalchemy import or_ from datetime import datetime, timedelta from backend.models.virtual_command_model import virtual_command_recorder_command_table, virtual_command_recorder_table from backend.recorder_adapters import get_defined_recorder_adapters from backend.tools.helpers import file_md5 metadata = MetaData() KNOWN_RECORDERS = {re.compile(r'(SMP)[\s]*([\d]+)[\s]*.?[\s]*([\S]*)'): 'SMP', re.compile( r'(LectureRecorder X2|LectureRecorder|VGADVI Recorder|DVI Broadcaster DL|DVIRecorderDL)'): 'Epiphan'} class RecorderModel(db.Model): __table_args__ = {'extend_existing': True} id = db.Column(db.Integer, autoincrement=True, primary_key=True) created_at = db.Column(db.DateTime, nullable=False, default=datetime.utcnow()) last_time_modified = db.Column(db.DateTime, nullable=True, default=None) record_adapter_id = db.Column(db.Unicode(63), unique=True, nullable=False) model_name = db.Column(db.Unicode(63), unique=True, nullable=False) notes = db.Column(db.Unicode(255), unique=False, nullable=True, default=None) recorder_commands = db.relationship('RecorderCommand', back_populates='recorder_model') recorders = db.relationship('Recorder', back_populates='recorder_model') checksum = db.Column(db.String(63), unique=True, nullable=False) # checksum of the recorder commands! (see: model_updater.py) _requires_user = db.Column(db.Integer, default=False, name='requires_user') _requires_password = db.Column(db.Integer, default=True, name='requires_password') @staticmethod def get_all(): return RecorderModel.query.all() @staticmethod def get_by_name(name): return RecorderModel.query.filter(RecorderModel.model_name == name).first() @staticmethod def get_by_adapter_id(name): return RecorderModel.query.filter(RecorderModel.record_adapter_id == name).first() @staticmethod def get_by_checksum(md5_sum): return RecorderModel.query.filter(RecorderModel.checksum == md5_sum).first() @hybrid_property def requires_user(self): return self._requires_user > 0 @requires_user.setter def requires_user(self, val: bool): self._requires_user = 1 if val else 0 @hybrid_property def requires_password(self): return self._requires_password > 0 @requires_password.setter def requires_password(self, val: bool): self._requires_password = 1 if val else 0 class Recorder(db.Model): __table_args__ = {'extend_existing': True} id = db.Column(db.Integer, autoincrement=True, primary_key=True) created_at = db.Column(db.DateTime, nullable=False, default=datetime.utcnow()) last_time_modified = db.Column(db.DateTime, nullable=False, default=datetime.utcnow()) name = db.Column(db.Unicode(63), unique=True, nullable=False) locked = db.Column(db.Boolean, default=False) lock_message = db.Column(db.String, nullable=True, default=None) offline = db.Column(db.Boolean, default=False) description = db.Column(db.Unicode(255), unique=False, nullable=True, default="") room_id = db.Column(db.Integer, db.ForeignKey('room.id')) room = db.relationship('Room', uselist=False, back_populates='recorder') # one-to-one relation (uselist=False) ip = db.Column(db.String(15), unique=True, nullable=True, default=None) configured_options_json_string = db.Column(db.UnicodeText, default='') firmware_version = db.Column(db.String, nullable=True, default=None) ip6 = db.Column(db.String(46), unique=True, nullable=True, default=None) network_name = db.Column(db.String(127), unique=True, nullable=True, default=None) telnet_port = db.Column(db.Integer, unique=False, nullable=False, default=23) ssh_port = db.Column(db.Integer, unique=False, nullable=False, default=22) username = db.Column(db.String, nullable=True, default=None) password = db.Column(db.String, nullable=True, default=None) recorder_model_id = db.Column(db.Integer, db.ForeignKey('recorder_model.id')) recorder_model = db.relationship('RecorderModel', back_populates='recorders') virtual_commands = db.relationship('VirtualCommand', secondary=virtual_command_recorder_table, back_populates='recorders') def __init__(self, **kwargs): super(Recorder, self).__init__(**kwargs) @staticmethod def get_by_name(name): return Recorder.query.filter(Recorder.name == name).first() @staticmethod def get_by_identifier(identifier): return Recorder.query.filter(Recorder.id == identifier).first() @staticmethod def get_all(): return Recorder.query.all() @validates('name') def validate_name(self, key, value): assert len(value) > 2 return value @hybrid_property def configured_options(self) -> list: return json.loads(self.configured_options_json_string) @configured_options.setter def configured_options(self, value: list): self.configured_options_json_string = json.dumps(value) def add_configured_option(self, value: str): self.configured_options_json_string = json.dumps(self.configured_options.append(value)) 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) class RecorderCommand(db.Model): __table_args__ = {'extend_existing': True} """Table containing permissions associated with groups.""" id = db.Column(db.Integer, autoincrement=True, primary_key=True) created_at = db.Column(db.DateTime, nullable=False, default=datetime.utcnow()) last_time_modified = db.Column(db.DateTime, nullable=True, default=datetime.utcnow()) name = db.Column(db.Unicode(63), unique=True, nullable=False) alternative_name = db.Column(db.Unicode(63), unique=True, nullable=True, default=None) disabled = db.Column(db.Boolean, default=False) description = db.Column(db.Unicode(511), nullable=True, default=None) parameters_string = db.Column(db.String(2047), nullable=True) recorder_model = db.relationship('RecorderModel', back_populates='recorder_commands') recorder_model_id = db.Column(db.Integer, db.ForeignKey('recorder_model.id')) virtual_commands = db.relationship('VirtualCommand', secondary=virtual_command_recorder_command_table, back_populates='recorder_commands') @staticmethod def get_all(): return RecorderCommand.query.all() @property def parameters(self): if self.parameters_string is None: return None return json.loads(self.parameters_string) @parameters.setter def parameters(self, parameters_dict: dict): self.parameters_string = json.dumps(parameters_dict) def pre_create_recorders(): models = set() f = os.path.abspath(os.path.join(os.path.dirname(__file__), os.path.pardir, 'models', 'initial_recorders.json')) with open(f, 'r') as json_file: recorders = json.load(json_file)['recorders'] for r in recorders: type = r.get('type') firmware_version = r.get('firmware_version', None) for k_r in KNOWN_RECORDERS: if match := k_r.search(type): print(KNOWN_RECORDERS[k_r]) print(match) print(match.groups()) models.add(r.get('type', '')) models.discard('') print(models) def pre_fill_table(): try: pre_create_recorders() print(db.session.commit()) except IntegrityError as e: db.session.rollback() if __name__ == '__main__': pass # pre_create_recorder_models()