first commit

Author: scripting-drafts

.gitignore

@@ -0,0 +1,3 @@
+geckodriver.*
+log_toast.log
+profile_path

Turquoise_Logger.py

@@ -0,0 +1,38 @@
+from datetime import datetime
+import logging
+import os
+
+logging.getLogger("scapy.runtime").setLevel(logging.ERROR)
+os.system('')
+
+class Logger:
+    def __init__(self):
+        self.script_name = 'Selenium'
+
+    def RGB(self, red=None, green=None, blue=None, bg=False):
+        '''Logger prettifier'''
+        if(bg == False and red != None and green != None and blue != None):
+            return f'\u001b[38;2;{red};{green};{blue}m'
+        elif(bg == True and red != None and green != None and blue != None):
+            return f'\u001b[48;2;{red};{green};{blue}m'
+        elif(red == None and green == None and blue == None):
+            return '\u001b[0m'
+
+    def logging(self):
+        '''Reusable logger'''
+        # defaults = str(datetime.today().strftime("%d-%m-%Y_%H-%M-%S"))
+        g0 = self.RGB()
+        g1 = self.RGB(127, 255, 212)
+        g2 = self.RGB(0, 0, 128)
+        bold = "\033[1m"
+        reset = "\033[0m"
+        logging.basicConfig(filename='log_toast.log', encoding='UTF-8', level=logging.DEBUG, format='%(asctime)s [%(name)s] %(message)s')
+        logger = logging.getLogger(f'{self.script_name}')
+        logger.setLevel(logging.DEBUG)
+        ch = logging.StreamHandler()
+        ch.setLevel(logging.DEBUG)
+        formatter = logging.Formatter('%(asctime)s,%(msecs)03d {}{}[%(name)s]{}{} %(message)s'.format(bold, g1, g0, reset), '%H:%M:%S')
+        ch.setFormatter(formatter)
+        logger.addHandler(ch)
+
+        return logger

sel-module.py

@@ -0,0 +1,60 @@
+from seleniumwire import webdriver
+from selenium.webdriver.firefox.options import Options
+import Turquoise_Logger
+
+
+logg = Turquoise_Logger.Logger()
+log = logg.logging()
+initial_url = "https://www.google.com"
+
+options = Options()
+options.add_argument('--headless')
+options.set_preference('dom.webnotifications.enabled', False)
+options.set_preference('dom.push.enabled', False)
+options.set_preference('dom.webdriver.enabled', False)
+options.set_preference('useAutomationExtension', False)
+options.set_preference('privacy.trackingprotection.enabled', True)
+profile_path = open('profile_path', 'r').read()
+
+driver = webdriver.Firefox(firefox_profile=profile_path, options=options)
+driver.implicitly_wait(10)
+
+def connection_attempts(initial_url=initial_url, attempts_count=2):
+    '''Commits attempts_count connection attempts to the given initial_url'''
+    error = None
+    while attempts_count:
+        try:
+            driver.get(initial_url)
+        except Exception as error:
+            log.error(f'{error}')
+        
+        attempts_count -= 1
+
+        if not error:
+            attempts_num = attempts_count
+            attempts_count = 0            
+            log.debug(f'Realized in {attempts_num} attempts')
+        elif attempts_count == 0:
+            log.debug('Failed to connect')
+        
+
+def requests_vars_get():
+    '''Outputs REQ (Request URL), STAT (Status Code) and CT (Content Type) within responses in requests'''
+    for request in driver.requests:
+        if request.response:
+            log.error(
+                f'REQ: {request.url}'
+            )
+
+            log.debug(
+                f'STAT: {request.response.status_code}'
+            )
+
+            log.debug(
+                'CT:' +  request.response.headers['content-type'] + '\n'
+            )
+
+connection_attempts()
+requests_vars_get()
+
+driver.quit()