Pre-sample/control connect

src/submissions/backend/db/models/__init__.py

@@ -1,8 +1,50 @@
 '''
 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 __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
 
-from tools import Base
 from .controls import *
 # import order must go: orgs, kit, subs due to circular import issues
 from .organizations import *