diff --git a/myeia/api.py b/myeia/api.py index a54348b..de59a59 100644 --- a/myeia/api.py +++ b/myeia/api.py @@ -15,8 +15,7 @@ level=logging.INFO, format="%(asctime)s - %(message)s", datefmt="%d-%b-%y %H:%M:%S" ) -load_dotenv() - +load_dotenv(verbose=True) class API: """ @@ -30,7 +29,13 @@ def __init__( self, token: Optional[str] = None, ): - self.token = token if token else os.getenv("EIA_TOKEN") + if token: + self.token = token + elif os.getenv("EIA_TOKEN"): + self.token = os.getenv("EIA_TOKEN") + else: + raise ValueError('EIA_TOKEN is not set. Please set it in your .env file or environment variables.') + self.base_url = "https://api.eia.gov/v2/" self.header = {"Accept": "*/*"}