Merge pull request #205 from rstudio/load-dotenv-app

ENH: load dotenv file into generated app.py file
rstudio · Jan 3, 2024 · efa0c52 · efa0c52
2 parents ffadf81 + cc57572
commit efa0c52
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -32,10 +32,12 @@ install_requires =
     requests
     pins>=0.7.1
     rsconnect-python>=1.11.0
+    python-dotenv
     plotly
     pip-tools
     httpx
 
+
 [options.extras_require]
 all =
     vetiver[dev]

diff --git a/vetiver/tests/test_write_app.py b/vetiver/tests/test_write_app.py
@@ -26,9 +26,11 @@ def test_write_app(vetiver_model_creation):
         assert (
             contents
             == f"""from vetiver import VetiverModel
+from dotenv import load_dotenv, find_dotenv
 import vetiver
 import pins
 
+load_dotenv(find_dotenv())
 
 b = pins.board_folder({repr(tempdir)}, allow_pickle_read=True)
 v = VetiverModel.from_pin(b, 'model', version = {repr(version)})

diff --git a/vetiver/write_fastapi.py b/vetiver/write_fastapi.py
@@ -104,7 +104,9 @@ def write_app(
         f = open(file, "x")
 
     app = f"""from vetiver import VetiverModel
+from dotenv import load_dotenv, find_dotenv
 {_glue_required_pkgs(infra_pkgs)}
+load_dotenv(find_dotenv())
 
 b = pins.{load_board}
 {pin_read}