# -*- coding: utf-8 -*- """ Models for lecture recorder """ import json from sqlalchemy import MetaData 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 metadata = MetaData() # This is the association table for the many-to-many relationship between # groups and permissions. recorder_model_recorder_command_table = db.Table('recorder_model_recorder_command', db.Column('recorder_model_id', db.Integer, db.ForeignKey('recorder_model.id', onupdate="CASCADE", ondelete="CASCADE"), primary_key=True), db.Column('recorder_command_id', db.Integer, db.ForeignKey('recorder_command.id', onupdate="CASCADE", ondelete="CASCADE"), primary_key=True)) class RecorderModel(db.Model): id = db.Column(db.Integer, autoincrement=True, primary_key=True) model_name = db.Column(db.Unicode(63), unique=True, nullable=False) model_checksum = db.Column(db.String(63), unique=True, nullable=True) notes = db.Column(db.Unicode(255), unique=False, nullable=True, default=None) recorder_commands = db.relationship('RecorderCommand', secondary=recorder_model_recorder_command_table, back_populates='recorder_models') recorders = db.relationship('Recorder', back_populates='recorder_model') class Recorder(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(63), unique=True, nullable=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) 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) 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', back_populates='recorders') def __init__(self, **kwargs): super(Recorder, self).__init__(**kwargs) @staticmethod def get_by_name(name): """ Find group by name :param name: :return: """ return Recorder.query.filter(Recorder.name == name).first() @staticmethod def get_all(): """ Return all groups :return: """ return Recorder.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) class RecorderCommand(db.Model): """Table containing permissions associated with groups.""" id = db.Column(db.Integer, autoincrement=True, primary_key=True) name = db.Column(db.Unicode(63), unique=True, nullable=False) description = db.Column(db.Unicode(511), nullable=True, default=None) parameters = db.Column(db.String(2047), nullable=True) recorder_models = db.relationship('RecorderModel', secondary=recorder_model_recorder_command_table, back_populates='recorder_commands') virtual_commands = db.relationship('VirtualCommand', secondary=virtual_command_recorder_command_table, back_populates='recorder_commands')