From 1a4a5e8c7a29031d9587efa2b63f7dfa5aed8478 Mon Sep 17 00:00:00 2001
From: lasseedfast <lasse.edfast@gmail.com>
Date: Thu, 10 Oct 2024 11:31:57 +0200
Subject: [PATCH] Refactor code to update .gitignore and include necessary
 files for repository

---
 .gitignore              | 7 ++-----
 env_manager/__init__.py | 4 +++-
 setup.py                | 2 +-
 3 files changed, 6 insertions(+), 7 deletions(-)

diff --git a/.gitignore b/.gitignore
index f1d17df..e18c66b 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,9 +1,6 @@
 # .gitignore
 *
-!env_manager.py
+!env_manager/
 !README.md
 !setup.py
-!.gitignore
-!__init__.py
-!/env_manager/*
-!/env_manager
\ No newline at end of file
+!.gitignore
\ No newline at end of file
diff --git a/env_manager/__init__.py b/env_manager/__init__.py
index 024ad02..5b3cd41 100644
--- a/env_manager/__init__.py
+++ b/env_manager/__init__.py
@@ -1,3 +1,5 @@
-from env_manager import get_env
+from .env_manager import get_env
+
+# Initialize environment variables
 get_env()
 print("Environment variables set from ArangoDB by env_manager.")
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 25d1609..e99ff7b 100644
--- a/setup.py
+++ b/setup.py
@@ -10,7 +10,7 @@ setup(
     ],
     entry_points={
         'console_scripts': [
-            'env_manager=env_manager:main',
+            'env_manager=env_manager.env_manager:main',
         ],
     },
 )
\ No newline at end of file