adjust imports

Author: BradleySappington

src/wfi_reference_pipeline/database/db_entry.py

@@ -1,6 +1,8 @@
 from datetime import datetime
 
 import romancal
+from rtb_db.constants.rfp_reef import DB_QC_INCOMPLETE
+from rtb_db.table_defs.wfi_rfp.log import RFPLogProTable
 
 from wfi_reference_pipeline import __version__ as rfp_version
 
@@ -31,8 +33,6 @@ def init_rfp_log_pro(self, ref_type, wfi_mode, reef_monitor):
         reef_monitor : bool
             Expecting external monitoring for this run.
         """
-        from rtb_db.constants.rfp_reef import DB_QC_INCOMPLETE
-        from rtb_db.table_defs.wfi_rfp.log import RFPLogProTable
         start_time = self.get_date_time_formatted()
         self.rfp_log_pro = RFPLogProTable(ref_type=ref_type,
                                           start_time=start_time,

src/wfi_reference_pipeline/utilities/db_handler.py

@@ -1,5 +1,11 @@
 import logging
 
+from rtb_db.utilities.login import connect_server
+from rtb_db.utilities.table_tools import (
+    add_to_tables_from_class_list,
+    ensure_connection_is_engine,
+)
+
 from wfi_reference_pipeline import constants
 from wfi_reference_pipeline.database.db_entry import DBEntry
 
@@ -80,8 +86,7 @@ def _connect(self, use_dsn, sql_server_str, sql_database_str, port, dsn_header_s
         """
 
         try:
-            from rtb_db.utilities.login import connect_server
-            from rtb_db.utilities.table_tools import ensure_connection_is_engine
+
             if use_dsn:
                 engine = connect_server(dsn_name=dsn_header_str)
             else:
@@ -115,6 +120,5 @@ def new_pipeline_db_entry(self, ref_type, wfi_mode, reef_monitor):
         reef_monitor : bool
             Expecting external monitoring for this run.
         """
-        from rtb_db.utilities.table_tools import add_to_tables_from_class_list
         self.db_entry.init_rfp_log_pro(ref_type, wfi_mode, reef_monitor)
         add_to_tables_from_class_list(self.db_engine, [self.db_entry.rfp_log_pro])

src/wfi_reference_pipeline/utilities/rtbdb_functions.py

@@ -1,5 +1,7 @@
 import logging
 
+from rtb_db.utilities import login, rfp_tools
+
 
 def get_ma_table_from_rtbdb(ma_table_number=None):
     """
@@ -15,8 +17,6 @@ def get_ma_table_from_rtbdb(ma_table_number=None):
     ma_table_dict: dictionary
         A python dictionary of MA Table information. See RTB Database examples for keys and values.
     """
-    from rtb_db.utilities import login, rfp_tools
-
     eng = login.connect_server(dsn_name='DWRINSDB')
     ma_table_dict = rfp_tools.query_ma_table(eng, ma_table_number)
     if ma_table_dict:
@@ -67,6 +67,5 @@ def write_metrics_db_dark(dark_file_dict, dark_dq_dict, dark_struc_dict, dark_am
     dark_amp_dict: dict, default=None
         This is the amplifier dictionary containing various metrics of sections of pixels according to amplifier.
     """
-    from rtb_db.utilities import login, rfp_tools
     eng = login.connect_server(dsn_name='DWRINSDB')
     rfp_tools.add_new_dark_file(eng, dark_file_dict, dark_dq_dict,dark_struc_dict, dark_amp_dict)