65 lines
1.8 KiB
Python
65 lines
1.8 KiB
Python
'''
|
|
Contains all models for sqlalchemy
|
|
'''
|
|
import sys
|
|
from sqlalchemy.orm import DeclarativeMeta, declarative_base
|
|
from sqlalchemy.ext.declarative import declared_attr
|
|
if 'pytest' in sys.modules:
|
|
from pathlib import Path
|
|
sys.path.append(Path(__file__).parents[4].absolute().joinpath("tests").__str__())
|
|
|
|
Base: DeclarativeMeta = declarative_base()
|
|
|
|
class BaseClass(Base):
|
|
"""
|
|
Abstract class to pass ctx values to all SQLAlchemy objects.
|
|
|
|
Args:
|
|
Base (DeclarativeMeta): Declarative base for metadata.
|
|
"""
|
|
__abstract__ = True
|
|
|
|
__table_args__ = {'extend_existing': True}
|
|
|
|
@declared_attr
|
|
def __tablename__(cls):
|
|
return f"_{cls.__name__.lower()}"
|
|
|
|
@declared_attr
|
|
def __database_session__(cls):
|
|
if not 'pytest' in sys.modules:
|
|
from tools import ctx
|
|
else:
|
|
from test_settings import ctx
|
|
return ctx.database_session
|
|
|
|
@declared_attr
|
|
def __directory_path__(cls):
|
|
if not 'pytest' in sys.modules:
|
|
from tools import ctx
|
|
else:
|
|
from test_settings import ctx
|
|
return ctx.directory_path
|
|
|
|
@declared_attr
|
|
def __backup_path__(cls):
|
|
if not 'pytest' in sys.modules:
|
|
from tools import ctx
|
|
else:
|
|
from test_settings import ctx
|
|
return ctx.backup_path
|
|
|
|
def save(self):
|
|
logger.debug(f"Saving {self}")
|
|
try:
|
|
self.__database_session__.add(self)
|
|
self.__database_session__.commit()
|
|
except Exception as e:
|
|
logger.critical(f"Problem saving object: {e}")
|
|
self.__database_session__.rollback()
|
|
|
|
from .controls import *
|
|
# import order must go: orgs, kit, subs due to circular import issues
|
|
from .organizations import *
|
|
from .kits import *
|
|
from .submissions import * |