From afaef06f4084bb753a9d71b8cf274f648d3d145c Mon Sep 17 00:00:00 2001
From: Azalea Gui <me@hydev.org>
Date: Sun, 29 Jan 2023 23:54:53 -0500
Subject: [PATCH] [+] Add setup_proxy

---
 hypy_utils/request_utils.py | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)
 create mode 100644 hypy_utils/request_utils.py

diff --git a/hypy_utils/request_utils.py b/hypy_utils/request_utils.py
new file mode 100644
index 0000000..c2c2ecf
--- /dev/null
+++ b/hypy_utils/request_utils.py
@@ -0,0 +1,27 @@
+import requests
+
+
+def setup_proxy(session: requests.Session, addr: str = 'socks5://localhost:9050', verbose: bool = True):
+    url = 'https://ifconfig.me/ip'
+
+    # Setup proxy
+    ip = session.get(url).text.strip()
+    session.proxies = {
+        'http': addr,
+        'https': addr
+    }
+    proxy_ip = session.get(url).text.strip()
+
+    # Print ip
+    if verbose:
+        print(f'Raw ip: {ip}')
+        print(f'Proxy ip: {proxy_ip}')
+
+    # ips shouldn't match
+    assert ip != proxy_ip, 'Proxy did not start correctly.'
+
+    # Disable default requests behavior
+    def warn(*args, **kwargs):
+        raise ReferenceError('Use session.get instead of requests.get')
+    requests.get = warn
+    requests.post = warn