IDEV-2412: Fix issue on importing openapi specs

jbabac · jbabac · commit 2ea7e46eb162 · 2026-01-16T14:44:01.000+08:00
diff --git a/domaintools/api.py b/domaintools/api.py
@@ -1,12 +1,14 @@
 from datetime import datetime, timedelta, timezone
 from hashlib import sha1, sha256
 from hmac import new as hmac
+from pathlib import Path
 from typing import Union
 
 import re
 import ssl
 import yaml
 
+
 from domaintools.constants import (
     Endpoint,
     OutputFormat,
@@ -111,17 +113,19 @@ def __init__(
             raise Exception("Proxy URL must be a string. For example: '127.0.0.1:8888'")
 
     def _initialize_specs(self):
+        package_root = Path(__file__).parent
         for spec_name, file_path in SPECS_MAPPING.items():
+            specs_file_path = f"{package_root}/specs/{file_path}"
             try:
-                with open(file_path, "r", encoding="utf-8") as f:
+                with open(specs_file_path, "r", encoding="utf-8") as f:
                     spec_content = yaml.safe_load(f)
                 if not spec_content:
                     raise ValueError("Spec file is empty or invalid.")
 
                 self.specs[spec_name] = spec_content
 
             except Exception as e:
-                print(f"Error loading {file_path}: {e}")
+                print(f"Error loading {specs_file_path}: {e}")
 
     def _get_ssl_default_context(self, verify_ssl: Union[str, bool]):
         return (
diff --git a/domaintools/constants.py b/domaintools/constants.py
@@ -58,6 +58,6 @@ class OutputFormat(Enum):
 }
 
 SPECS_MAPPING = {
-    "iris": "domaintools/specs/iris-openapi.yaml",
-    # "rttf": "domaintools/specs/feeds-openapi.yaml",
+    "iris": "iris-openapi.yaml",
+    # "rttf": "feeds-openapi.yaml",
 }

Original file line number	Diff line number	Diff line change
`@@ -58,6 +58,6 @@ class OutputFormat(Enum):`
`58`	`58`	`}`
`59`	`59`
`60`	`60`	`SPECS_MAPPING = {`
`61`		`- "iris": "domaintools/specs/iris-openapi.yaml",`
`62`		`- # "rttf": "domaintools/specs/feeds-openapi.yaml",`
	`61`	`+ "iris": "iris-openapi.yaml",`
	`62`	`+ # "rttf": "feeds-openapi.yaml",`
`63`	`63`	`}`