KickViewBot/KickViewBotHEADLESS.py at main · Infinity242/KickViewBot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
from undetected_chromedriver import ChromeOptions
from selenium.webdriver.common.by import By
from selenium.common.exceptions import NoSuchElementException
from threading import Thread
import undetected_chromedriver as uc
from time import sleep
from multiprocessing import Process
from selenium.webdriver import Firefox, FirefoxOptions
import requests

class KickBot():
    def __init__(self, url):
        self.url = url
        self.driver = None

    def setupWebBrowser(self):
        chrome_options = Options()
        chrome_options.add_argument("--headless")
        chrome_options.add_argument("--mute-audio")
        chrome_options.add_argument("--no-sandbox")
        chrome_options.add_argument("--disable-dev-shm-usage")
        chrome_options.add_argument("--disable-gpu")
        self.driver = uc.Chrome(options=chrome_options)

    def getToken(self):
        referer = self.driver.current_url
        response = requests.get(referer)
        csrf_token = response.headers.get('X-CSRFToken')
        return csrf_token

    def doTest(self):
        self.setupWebBrowser()
        self.driver.get(self.url)
        not_found_count = 0
        csrf_token = self.getToken()
        while True:
            try:

                button = self.driver.find_element(By.CSS_SELECTOR, "button.variant-action.size-sm")

                button.click()
                print("Button clicked!")
                not_found_count = 0
            except NoSuchElementException:

                if "Oops, Something went wrong" in self.driver.page_source:
                    self.driver.refresh()
                    print("Page refreshed!")
                    not_found_count = 0
                elif not_found_count >= 10:
                    print("Watch now button not found after 10 attempts. Stopping search.")
                    break
                elif "Checking if the site connection is secure" in self.driver.page_source:
                    self.driver.close()
                    sleep(5)
                    self.setupWebBrowser()
                    self.driver.get(self.url)
                    print("Browser restarted")
                    csrf_token = self.getToken()
                else:
                    print("Watch now button not found on this page")
                not_found_count += 1
            sleep(10)


        self.driver.execute_script("document.querySelector('#vjs_video_3 > div:nth-child(1) > video').style.display = 'none'")


        while True:
            pass

def run_script(url, num_threads):
    processes = []
    num_windows = 0
    for i in range(num_threads):
        bot = KickBot(url)
        process = Process(target=bot.doTest)
        process.start()
        processes.append(process)
        num_windows += 1
        if num_windows == 4:
            num_windows = 0
            sleep(6)
    for process in processes:
        process.join()

if __name__ == '__main__':
    url = input("Enter the website URL: ")
    num_threads = int(input('Enter the number of threads to use: '))
    run_script(url, num_threads)