Use TypedDict for JSON output typing.

Flameeyes · Aug 16, 2024 · d9e6a92 · d9e6a92
1 parent d4dad25
commit d9e6a92
Showing 1 changed file with 14 additions and 3 deletions.
diff --git a/ipv6_in_real_life/json.py b/ipv6_in_real_life/json.py
@@ -4,12 +4,23 @@
 
 import json
 import time
-from typing import Union
+from typing import TypedDict
 
 from . import data_structures
 
-HostJson = dict[str, Union[bool, str, None]]
-EntityJson = dict[str, Union[str, list[HostJson], HostJson]]
+
+class HostJson(TypedDict):
+ name: str
+ has_ipv4_address: bool | None
+ has_ipv6_address: bool | None
+
+
+class EntityJson(TypedDict):
+ name: str
+ main_host: HostJson
+ additional_hosts: list[HostJson]
+
+
 CategoryJson = list[EntityJson]
 CountryDataJson = dict[str, CategoryJson]
 SourceJson = dict[str, CountryDataJson]