chore: cleanup

src/aind_qc_portal/database.py → src/aind_qc_portal/docdb/database.py

@@ -2,14 +2,11 @@
 from aind_data_access_api.document_db import MetadataDbClient
 import numpy as np
 import panel as pn
+import os
 
-# API_GATEWAY_HOST = "api.allenneuraldynamics.org"
-# DATABASE = "metadata_index"
-# COLLECTION = "data_assets"
-
-API_GATEWAY_HOST = "api.allenneuraldynamics-test.org"
-DATABASE = "test"
-COLLECTION = "data_assets"
+API_GATEWAY_HOST = os.getenv("API_GATEWAY_HOST", "api.allenneuraldynamics-test.org")
+DATABASE = os.getenv("DATABASE", "test")
+COLLECTION = os.getenv("COLLECTION", "data_assets")
 
 TIMEOUT_1M = 60
 TIMEOUT_1H = 60 * 60

src/aind_qc_portal/evaluation.py → src/aind_qc_portal/panel/evaluation.py

@@ -1,5 +1,5 @@
 from aind_data_schema.core.quality_control import QCEvaluation
-from aind_qc_portal.metric import QCMetricPanel
+from aind_qc_portal.panel.metric import QCMetricPanel
 from aind_qc_portal.utils import md_style, status_html
 import panel as pn
 

src/aind_qc_portal/quality_control.py → src/aind_qc_portal/panel/quality_control.py

@@ -2,9 +2,9 @@
 import panel as pn
 import param
 import json
-from aind_qc_portal.metric import QCMetricPanel
-from aind_qc_portal.evaluation import QCEvalPanel
-from aind_qc_portal.database import qc_from_id, qc_update_to_id
+from aind_qc_portal.panel.metric import QCMetricPanel
+from aind_qc_portal.panel.evaluation import QCEvalPanel
+from aind_qc_portal.docdb.database import qc_from_id, qc_update_to_id
 from aind_qc_portal.utils import md_style, status_html
 from aind_data_schema.core.quality_control import QualityControl, QCEvaluation
 

src/aind_qc_portal/qc_app.py

@@ -6,7 +6,7 @@
 
 alt.data_transformers.disable_max_rows()
 # from aind_qc_portal.search import search_bar
-from aind_qc_portal.quality_control import QCPanel
+from aind_qc_portal.panel.quality_control import QCPanel
 
 alt.data_transformers.disable_max_rows()
 pn.extension("vega", "ace", "jsoneditor")

src/aind_qc_portal/qc_asset_app.py

@@ -5,7 +5,7 @@
 import altair as alt
 import json
 
-from aind_qc_portal.database import get_subj_from_id, get_assets_by_subj, _raw_name_from_derived
+from aind_qc_portal.docdb.database import get_subj_from_id, get_assets_by_subj, _raw_name_from_derived
 from aind_qc_portal.utils import QC_LINK_PREFIX, qc_color, df_timestamp_range
 
 from aind_data_schema.core.quality_control import QualityControl

src/aind_qc_portal/qc_portal_app.py

@@ -6,7 +6,7 @@
 import param
 from datetime import datetime
 
-from aind_qc_portal.database import get_meta
+from aind_qc_portal.docdb.database import get_meta
 from aind_qc_portal.utils import ASSET_LINK_PREFIX, QC_LINK_PREFIX, qc_color
 
 

src/aind_qc_portal/utils.py

@@ -7,16 +7,16 @@
 
 
 def status_html(status: Status):
-    if status.status.value == "Pass":
+    if status.value == "Pass":
         color = "green"
-    elif status.status.value == "Pending":
+    elif status.value == "Pending":
         color = "blue"
-    elif status.status.value == "Fail":
+    elif status.value == "Fail":
         color = "red"
     else:
         color = "#F5BB00"
 
-    return f'<span style="color:{color};">{status.status.value}</span>'
+    return f'<span style="color:{color};">{status.value}</span>'
 
 
 def df_timestamp_range(df, column="timestamp"):