Patch event loop class on Windows so REPL doesn't fail

Author: SaladDais

hippolyzer/apps/proxy.py

@@ -12,6 +12,7 @@
 import outleap
 
 from hippolyzer.lib.base import llsd
+from hippolyzer.lib.base.helpers import patch_loop_factory_for_ptpython
 from hippolyzer.lib.proxy.addons import AddonManager
 from hippolyzer.lib.proxy.addon_utils import BaseAddon
 from hippolyzer.lib.proxy.ca_utils import setup_ca
@@ -206,6 +207,7 @@ def _windows_timeout_killer(pid: int):
 
 
 def main():
+    patch_loop_factory_for_ptpython()
     multiprocessing.set_start_method("spawn")
     start_proxy(SessionManager(ProxySettings()))
 

hippolyzer/apps/proxy_gui.py

@@ -24,7 +24,8 @@
 from hippolyzer.apps.proxy import start_proxy
 from hippolyzer.lib.base import llsd
 from hippolyzer.lib.base.datatypes import UUID
-from hippolyzer.lib.base.helpers import bytes_unescape, bytes_escape, get_resource_filename, create_logged_task
+from hippolyzer.lib.base.helpers import bytes_unescape, bytes_escape, get_resource_filename, create_logged_task, \
+    patch_loop_factory_for_ptpython
 from hippolyzer.lib.base.message.llsd_msg_serializer import LLSDMessageSerializer
 from hippolyzer.lib.base.message.message import Block, Message
 from hippolyzer.lib.base.message.message_formatting import (
@@ -924,6 +925,7 @@ def _removeCurrentPressed(self):
 
 
 def gui_main():
+    patch_loop_factory_for_ptpython()
     multiprocessing.set_start_method('spawn')
     QtCore.QCoreApplication.setAttribute(QtCore.Qt.AA_ShareOpenGLContexts)
     app = QtWidgets.QApplication(sys.argv)

hippolyzer/lib/base/helpers.py

@@ -5,6 +5,7 @@
 import functools
 import logging
 import os
+import platform
 
 import lazy_object_proxy
 import pkg_resources
@@ -213,3 +214,17 @@ def reorient_coord(coord, new_orientation, min_val: int | float = 0):
     if coord.__class__ in (list, tuple):
         return coord.__class__(coords)
     return coord.__class__(*coords)
+
+
+def patch_loop_factory_for_ptpython() -> None:
+    # This patch can be removed when https://github.com/prompt-toolkit/ptpython/issues/582 is fixed
+    from asyncio import get_event_loop_policy
+    if platform.system() != "Windows":
+        return
+
+    policy = get_event_loop_policy()
+    # Note: this patches the underlying event loop class, already existing subclasses should pick up these
+    # new additions as well. This is important for the Qt event loops.
+    if loop_factory := getattr(policy, "_loop_factory", None):
+        for attr in ("add_signal_handler", "remove_signal_handler"):
+            setattr(loop_factory, attr, lambda *args, **kwargs: None)