-
Notifications
You must be signed in to change notification settings - Fork 4
/
wtrobot.py
73 lines (58 loc) · 2.29 KB
/
wtrobot.py
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
import sys, os, json
import logging
from src import commmandParser
def logger_init(filename):
logging.basicConfig(
filename=filename,
level=logging.INFO,
format="%(levelname)s - %(asctime)s - %(filename)s - %(lineno)d - %(message)s",
)
console = logging.StreamHandler()
console.setLevel(logging.INFO)
formatter = logging.Formatter(
"%(levelname)s - %(asctime)s - %(filename)s - %(lineno)d - %(message)s"
)
console.setFormatter(formatter)
logging.getLogger("").addHandler(console)
# marker to denote new log starting
logging.info("------------------new---------------------")
if __name__ == "__main__":
config_file_name = "config.json"
config = dict()
count = 0
try:
with open(config_file_name) as fobj:
config = json.load(fobj)
count = len(config)
except (OSError, json.decoder.JSONDecodeError):
print(
"config file missing or incorrect, please provide following configurations..."
)
if config is None or "script_filepath" not in config.keys():
config["script_filepath"] = (
input("Test script filename: test.yaml ? ") or "test.yaml"
)
if config is None or "browser" not in config.keys():
config["browser"] = input("Which browser : firefox ? ") or "firefox"
if config is None or "webdriver_path" not in config.keys():
config["webdriver_path"] = (
input("Selenium webdriver : ./selenium_drivers/geckodriver ? ")
or "./selenium_drivers/geckodriver"
)
if config is None or "locale" not in config.keys():
config["locale"] = input("Which browser locale : en_US ? ") or "en_US"
if config is None or "log" not in config.keys():
config["log"] = (
input("WTRobot execution log file path : wtlog.log ? ") or "wtlog.log"
)
logger_init(config["log"])
if not os.path.exists(config["script_filepath"]):
logging.error("Invalid script file path")
sys.exit(0)
if not os.path.exists(config["webdriver_path"]):
logging.error("Invalid webdriver file path")
sys.exit(0)
if count < len(config):
with open(config_file_name, "w") as fobj:
json.dump(config, fobj, indent=4)
obj = commmandParser(config)