simple_mouse_tracker/track_mouse.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
from pymouse import PyMouse
import threading
from datetime import datetime
import time

class MouseTracker():
    def __init__(self, save_folder):
        now = datetime.now()
        self.save_folder = save_folder
        self.file_name = "mouse_track-{}{}{}_{}{}{}.tracked".format(
                now.year,
                now.month,
                now.day,
                now.hour,
                now.minute,
                now.second)

        self.mouse = PyMouse()

        self.tracking = False
        self.paused = False
        self.thread = threading.Thread(target=self.start)

        self.tracked = 0
        self.sf = None
        self.is_last_paused = False


    def start(self):
        self.tracking = True
        last_pos = (-1,-1)

        file_path = self.save_folder + "/" + self.file_name

        try:
            self.sf = open(file_path, 'a')
            while self.tracking:
                if not self.paused:
                    mouse_x, mouse_y = self.mouse.position()
                    now = time.time()


                    if last_pos != (mouse_x, mouse_y):
                        self.sf.write(str(mouse_x) + ',' + str(mouse_y) + "," + str(now) + "\n")
                        self.is_last_paused = False
                        last_pos = (mouse_x, mouse_y)
                        self.tracked += 1

            self.sf.close()
        except FileNotFoundError:
            print("Save folder does not exist.")

    def terminate(self):
        self.tracking = False
        self.paused = False