changed a lot regarding rec state, etc.
This commit is contained in:
11
backend/models/model_base.py
Normal file
11
backend/models/model_base.py
Normal file
@@ -0,0 +1,11 @@
|
||||
|
||||
class ModelBase:
|
||||
def get(self, attribute_name, default_value=None):
|
||||
if hasattr(self, attribute_name):
|
||||
return getattr(self, attribute_name)
|
||||
elif False: # a check for properties?
|
||||
pass
|
||||
elif default_value is not None:
|
||||
return default_value
|
||||
else:
|
||||
raise KeyError("{} not found".format(attribute_name))
|
||||
@@ -18,12 +18,13 @@ from backend import db, app, login_manager, LrcException
|
||||
from sqlalchemy import or_
|
||||
from datetime import datetime, timedelta
|
||||
|
||||
from backend.models.model_base import ModelBase
|
||||
from backend.models.virtual_command_model import virtual_command_recorder_command_table, virtual_command_recorder_table
|
||||
|
||||
metadata = MetaData()
|
||||
|
||||
|
||||
class RecorderModel(db.Model):
|
||||
class RecorderModel(db.Model, ModelBase):
|
||||
__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())
|
||||
@@ -43,6 +44,10 @@ class RecorderModel(db.Model):
|
||||
def get_all():
|
||||
return RecorderModel.query.all()
|
||||
|
||||
@staticmethod
|
||||
def get_by_id(id):
|
||||
return RecorderModel.query.filter(RecorderModel.id == id).first()
|
||||
|
||||
@staticmethod
|
||||
def get_by_name(name):
|
||||
return RecorderModel.query.filter(RecorderModel.model_name == name).first()
|
||||
@@ -79,7 +84,7 @@ class RecorderModel(db.Model):
|
||||
return self.model_name + " (record adapter: {})".format(self.record_adapter_id)
|
||||
|
||||
|
||||
class Recorder(db.Model):
|
||||
class Recorder(db.Model, ModelBase):
|
||||
__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())
|
||||
@@ -209,7 +214,7 @@ class Recorder(db.Model):
|
||||
sort_keys=True, indent=4)
|
||||
|
||||
|
||||
class RecorderCommand(db.Model):
|
||||
class RecorderCommand(db.Model, ModelBase):
|
||||
__table_args__ = {'extend_existing': True}
|
||||
"""Table containing permissions associated with groups."""
|
||||
id = db.Column(db.Integer, autoincrement=True, primary_key=True)
|
||||
|
||||
Reference in New Issue
Block a user