Fix Python typing includes for python < 3.12 (closes #55)

Author: Martin Stadler

src/icon/client/client.py

@@ -1,7 +1,12 @@
 import asyncio
 import logging
 from io import StringIO
-from typing import TYPE_CHECKING, Any, TypedDict
+from typing import TYPE_CHECKING, Any
+import sys
+if sys.version_info < (3, 12):
+    from typing_extensions import TypedDict
+else:
+    from typing import TypedDict
 
 import pandas as pd
 import pydase

src/icon/server/api/models/device_dict.py

@@ -1,4 +1,8 @@
-from typing import TypedDict
+import sys
+if sys.version_info < (3, 12):
+    from typing_extensions import TypedDict
+else:
+    from typing import TypedDict
 
 
 class DeviceDict(TypedDict):

src/icon/server/api/models/experiment_dict.py

@@ -1,4 +1,9 @@
-from typing import Any, TypedDict
+from typing import Any
+import sys
+if sys.version_info < (3, 12):
+    from typing_extensions import TypedDict
+else:
+    from typing import TypedDict
 
 from icon.server.api.models.parameter_metadata import ParameterMetadata
 

src/icon/server/api/models/parameter_metadata.py

@@ -1,4 +1,9 @@
-from typing import Any, TypedDict
+from typing import Any
+import sys
+if sys.version_info < (3, 12):
+    from typing_extensions import TypedDict
+else:
+    from typing import TypedDict
 
 
 class ParameterMetadata(TypedDict):

src/icon/server/api/models/scan_parameter.py

@@ -1,6 +1,17 @@
 from __future__ import annotations
 
-from typing import Any, NotRequired, TypedDict
+from typing import Any
+
+import sys
+if sys.version_info < (3, 11):
+    from typing_extensions import NotRequired
+else:
+    from typing import NotRequired
+
+if sys.version_info < (3, 12):
+    from typing_extensions import TypedDict
+else:
+    from typing import TypedDict
 
 
 class ScanParameter(TypedDict):

src/icon/server/data_access/repositories/experiment_data_repository.py

@@ -2,7 +2,12 @@
 import logging
 import os
 from datetime import datetime
-from typing import TYPE_CHECKING, Literal, TypedDict, cast
+from typing import TYPE_CHECKING, Literal, cast
+import sys
+if sys.version_info < (3, 12):
+    from typing_extensions import TypedDict
+else:
+    from typing import TypedDict
 
 import h5py  # type: ignore
 import numpy as np

src/icon/server/data_access/repositories/pycrystal_library_repository.py

@@ -1,7 +1,12 @@
 import asyncio
 import json
 from pathlib import Path
-from typing import Any, TypedDict
+from typing import Any
+import sys
+if sys.version_info < (3, 12):
+    from typing_extensions import TypedDict
+else:
+    from typing import TypedDict
 
 from icon.config.config import get_config
 from icon.server.api.models.experiment_dict import (

src/icon/server/utils/types.py

@@ -1,4 +1,15 @@
-from typing import Literal, NotRequired, TypedDict
+from typing import Literal
+
+import sys
+if sys.version_info < (3, 11):
+    from typing_extensions import NotRequired
+else:
+    from typing import NotRequired
+
+if sys.version_info < (3, 12):
+    from typing_extensions import TypedDict
+else:
+    from typing import TypedDict
 
 from icon.server.data_access.db_context.influxdb_v1 import DatabaseValueType
 

src/icon/server/web_server/socketio_emit_queue.py

@@ -1,7 +1,18 @@
 from __future__ import annotations
 
 import multiprocessing
-from typing import Any, NotRequired, TypedDict
+from typing import Any
+
+import sys
+if sys.version_info < (3, 11):
+    from typing_extensions import NotRequired
+else:
+    from typing import NotRequired
+
+if sys.version_info < (3, 12):
+    from typing_extensions import TypedDict
+else:
+    from typing import TypedDict
 
 
 class EmitEvent(TypedDict):