Merge pull request #5 from github/env-load

Load the env from file for local use
github · Oct 6, 2023 · 1abd215 · 1abd215
2 parents 1c9df51 + dc0666d
commit 1abd215
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/env.py b/env.py
@@ -2,6 +2,8 @@
 
 import os
 from typing import List
+from os.path import dirname, join
+from dotenv import load_dotenv
 
 
 def get_env_vars() -> tuple[str, str, List[str]]:
@@ -15,6 +17,10 @@ def get_env_vars() -> tuple[str, str, List[str]]:
         str: the start date to get contributor information from
         str: the end date to get contributor information to.
     """
+    # Load from .env file if it exists
+    dotenv_path = join(dirname(__file__), ".env")
+    load_dotenv(dotenv_path)
+
     organization = os.getenv("ORGANIZATION")
     repository = os.getenv("REPOSITORY")
     # Either organization or repository must be set

diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,4 @@
 github3.py==4.0.1
+python-dotenv==1.0.0
 pytest==7.4.2
-pytest-cov==4.1.0
-
+pytest-cov==4.1.0