simple_mouse_tracker/config.py at master · rorro/simple_mouse_tracker · 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
from pathlib import Path

class Config():
    def __init__(self):
        self.SCRIPT_FOLDER_PATH = str(Path(__file__).parent)
        self.HOME_FOLDER = str(Path.home())
        self.CONFIG_FILE = self.HOME_FOLDER + "/" + ".simplemousetracker"

        self.DEFAULT_CONFIG = [
                "# The keybinding for starting/stopping the mouse tracking.",
                "# Valid modifiers: control, shift, super (windows key), alt.",
                "# Read more about valid bindings at https://github.com/timeyyy/system_hotkey",
                "start_tracking=control-s",
                "",
                "# The keybinding for pausing/resuming the mouse tracking.",
                "pause_tracking=control-p",
                "",
                "# Where to save the tracked files. Leave blank to save in same",
                "# folder as the script. The path should be absolute.",
                "# If folder doesn't exist, nothing will be saved.",
                "save_folder=",
                "",
                "# Colored lines when exporting/showing tracked data.",
                "# Only true is true, everything else is false.",
                "colored_speed=true",
                "",
                "# Default color for drawing tracked data.",
                "# Only applies when colored_speed is set to false.",
                "default_color=#000000",
                ]

        # Default config values.
        # They are here to make sure the program works if the config
        # file is not configured correctly.
        self.start_tracking_binding = ("control", "s")
        self.pause_tracking_binding = ("control", "p")
        self.save_folder = self.SCRIPT_FOLDER_PATH
        self.colored_speed = True
        self.default_color = "#000000"

        if not Path(self.CONFIG_FILE).is_file():
            print("Config file doesn't exist. Creating default.")
            config = open(self.CONFIG_FILE, "w")
            config.write("\n".join(self.DEFAULT_CONFIG))
            config.close()
        else:
            config = open(self.CONFIG_FILE, "r")

            for line in config:
                line = line.rstrip()
                if not line or line.startswith("#"):
                    continue

                split = line.split("=")
                lhs = split[0]
                rhs = split[1]

                if lhs == "start_tracking":
                    keys = rhs.split("-")
                    if len(keys) == 1:
                        self.start_tracking_binding = rhs
                    else:
                        self.start_tracking_binding = tuple(keys)
                elif lhs == "pause_tracking":
                    keys = rhs.split("-")
                    if len(keys) == 1:
                        self.pause_tracking_binding = rhs
                    else:
                        self.pause_tracking_binding = tuple(keys)
                elif lhs == "save_folder":
                    if rhs:
                        self.save_folder = rhs
                    else:
                        self.save_folder = self.SCRIPT_FOLDER_PATH
                elif lhs == "colored_speed":
                    if rhs != "true":
                        self.colored_speed = False
                elif lhs == "default_color":
                    self.default_color = rhs

            config.close()