Improve NIST NVD API key import (#262)

* Better warning message for a missing NIST API key * Fix default configuration file paths
cve-search · Jan 29, 2024 · 5bc6922 · 5bc6922
1 parent a0f43a0
commit 5bc6922
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/CveXplore/core/nvd_nist/nvd_nist_api.py b/CveXplore/core/nvd_nist/nvd_nist_api.py
@@ -55,7 +55,8 @@ def __init__(
             self.api_key_limit = False
         else:
             self.logger.warning(
-                "Could not find a NIST API Key in the '~/.cvexplore/.env' file; "
+                "Could not find a NIST API Key in the environment variable 'NVD_NIST_API_KEY' "
+                "(e.g. from the '~/.cvexplore/.env' file); "
                 "you could request one at: https://nvd.nist.gov/developers/request-an-api-key"
             )
             self.api_key_limit = True

diff --git a/CveXplore/main.py b/CveXplore/main.py
@@ -14,15 +14,15 @@
 
 if not os.path.exists(os.path.join(user_wd, ".env")):
     shutil.copyfile(
-        os.path.join(os.path.dirname(__file__), ".env_example"),
+        os.path.join(os.path.dirname(__file__), "common/.env_example"),
         os.path.join(user_wd, ".env"),
     )
 
 load_dotenv(os.path.join(user_wd, ".env"))
 
 if not os.path.exists(os.path.join(user_wd, ".sources.ini")):
     shutil.copyfile(
-        os.path.join(os.path.dirname(__file__), ".sources.ini"),
+        os.path.join(os.path.dirname(__file__), "common/.sources.ini"),
         os.path.join(user_wd, ".sources.ini"),
     )