Skip to content

Commit 7f2d806

Browse files
committed
Fix: Restore HostResult.from_dict() for get_host and get_domain
1 parent f6a0e40 commit 7f2d806

1 file changed

Lines changed: 6 additions & 10 deletions

File tree

leakix/client.py

Lines changed: 6 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -148,12 +148,10 @@ def get_host(self, ipv4: str) -> AbstractResponse:
148148
r = self.__get(url, params=None)
149149
if r.is_success():
150150
response_json = r.json()
151-
# Handle both old (uppercase) and new (lowercase) API response formats
152-
services = response_json.get("services") or response_json.get("Services") or []
153-
leaks = response_json.get("leaks") or response_json.get("Leaks") or []
151+
formatted_result = HostResult.from_dict(response_json)
154152
response_json = {
155-
"services": services,
156-
"leaks": leaks,
153+
"services": formatted_result.Services,
154+
"leaks": formatted_result.Leaks,
157155
}
158156
r.response_json = response_json
159157
return r
@@ -255,12 +253,10 @@ def get_domain(self, domain: str) -> AbstractResponse:
255253
r = self.__get(url, params=None)
256254
if r.is_success():
257255
response_json = r.json()
258-
# Handle both old (uppercase) and new (lowercase) API response formats
259-
services = response_json.get("services") or response_json.get("Services") or []
260-
leaks = response_json.get("leaks") or response_json.get("Leaks") or []
256+
formatted_result = HostResult.from_dict(response_json)
261257
response_json = {
262-
"services": services,
263-
"leaks": leaks,
258+
"services": formatted_result.Services,
259+
"leaks": formatted_result.Leaks,
264260
}
265261
r.response_json = response_json
266262
return r

0 commit comments

Comments
 (0)