:gem: [Feature] OSEnver: Auto set proxy for creator and connector

conversations/conversation_connector.py

@@ -9,8 +9,9 @@ from networks import (
     OpenaiStreamOutputer,
 )
 from utils.logger import logger
+from utils.enver import enver
 
-http_proxy = "http://localhost:11111"  # Replace with yours
+enver.set_envs(proxies=True)
 
 
 class ConversationConnector:
@@ -61,7 +62,7 @@ class ConversationConnector:
         self.wss = await self.aiohttp_session.ws_connect(
             self.ws_url,
             headers=headers_constructor.request_headers,
-            proxy=http_proxy,
+            proxy=enver.envs["http_proxy"],
         )
         await self.init_handshake()
 

conversations/conversation_creator.py

@@ -1,8 +1,9 @@
 import httpx
 import json
 from pprint import pprint
+from utils.enver import enver
 
-http_proxy = "http://localhost:11111"  # Replace with yours
+enver.set_envs(proxies=True)
 
 
 class ConversationCreator:
@@ -23,13 +24,13 @@ class ConversationCreator:
             "X-Forwarded-For": "65.49.22.66",
         }
 
-    def create(self, proxy: str = None):
+    def create(self):
         self.construct_cookies()
         self.construct_headers()
         self.response = httpx.get(
             self.conversation_create_url,
             headers=self.request_headers,
-            proxies=http_proxy if proxy is None else proxy,
+            proxies=enver.envs["http_proxy"],
             cookies=self.httpx_cookies,
         )
         self.response_content = json.loads(self.response.content.decode("utf-8"))

utils/enver.py

@@ -0,0 +1,54 @@
+import json
+import os
+
+from pathlib import Path
+
+
+class OSEnver:
+    def __init__(self, global_scope=True):
+        self.envs_stack = []
+        self.global_scope = global_scope
+        self.envs = os.environ.copy()
+
+    def store_envs(self):
+        self.envs_stack.append(self.envs)
+
+    def restore_envs(self):
+        self.envs = self.envs_stack.pop()
+        if self.global_scope:
+            os.environ = self.envs
+
+    def set_envs(self, secrets=True, proxies=None, store_envs=True):
+        # caller_info = inspect.stack()[1]
+        # logger.back(f"OS Envs is set by: {caller_info.filename}")
+
+        if store_envs:
+            self.store_envs()
+
+        if secrets:
+            secrets_path = Path(__file__).parents[1] / "secrets.json"
+            if secrets_path.exists():
+                with open(secrets_path, "r") as rf:
+                    secrets = json.load(rf)
+            else:
+                secrets = {}
+
+        if proxies:
+            for proxy_env in ["http_proxy", "https_proxy"]:
+                if isinstance(proxies, str):
+                    self.envs[proxy_env] = proxies
+                elif "http_proxy" in secrets.keys():
+                    self.envs[proxy_env] = secrets["http_proxy"]
+                elif os.getenv("http_proxy"):
+                    self.envs[proxy_env] = os.getenv("http_proxy")
+                else:
+                    continue
+                print(f"Set {proxy_env} to {self.envs[proxy_env]}")
+            else:
+                pass
+
+        if self.global_scope:
+            os.environ = self.envs
+
+
+enver = OSEnver()