use of pkg ressources to get file paths

debiai · Jul 12, 2024 · 8da0c2d · 8da0c2d
1 parent 3beb5e1
commit 8da0c2d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/debiaiServer/config/init_config.py b/debiaiServer/config/init_config.py
@@ -1,9 +1,9 @@
 from configparser import ConfigParser
 from termcolor import colored
-
+import pkg_resources
 import os
 
-config_path = "debiaiServer/config/config.ini"
+config_path = pkg_resources.resource_filename("debiaiServer", "config/config.ini")
 config_parser = ConfigParser()
 
 DEBUG_COLOR = "light_blue"

diff --git a/debiaiServer/utils/utils.py b/debiaiServer/utils/utils.py
@@ -1,13 +1,16 @@
 import time
 import yaml
+import pkg_resources
 from yaml.loader import SafeLoader
 from urllib.parse import urlparse
 
 
 def get_app_version():
     # Read the version from the API YAML file
+    yaml_path = pkg_resources.resource_filename("debiaiServer", "swagger.yaml")
+
     try:
-        with open("debiaiServer/swagger.yaml") as f:
+        with open(yaml_path, "r") as f:
             data = yaml.load(f, Loader=SafeLoader)
             return data["info"]["version"]
     except Exception as e: