Refactor of caching & data storage part 3 #118

sigrefdataproviders/sig_ref_data_provider.py

@@ -2,19 +2,25 @@ from datetime import datetime
 
 import pytz
 
+from core.data_store import DATA_STORE
+
 
 class SIGRefDataProvider:
     """Generic SIG reference data provider class. Subclasses of this query the individual URLs or files for data."""
 
-    def __init__(self, name, provider_config):
+    def __init__(self, sig_name, provider_config):
         """Constructor"""
 
-        self.name = name
+        self.sig_name = sig_name
         self.enabled = provider_config["enabled"]
         self.last_update_time = datetime.min.replace(tzinfo=pytz.UTC)
         self.last_spot_time = datetime.min.replace(tzinfo=pytz.UTC)
         self.status = "Not Started" if self.enabled else "Disabled"
 
+        # Create an empty dict to store data if one doesn't already exist
+        if not sig_name in DATA_STORE.sigrefs:
+            DATA_STORE.sigrefs[sig_name] = {}
+
 
     def start(self):
         """Start the provider. This should return immediately after spawning threads to access the remote resources"""
@@ -26,3 +32,9 @@ class SIGRefDataProvider:
         """Stop any threads and prepare for application shutdown"""
 
         raise NotImplementedError("Subclasses must implement this method")
+
+    def _replace_data(self, new_data):
+        """Replace all data for the named sig with the new data. new_data should be a map of reference ID to SIGRef
+        objects."""
+
+        DATA_STORE.sigrefs[self.sig_name] = new_data