From 035058b210d39a920242efc2be8de9d400454036 Mon Sep 17 00:00:00 2001
From: Lukas <m@hler.eu>
Date: Sat, 31 Oct 2020 17:55:35 +0100
Subject: [PATCH] Added Proxy functionality + proxy test function

---
 main.py      | 26 +++++++++++++++++++++++---
 src/proxy.py | 32 +++++++++++++++++++++++++++++++-
 2 files changed, 54 insertions(+), 4 deletions(-)

diff --git a/main.py b/main.py
index 078b3c6..fb1bf25 100644
--- a/main.py
+++ b/main.py
@@ -5,9 +5,19 @@ import src.proxy as proxy
 import os
 import glob
 import json
+from dotenv import load_dotenv
 from colorama import Style, Fore, init
 
 
+def initialize():
+
+    # Init Environment
+    load_dotenv()
+
+    # Init Colorama
+    init(autoreset=True)
+
+
 def readsites(p=None):
 
     dod = {}
@@ -28,14 +38,24 @@ def readsites(p=None):
     print(dod)
 
 
+def proxytest():
+    one = proxy.Proxy
+    one.name = "TEST PROXY"
+    one.usr = os.getenv("PROXYUSR")
+    one.pwd = os.getenv("PROXYPW")
+    one.url = os.getenv("PROXYURL")
+    one.port = os.getenv("PROXYPORT")
+
+    proxy.test(one)
+
+
 if __name__ == "__main__":
 
     num_monitors = 10
 
-    # Initialize Colorama
-    init(autoreset=True)
-
+    initialize()
     readsites()
+    proxytest()
 
     tools.clear()
     print(f"Creating {num_monitors} new Monitors...\n{tools.spacer}")
diff --git a/src/proxy.py b/src/proxy.py
index 3223274..44c44d9 100644
--- a/src/proxy.py
+++ b/src/proxy.py
@@ -1,9 +1,39 @@
+import requests
+
+
+def test(proxy):
+
+    print(f'http://{proxy.usr}:{proxy.pwd}@{proxy.url}:{proxy.port}')
+
+    try:
+        r = requests.get(
+            "http://example.com",
+            proxies={
+                'http': f'http://{proxy.usr}:{proxy.pwd}@{proxy.url}:{proxy.port}',
+                'https': f'http://{proxy.usr}:{proxy.pwd}@{proxy.url}:{proxy.port}'
+            }
+        )
+
+        if r.status_code == 200:
+            proxy.isworking = True
+        else:
+            print(r.text)
+            proxy.isworking = False
+
+        print(f"{r.status_code} -> {proxy.isworking}")
+
+    except Exception as e:
+        print(f"Connection error! (Check proxy)\n\n{e}")
+        proxy.isworking = False
+
+
 class Proxy:
 
     def __init__(self):
 
         self.name = None
         self.url = None
+        self.port = None
         self.usr = None
         self.pwd = None
-        self.status = None
+        self.isworking = None