From 2ed9f3345dcaafa5dd84631e6c322d916e956846 Mon Sep 17 00:00:00 2001
From: root <root@docs.tailcf205.ts.net>
Date: Fri, 28 Jun 2024 20:10:47 -0500
Subject: [PATCH] added some dotenv stuff

---
 pyproject.toml          | 2 +-
 requirements.txt        | 3 ++-
 src/restic/backup.py    | 7 +++++++
 src/restic/snapshots.py | 3 +++
 4 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 2d23d73..4f9071a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -12,4 +12,4 @@ authors = [
 license = { file = "LICENSE" }
 requires-python = ">=3.10"
 
-dependencies = ["rich", "requests", "click", "docker"]
+dependencies = ["rich", "requests", "click", "docker", "python-dotenv"]
diff --git a/requirements.txt b/requirements.txt
index c622efd..e58d932 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,4 +3,5 @@ ruff
 rich
 requests
 click
-docker
\ No newline at end of file
+docker
+python-dotenv
\ No newline at end of file
diff --git a/src/restic/backup.py b/src/restic/backup.py
index 39b74e9..85d2995 100755
--- a/src/restic/backup.py
+++ b/src/restic/backup.py
@@ -81,6 +81,7 @@ def main(
     logging.getLogger('urllib3.connectionpool').setLevel('WARNING')
 
     if project is not None and services is not None:
+        logger.debug(f'Using project {project} and stopping services: {services}')
         decorator = manage_containers(project=project, services=services.split(','))
         func = decorator(run)
     else:
@@ -96,4 +97,10 @@ def main(
 
 
 if __name__ == '__main__':
+    from dotenv import load_dotenv
+    from pathlib import Path
+
+    dotenv_file = Path.cwd() / '.env'
+    print(dotenv_file)
+    load_dotenv(dotenv_path=dotenv_file)
     main()
diff --git a/src/restic/snapshots.py b/src/restic/snapshots.py
index 7121931..00b87e3 100644
--- a/src/restic/snapshots.py
+++ b/src/restic/snapshots.py
@@ -35,4 +35,7 @@ def main(loki_url: str = None):
 
 
 if __name__ == '__main__':
+    from dotenv import load_dotenv
+    
+    load_dotenv()
     main()