Swoo_Hack.py

import pytesseract
import pyscreenshot as ImageGrab
from bs4 import BeautifulSoup
from selenium import webdriver
import time
import threading


def replacements():
    global toSearch
    normal_replaces = {'The ', 'Who ', 'What ', 'Which ', 'Whom ', 'of ', 'these ', 'which'}
    plus_replaces = {' following ', ' is ', ' these ', ' an ', ' a ', ' the ', ' which ', ' not ', ' what ', ' who '}

    for rep in normal_replaces:
        toSearch = toSearch.replace(rep, '')
    toSearch = toSearch.lower()

    for rep in plus_replaces:
        toSearch = toSearch.replace(rep, '+')

    toSearch = toSearch.replace('?', '').replace('\'', '').replace('\n', '+').replace(' ', '+').replace('++', '+').replace(',', '')
    print("\n" + toSearch + "\n")


def printer(a, results):
    print(a, end = "")
    counter_imp = [0] * 3
    counter = [0] * 3
    for result in results:
        result = result.getText().lower()
        for i in range(0, 3):
            counter_imp[i] = counter_imp[i] + result.count(options[i])
            words = options[i].replace('the ', '').strip().split(' ')
            for word in words:
                counter[i] = counter[i] + result.count(word.strip())
    start = "\033[1m"
    end = "\033[0;0m"

    print(start)
    for i in range(0, 3):
        print(str(counter_imp[i]), end="  ")
    print(end)
    for i in range(0, 3):
        print(str(counter[i]), end="  ")
    print("\n")


def bing(url, class_div, driver):
    global count
    try:
        driver.get(url)
        html = driver.page_source
        soup = BeautifulSoup(html, "html.parser")
    except Exception as e:
        print("ERROR")
        if (time.time() - start_time) >= 10 or count > 10:
            print("Timeout " + str(count))
            count = count + 1
            return
        else:
            count = count + 1
        bing(url, class_div, driver)

    bing_results = soup.findAll("div", {"class": class_div})
    printer("Bing", bing_results)


def google(url, class_div, driver):
    global count
    try:
        driver.get(url)
        html = driver.page_source
        soup = BeautifulSoup(html, "html.parser")
    except Exception as e:
        print("ERROR")
        if (time.time() - start_time) >= 10 or count > 10:
            print("Timeout " + str(count))
            count = count + 1
            return
        else:
            count = count + 1
        google(url, class_div, driver)

    google_results = soup.findAll("div", {"class": class_div})
    printer("Google", google_results)

start_time = time.time()

SWOO = (120, 658, 550, 840)

im1 = ImageGrab.grab(SWOO)
options = pytesseract.image_to_string(im1, lang='eng')

options = options.replace(options[:3], '').strip()
options = options.split("\n\n")
toSearch = options[0]
options.pop(0)
options[0] = options[0].lower()
options[1] = options[1].lower()
options[2] = options[2].lower()

replacements()

URL1 = "http://www.bing.com/search?q=" + toSearch
CLASS1 = "b_caption"

URL2 = "http://www.google.com/search?q=" + toSearch
CLASS2 = "rc"

chromeOptions = webdriver.ChromeOptions()
chromeOptions.add_argument("headless")
driver = webdriver.Chrome(chrome_options=chromeOptions)

#thread1 = threading.Thread(target=bing(URL1, CLASS1, driver), args=())
#thread2 = threading.Thread(target=google(URL2, CLASS2, driver), args=())

bing(URL1, CLASS1, driver)
google(URL2, CLASS2, driver)

driver.close()
driver.quit()

"""
thread1.start()
thread2.start()

thread1.join()
thread2.join()
"""

print("--- %s seconds ---" % (time.time() - start_time))