_config.py

import os
from dotenv import load_dotenv
from flask import Flask
import json

load_dotenv()

VERSION = "0.3.20"

MAX_RESULTS = int(10e6)
MAX_COMPATIBILITY_RESULTS = int(3650)

SQLALCHEMY_DATABASE_URI = os.environ.get("SQLALCHEMY_DATABASE_URI", "sqlite:///test.db")
SQLALCHEMY_ENGINE_OPTIONS = json.loads(os.environ.get("SQLALCHEMY_ENGINE_OPTIONS", "{}"))
SECRET = os.environ.get("FLASK_SECRET", "secret")
URL_PREFIX = os.environ.get("FLASK_PREFIX", "/")

AUTH = {
    "twitter": os.environ.get("SECRET_TWITTER"),
    "ght": os.environ.get("SECRET_GHT"),
    "fluview": os.environ.get("SECRET_FLUVIEW"),
    "cdc": os.environ.get("SECRET_CDC"),
    "sensors": os.environ.get("SECRET_SENSORS"),
    "quidel": os.environ.get("SECRET_QUIDEL"),
    "norostat": os.environ.get("SECRET_NOROSTAT"),
    "afhsb": os.environ.get("SECRET_AFHSB"),
}

# begin sensor query authentication configuration
#   A multimap of sensor names to the "granular" auth tokens that can be used to access them; excludes the "global" sensor auth key that works for all sensors:
GRANULAR_SENSOR_AUTH_TOKENS = {
    "twtr": os.environ.get("SECRET_SENSOR_TWTR", "").split(","),
    "gft": os.environ.get("SECRET_SENSOR_GFT", "").split(","),
    "ght": os.environ.get("SECRET_SENSOR_GHT", "").split(","),
    "ghtj": os.environ.get("SECRET_SENSOR_GHTJ", "").split(","),
    "cdc": os.environ.get("SECRET_SENSOR_CDC", "").split(","),
    "quid": os.environ.get("SECRET_SENSOR_QUID", "").split(","),
    "wiki": os.environ.get("SECRET_SENSOR_WIKI", "").split(","),
}

#   A set of sensors that do not require an auth key to access:
OPEN_SENSORS = [
    "sar3",
    "epic",
    "arch",
]

REGION_TO_STATE = {
    "hhs1": ["VT", "CT", "ME", "MA", "NH", "RI"],
    "hhs2": ["NJ", "NY"],
    "hhs3": ["DE", "DC", "MD", "PA", "VA", "WV"],
    "hhs4": ["AL", "FL", "GA", "KY", "MS", "NC", "TN", "SC"],
    "hhs5": ["IL", "IN", "MI", "MN", "OH", "WI"],
    "hhs6": ["AR", "LA", "NM", "OK", "TX"],
    "hhs7": ["IA", "KS", "MO", "NE"],
    "hhs8": ["CO", "MT", "ND", "SD", "UT", "WY"],
    "hhs9": ["AZ", "CA", "HI", "NV"],
    "hhs10": ["AK", "ID", "OR", "WA"],
    "cen1": ["CT", "ME", "MA", "NH", "RI", "VT"],
    "cen2": ["NJ", "NY", "PA"],
    "cen3": ["IL", "IN", "MI", "OH", "WI"],
    "cen4": ["IA", "KS", "MN", "MO", "NE", "ND", "SD"],
    "cen5": ["DE", "DC", "FL", "GA", "MD", "NC", "SC", "VA", "WV"],
    "cen6": ["AL", "KY", "MS", "TN"],
    "cen7": ["AR", "LA", "OK", "TX"],
    "cen8": ["AZ", "CO", "ID", "MT", "NV", "NM", "UT", "WY"],
    "cen9": ["AK", "CA", "HI", "OR", "WA"],
}
NATION_REGION = "nat"