96 lines
3.8 KiB
Python
96 lines
3.8 KiB
Python
# -*- 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
|
|
|
|
|
|
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)
|
|
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)
|
|
use_telnet_instead_ssh = db.Column(db.Boolean, nullable=False, default=False)
|
|
recorder_model_id = db.Column(db.Integer, db.ForeignKey('recorder_model.id'))
|
|
recorder_model = db.relationship('RecorderModel', 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)
|
|
command = db.Column(db.String(2047), nullable=False)
|
|
recorder_models = db.relationship('RecorderModel', secondary=recorder_model_recorder_command_table,
|
|
back_populates='recorder_commands')
|