[utils] WebSocketsWrapper: Allow omitting `__enter__` invocation (#3187)

Authored by: Lesmiscore

yt_dlp/extractor/fc2.py

@@ -212,7 +212,6 @@ class FC2LiveIE(InfoExtractor):
             'Accept': '*/*',
             'User-Agent': std_headers['User-Agent'],
         })
-        ws.__enter__()
 
         self.write_debug('[debug] Sending HLS server request')
 

yt_dlp/utils.py

@@ -5436,14 +5436,17 @@ class Config:
 class WebSocketsWrapper():
     """Wraps websockets module to use in non-async scopes"""
 
-    def __init__(self, url, headers=None):
+    def __init__(self, url, headers=None, connect=True):
         self.loop = asyncio.events.new_event_loop()
         self.conn = compat_websockets.connect(
             url, extra_headers=headers, ping_interval=None,
             close_timeout=float('inf'), loop=self.loop, ping_timeout=float('inf'))
+        if connect:
+            self.__enter__()
         atexit.register(self.__exit__, None, None, None)
 
     def __enter__(self):
+        if not self.pool:
             self.pool = self.run_with_loop(self.conn.__aenter__(), self.loop)
         return self