binaryPuzzleSolver/helpers.py at master · Haolicopter/binaryPuzzleSolver · 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
#!/usr/bin/env python3

from selenium import webdriver
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.common.action_chains import ActionChains
import os


def getChromeDriver():
    chromedriver = "/usr/local/bin/chromedriver"
    os.environ["webdriver.chrome.driver"] = chromedriver
    chrome_options = Options()
    # This make Chromium reachable
    chrome_options.add_argument("--no-sandbox")
    # Overrides default choices
    chrome_options.add_argument("--no-default-browser-check")
    chrome_options.add_argument("--disable-user-media-security=true")
    chrome_options.add_argument("--use-fake-ui-for-media-stream")
    return webdriver.Chrome(chromedriver, chrome_options=chrome_options)


def drawCell(browser, cell, val):
    if val == 1:
        drawOne(browser, cell)
    elif val == 0:
        drawZero(browser, cell)
    else:
        drawNone(browser, cell)


def drawOne(browser, cell):
    if cell.text == 0:
        ActionChains(browser).click(cell).perform()
    elif cell.text == '':
        ActionChains(browser).click(cell).click(cell).perform()


def drawZero(browser, cell):
    if cell.text == '':
        ActionChains(browser).click(cell).perform()
    elif cell.text == '1':
        ActionChains(browser).click(cell).click(cell).perform()


def drawNone(browser, cell):
    if cell.text == '1':
        ActionChains(browser).click(cell).perform()
    elif cell.text == '0':
        ActionChains(browser).click(cell).click(cell).perform()