making dataset_id into constants

Author: aysim319

nhsn/delphi_nhsn/constants.py

@@ -2,6 +2,9 @@
 
 GEOS = ["state", "nation", "hhs"]
 
+MAIN_DATASET_ID = "ua7e-t2fy"
+PRELIM_DATASET_ID = "mpgq-jmmr"
+
 # column name from socrata
 TOTAL_ADMISSION_COVID_API = "totalconfc19newadm"
 TOTAL_ADMISSION_FLU_API = "totalconfflunewadm"

nhsn/delphi_nhsn/pull.py

@@ -8,7 +8,8 @@
 from delphi_utils import create_backup_csv
 from sodapy import Socrata
 
-from .constants import PRELIM_SIGNALS_MAP, PRELIM_TYPE_DICT, SIGNALS_MAP, TYPE_DICT
+from .constants import PRELIM_SIGNALS_MAP, PRELIM_TYPE_DICT, SIGNALS_MAP, TYPE_DICT, \
+    MAIN_DATASET_ID, PRELIM_DATASET_ID
 
 
 def pull_data(socrata_token: str, dataset_id: str):
@@ -89,7 +90,7 @@ def pull_nhsn_data(
     """
     # Pull data from Socrata API
     df = (
-        pull_data(socrata_token, dataset_id="ua7e-t2fy")
+        pull_data(socrata_token, dataset_id=MAIN_DATASET_ID)
         if not custom_run
         else pull_data_from_file(backup_dir, issue_date, logger, prelim_flag=False)
     )
@@ -145,7 +146,7 @@ def pull_preliminary_nhsn_data(
         Dataframe as described above.
     """
     df = (
-        pull_data(socrata_token, dataset_id="mpgq-jmmr")
+        pull_data(socrata_token, dataset_id=PRELIM_DATASET_ID)
         if not custom_run
         else pull_data_from_file(backup_dir, issue_date, logger, prelim_flag=True)
     )