Merge pull request #7 from Flowseal/copilot/fix-attribute-error-dict-add

Fix AttributeError when handling 302 redirects: initialize _ws_blacklist as set()

README.md

@@ -4,6 +4,8 @@
   
 **Ожидаемый результат аналогичен прокидыванию hosts для Web Telegram**: ускорение загрузки и скачивания файлов, загрузки сообщений и части медиа.
 
+<img width="529" height="487" alt="image" src="https://github.com/user-attachments/assets/6a4cf683-0df8-43af-86c1-0e8f08682b62" />
+
 ## Как это работает
 
 ```
@@ -114,4 +116,4 @@ pyinstaller tg_ws_proxy.spec
 
 ## Лицензия
 
-[MIT License](LICENSE)
+[MIT License](LICENSE)

tg_ws_proxy.py

@@ -39,7 +39,7 @@ _dc_opt: Dict[int, Optional[str]] = {}
 # DCs where WS is known to fail (302 redirect)
 # Raw TCP fallback will be used instead
 # Keyed by (dc, is_media)
-_ws_blacklist: Set[Tuple[int, bool]] = {}
+_ws_blacklist: Set[Tuple[int, bool]] = set()
 
 # Rate-limit re-attempts per (dc, is_media)
 _dc_fail_until: Dict[Tuple[int, bool], float] = {}
@@ -790,7 +790,14 @@ async def _run(port: int, dc_opt: Dict[int, Optional[str]],
         async def wait_stop():
             await stop_event.wait()
             server.close()
-            await server.wait_closed()
+            me = asyncio.current_task()
+            for task in list(asyncio.all_tasks()):
+                if task is not me:
+                    task.cancel()
+            try:
+                await server.wait_closed()
+            except asyncio.CancelledError:
+                pass
         asyncio.create_task(wait_stop())
 
     async with server:

tg_ws_tray.py

@@ -51,6 +51,7 @@ _stop_event: Optional[threading.Event] = None
 _async_stop: Optional[object] = None
 _tray_icon: Optional[object] = None
 _config: dict = {}
+_exiting: bool = False
 
 log = logging.getLogger("tg-ws-tray")
 
@@ -198,7 +199,7 @@ def stop_proxy():
         loop, stop_ev = _async_stop
         loop.call_soon_threadsafe(stop_ev.set)
         if _proxy_thread:
-            _proxy_thread.join(timeout=5)
+            _proxy_thread.join(timeout=2)
     _proxy_thread = None
     log.info("Proxy stopped")
 
@@ -401,8 +402,18 @@ def _on_open_logs(icon=None, item=None):
 
 
 def _on_exit(icon=None, item=None):
+    global _exiting
+    if _exiting:
+        os._exit(0)
+        return
+    _exiting = True
     log.info("User requested exit")
-    stop_proxy()
+
+    def _force_exit():
+        time.sleep(3)
+        os._exit(0)
+    threading.Thread(target=_force_exit, daemon=True, name="force-exit").start()
+
     if icon:
         icon.stop()