-
Notifications
You must be signed in to change notification settings - Fork 10
/
lifelights.py
80 lines (58 loc) · 2.21 KB
/
lifelights.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
74
75
76
77
78
79
80
import time
import sys
import os
import itertools
from util import Util
from widthwatcher import WidthWatcher
import yaml
sys.path.append(
os.path.join(os.path.dirname(os.path.realpath(__file__)), '..'))
def main():
"""Main entrypoint for script."""
profiles = next(os.walk('profiles'))[2]
if len(profiles) == 0:
Util.log(
"Unable to locate any game profiles in the profiles directory", "ERROR")
sys.exit()
for index, profile in enumerate(profiles):
# Use the filename as the name of the profile
print "%d - %s" % (index + 1, profile.replace("_", " ").rsplit('.', 1)[0].title())
profile_id = int(input('Enter the profile number to load: ')) - 1
if profile_id > len(profiles) or profile_id < 0:
Util.log("Profile number out of range, must be between 1 and %d" %
len(profiles), "ERROR")
sys.exit()
config_file = open("profiles/%s" % profiles[profile_id])
settings = yaml.safe_load(config_file)
config_file.close()
config_error = Util.has_valid_config(settings)
if config_error:
Util.log("Error found in configuration file -- %s" %
config_error, "ERROR")
sys.exit()
spinner = itertools.cycle(['-', '/', '|', '\\'])
watcher_list = [WidthWatcher(w) for w in settings["watchers"]]
window = Util.find_window_by_title(settings["window_title"])
if window is not None:
window = Util.resize_capture_area(window, settings)
while True:
if window is None:
sys.stdout.write("Waiting for window ... " + spinner.next() + "\r")
sys.stdout.flush()
window = Util.find_window_by_title(settings["window_title"])
if window is not None:
window = Util.resize_capture_area(window, settings)
time.sleep(0.3)
continue
time.sleep(float(settings["scan_interval"]))
screen = Util.screenshot(window)
if not screen.any():
continue
for watch in watcher_list:
watch.scan(screen)
watch.process()
if __name__ == '__main__':
try:
main()
except (KeyboardInterrupt, SystemExit):
print "Goodbye, hero."