From 6e5a126746faaa9720e0097cf342aff0ffc50bbb Mon Sep 17 00:00:00 2001
From: wuttinanhi <wuttinanhi@gmail.com>
Date: Wed, 21 Sep 2022 06:27:42 +0000
Subject: [PATCH] fix env module ignore by .gitignore

---
 env_wrapper/__init__.py | 13 +++++++++++++
 main.py                 |  2 +-
 mock/run.py             |  2 +-
 3 files changed, 15 insertions(+), 2 deletions(-)
 create mode 100644 env_wrapper/__init__.py

diff --git a/env_wrapper/__init__.py b/env_wrapper/__init__.py
new file mode 100644
index 0000000..54b703e
--- /dev/null
+++ b/env_wrapper/__init__.py
@@ -0,0 +1,13 @@
+"""
+    env module
+"""
+
+
+def load_env():
+    import os
+
+    from dotenv import load_dotenv
+    if os.getenv("env") == "production":
+        load_dotenv(dotenv_path=".env")
+    else:
+        load_dotenv(dotenv_path="dev.env")
diff --git a/main.py b/main.py
index ee1ad87..8db10c8 100644
--- a/main.py
+++ b/main.py
@@ -7,7 +7,7 @@
 from auth import auth_blueprint
 from car import car_blueprint
 from database.database import db_session, init_db
-from env import load_env
+from env_wrapper import load_env
 from parking_lot import parking_lot_blueprint
 
 # load env
diff --git a/mock/run.py b/mock/run.py
index b070ab5..ed9c492 100644
--- a/mock/run.py
+++ b/mock/run.py
@@ -3,7 +3,7 @@
 """
 
 if __name__ == "__main__":
-    from env import load_env
+    from env_wrapper import load_env
     load_env()
     from mock.mock import Mock
     Mock.clean_db()