From 6caa8ff135afe53bc4445254344241ca97efec9c Mon Sep 17 00:00:00 2001
From: Darren Chaddock <dchaddoc@ucalgary.ca>
Date: Fri, 24 Jan 2025 14:14:15 +0000
Subject: [PATCH] class import updates

---
 pyaurorax/search/__init__.py | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/pyaurorax/search/__init__.py b/pyaurorax/search/__init__.py
index 6c826b22..82d94299 100644
--- a/pyaurorax/search/__init__.py
+++ b/pyaurorax/search/__init__.py
@@ -111,6 +111,36 @@ def __init__(self, aurorax_obj):
         self.__data_products = DataProductsManager(self.__aurorax_obj)
         self.__conjunctions = ConjunctionsManager(self.__aurorax_obj)
 
+        # initialize class vars
+        self.DataSource = DataSource
+        self.Location = Location
+        self.AvailabilityResult = AvailabilityResult
+        self.EphemerisData = EphemerisData
+        self.EphemerisSearch = EphemerisSearch
+        self.DataProductData = DataProductData
+        self.DataProductSearch = DataProductSearch
+        self.Conjunction = Conjunction
+        self.ConjunctionSearch = ConjunctionSearch
+
+        # initialize static vars
+        self.FORMAT_BASIC_INFO = FORMAT_BASIC_INFO
+        self.FORMAT_BASIC_INFO_WITH_METADATA = FORMAT_BASIC_INFO_WITH_METADATA
+        self.FORMAT_FULL_RECORD = FORMAT_FULL_RECORD
+        self.FORMAT_IDENTIFIER_ONLY = FORMAT_IDENTIFIER_ONLY
+        self.FORMAT_DEFAULT = FORMAT_DEFAULT
+        self.SOURCE_TYPE_EVENT_LIST = SOURCE_TYPE_EVENT_LIST
+        self.SOURCE_TYPE_GROUND = SOURCE_TYPE_GROUND
+        self.SOURCE_TYPE_HEO = SOURCE_TYPE_HEO
+        self.SOURCE_TYPE_LEO = SOURCE_TYPE_LEO
+        self.SOURCE_TYPE_LUNAR = SOURCE_TYPE_LUNAR
+        self.DATA_PRODUCT_TYPE_KEOGRAM = DATA_PRODUCT_TYPE_KEOGRAM
+        self.DATA_PRODUCT_TYPE_MONTAGE = DATA_PRODUCT_TYPE_MONTAGE
+        self.DATA_PRODUCT_TYPE_MOVIE = DATA_PRODUCT_TYPE_MOVIE
+        self.DATA_PRODUCT_TYPE_SUMMARY_PLOT = DATA_PRODUCT_TYPE_SUMMARY_PLOT
+        self.DATA_PRODUCT_TYPE_DATA_AVAILABILITY = DATA_PRODUCT_TYPE_DATA_AVAILABILITY
+        self.CONJUNCTION_TYPE_NBTRACE = CONJUNCTION_TYPE_NBTRACE
+        self.CONJUNCTION_TYPE_SBTRACE = CONJUNCTION_TYPE_SBTRACE
+
     # ------------------------------------------
     # properties for submodule managers
     # ------------------------------------------