created a main() function

sryu1 · Dec 5, 2022 · 82bb90e · 82bb90e
1 parent 9a78f5a
commit 82bb90e
Showing 1 changed file with 100 additions and 99 deletions.
diff --git a/Bot.py b/Bot.py
@@ -7,38 +7,9 @@
 import chromedriver_autoinstaller
 
 
-def config():
-    if not os.path.isfile('config.json'):
-        def hdls(headless):
-            if headless == "y":
-                return True
-            if headless == "n":
-                return False
-
-        headless = input(
-            "Would you like to run the bot in headless mode? (y/n): ")
-
-        def sdop(mute):
-            if mute == "y":
-                return True
-            if mute == "n":
-                return False
-
-        sound = input(
-            "Would you like to mute the videos while they are playing? (y/n): ")
-        configs = {
-            "Headless": str(hdls(headless)),
-            "Mute": str(sdop(sound))
-        }
-        json_file = json.dumps(configs)
-        with open("config.json", "w") as jsonfile:
-            jsonfile.write(json_file)
-    else:
-        config_options = input(
-            "Would you like to use the previous settings for Headless mode and Sound? (y/n): ")
-        if config_options == "y":
-            pass
-        else:
+def main():
+    def config():
+        if not os.path.isfile('config.json'):
             def hdls(headless):
                 if headless == "y":
                     return True
@@ -63,70 +34,100 @@ def sdop(mute):
             json_file = json.dumps(configs)
             with open("config.json", "w") as jsonfile:
                 jsonfile.write(json_file)
-
-
-config()
-
-chromedriver_autoinstaller.install()
-viewcount = 0
-views = input("how many views would you like: ")
-number_of_drivers = int(input("Enter the number of tabs you want open: "))
-time_to_refresh = int(input("Choose your watch time (seconds): "))
-url = input("Enter Video URL: ")
-drivers = []
-sites = ['https://search.yahoo.com/', 'https://duckduckgo.com/', 'https://www.google.com/',
-         'https://www.bing.com/', 'https://t.co/', 'https://youtube.com']
-
-wsviews = open("Bot Status/views.txt", 'w')
-wsurl = open("Bot Status/url.txt", 'w')
-wsviews.write(views)
-wsviews.close()
-wsurl.write(url)
-wsurl.close()
-
-with open("config.json", "r") as jsonfile:
-    json_options = json.load(jsonfile)
-    jsonfile.close()
-
-
-def wsviewcount():
-    wsvc = open("Bot Status/viewcount.txt", 'w')
-    wsvc.write(str(viewcount))
-    wsvc.close()
-
-
-def play_video(drivers):
-    ActionChains(drivers[i]) \
-        .send_keys("k") \
-        .perform()
-
-
-for i in range(number_of_drivers):
-    options = webdriver.ChromeOptions()
-    if json_options["Headless"] == "True":
-        options.add_argument("--headless")
-    if json_options["Mute"] == "False":
-        pass
-    else:
-        options.add_argument("--mute-audio")
-    options.add_experimental_option("excludeSwitches", ["enable-logging"])
-    drivers.append(webdriver.Chrome(options=options,
-                                    executable_path=r"chromedriver"))
-    drivers[i].get(random.choice(sites))
-    drivers[i].get(url)
-    play_video(drivers)
-
-while True:
-    time.sleep(time_to_refresh)
-    viewcount += number_of_drivers
-    wsviewcount()
-
-    print("view count = " + str(viewcount))
-    if int(views) <= int(viewcount):
-        drivers[i].quit()
-        print("the listed amount has been viewed")
-        os.system("pause")
-        exit()
-    elif int(views) > int(viewcount):
-        for i in range(number_of_drivers):
-            drivers[i].refresh()
+        else:
+            config_options = input(
+                "Would you like to use the previous settings for Headless mode and Sound? (y/n): ")
+            if config_options == "y":
+                pass
+            else:
+                def hdls(headless):
+                    if headless == "y":
+                        return True
+                    if headless == "n":
+                        return False
+
+                headless = input(
+                    "Would you like to run the bot in headless mode? (y/n): ")
+
+                def sdop(mute):
+                    if mute == "y":
+                        return True
+                    if mute == "n":
+                        return False
+
+                sound = input(
+                    "Would you like to mute the videos while they are playing? (y/n): ")
+                configs = {
+                    "Headless": str(hdls(headless)),
+                    "Mute": str(sdop(sound))
+                }
+                json_file = json.dumps(configs)
+                with open("config.json", "w") as jsonfile:
+                    jsonfile.write(json_file)
+
+    config()
+
+    chromedriver_autoinstaller.install()
+    viewcount = 0
+    views = input("how many views would you like: ")
+    number_of_drivers = int(input("Enter the number of tabs you want open: "))
+    time_to_refresh = int(input("Choose your watch time (seconds): "))
+    url = input("Enter Video URL: ")
+    drivers = []
+    sites = ['https://search.yahoo.com/', 'https://duckduckgo.com/', 'https://www.google.com/',
+             'https://www.bing.com/', 'https://t.co/', 'https://youtube.com']
+
+    wsviews = open("Bot Status/views.txt", 'w')
+    wsurl = open("Bot Status/url.txt", 'w')
+    wsviews.write(views)
+    wsviews.close()
+    wsurl.write(url)
+    wsurl.close()
+
+    with open("config.json", "r") as jsonfile:
+        json_options = json.load(jsonfile)
+        jsonfile.close()
+
+    def wsviewcount():
+        wsvc = open("Bot Status/viewcount.txt", 'w')
+        wsvc.write(str(viewcount))
+        wsvc.close()
+
+    def play_video(drivers):
+        ActionChains(drivers[i]) \
+            .send_keys("k") \
+            .perform()
+
+    for i in range(number_of_drivers):
+        options = webdriver.ChromeOptions()
+        if json_options["Headless"] == "True":
+            options.add_argument("--headless")
+        if json_options["Mute"] == "False":
+            pass
+        else:
+            options.add_argument("--mute-audio")
+        options.add_experimental_option("excludeSwitches", ["enable-logging"])
+        drivers.append(webdriver.Chrome(options=options,
+                                        executable_path=r"chromedriver"))
+        drivers[i].get(random.choice(sites))
+        drivers[i].get(url)
+        play_video(drivers)
+
+    while True:
+        time.sleep(time_to_refresh)
+        viewcount += number_of_drivers
+        wsviewcount()
+
+        print("view count = " + str(viewcount))
+        if int(views) <= int(viewcount):
+            drivers[i].quit()
+            print("the listed amount has been viewed")
+            os.system("pause")
+            exit()
+        elif int(views) > int(viewcount):
+            for i in range(number_of_drivers):
+                drivers[i].refresh()
+
+
+if __name__ == "__main__":
+    main()