oneshot gui improvements

Author: scripting-drafts

sel-module.py

@@ -6,7 +6,7 @@
 
 class SeleniumWireModule:
     def __init__(self):
-        gui_enhancements.run()
+        gui_enhancements.run_sel()
         logg = turquoise_logger.Logger()
         path_firefox_binary = 'tools/geckodriver.exe'
         path_geckodriver_log = path.abspath('resources/geckodriver.log')

tools/mod_initializer.py

@@ -1,5 +1,5 @@
 
-def run():
+def run_sel():
     print(r'''
      _____      _            _                 _    _ _          
     /  ___|    | |          (_)               | |  | (_)         
@@ -9,6 +9,32 @@ def run():
     \____/ \___|_|\___|_| |_|_|\__,_|_| |_| |_|\/  \/|_|_|  \___|
     ''')
 
+    attribution()
+    
+def run_useragent():
+    print(r'''
+     _   _                ___                   _   
+    | | | |              / _ \                 | |  
+    | | | |___  ___ _ __/ /_\ \ __ _  ___ _ __ | |_ 
+    | | | / __|/ _ \ '__|  _  |/ _` |/ _ \ '_ \| __|
+    | |_| \__ \  __/ |  | | | | (_| |  __/ | | | |_ 
+     \___/|___/\___|_|  \_| |_/\__, |\___|_| |_|\__|
+                                __/ |               
+                               |___/                
+     _____                                          
+    /  ___|                                         
+    \ `--.  ___  _   _ _ __ ___ ___ _ __            
+     `--. \/ _ \| | | | '__/ __/ _ \ '__|           
+    /\__/ / (_) | |_| | | | (_|  __/ |              
+    \____/ \___/ \__,_|_|  \___\___|_|              
+                                                    
+                                                    
+                                                    
+    ''')
+
+    attribution()
+
+def attribution():
     print('''\U0001F310 Developed by Gerard Sala @ https://www.github.com/scripting-drafts
           
           ''')

tools/useragents_listgen.py renamed to tools/useragent_sourcer.py

@@ -5,7 +5,9 @@
 from selenium.webdriver.firefox.options import Options
 from selenium.webdriver.firefox.service import Service
 from selenium.webdriver.common.by import By
+import mod_initializer as gui_enhancements
 
+gui_enhancements.run_useragent()
 input_file = '../resources/useragents.txt'
 userAgents = open(input_file, 'w')