prevent merging duplicate methods

spyoungtech · Aug 23, 2023 · 47c3b64 · 47c3b64
1 parent 06a38f8
commit 47c3b64
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/ahk/_async/engine.py b/ahk/_async/engine.py
@@ -145,9 +145,13 @@ def __init__(
             is_async = False
             is_async = True  # unasync: remove
             if is_async:
-                extensions = [entry.extension for name, entry in _extension_method_registry.async_methods.items()]
+                extensions = list(
+                    set(entry.extension for name, entry in _extension_method_registry.async_methods.items())
+                )
             else:
-                extensions = [entry.extension for name, entry in _extension_method_registry.sync_methods.items()]
+                extensions = list(
+                    set(entry.extension for name, entry in _extension_method_registry.sync_methods.items())
+                )
             self._extension_registry = _extension_method_registry
             self._extensions = extensions
         else:

diff --git a/ahk/_sync/engine.py b/ahk/_sync/engine.py
@@ -133,16 +133,16 @@ def __init__(
         TransportClass: Optional[Type[Transport]] = None,
         directives: Optional[list[Directive | Type[Directive]]] = None,
         executable_path: str = '',
-        extensions: list[Extension] | None | Literal['auto'] = None
+        extensions: list[Extension] | None | Literal['auto'] = None,
     ):
         self._extension_registry: _ExtensionMethodRegistry
         self._extensions: list[Extension]
         if extensions == 'auto':
             is_async = False
             if is_async:
-                extensions = [entry.extension for name, entry in _extension_method_registry.async_methods.items()]
+                extensions = list(set(entry.extension for name, entry in _extension_method_registry.async_methods.items()))
             else:
-                extensions = [entry.extension for name, entry in _extension_method_registry.sync_methods.items()]
+                extensions = list(set(entry.extension for name, entry in _extension_method_registry.sync_methods.items()))
             self._extension_registry = _extension_method_registry
             self._extensions = extensions
         else:
@@ -151,7 +151,6 @@ def __init__(
             for ext in self._extensions:
                 self._extension_registry.merge(ext._extension_method_registry)
 
-
         if TransportClass is None:
             TransportClass = DaemonProcessTransport
         assert TransportClass is not None