forked from bpython/bpython
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathfilewatch.py
More file actions
77 lines (67 loc) · 2.5 KB
/
filewatch.py
File metadata and controls
77 lines (67 loc) · 2.5 KB
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
import time
import os
from collections import defaultdict
from bpython import importcompletion
try:
from watchdog.observers import Observer
from watchdog.events import FileSystemEventHandler
except ImportError:
def ModuleChangedEventHandler(*args):
return None
else:
class ModuleChangedEventHandler(FileSystemEventHandler):
def __init__(self, paths, on_change):
self.dirs = defaultdict(set)
self.on_change = on_change
self.modules_to_add_later = []
self.observer = Observer()
self.old_dirs = defaultdict(set)
self.started = False
for path in paths:
self.add_module(path)
def reset(self):
self.dirs = defaultdict(set)
del self.modules_to_add_later[:]
self.old_dirs = defaultdict(set)
self.observer.unschedule_all()
def add_module(self, path):
"""Add a python module to track changes to"""
path = os.path.abspath(path)
for suff in importcompletion.SUFFIXES:
if path.endswith(suff):
path = path[:-len(suff)]
break
dirname = os.path.dirname(path)
if dirname not in self.dirs:
self.observer.schedule(self, dirname, recursive=False)
self.dirs[os.path.dirname(path)].add(path)
def add_module_later(self, path):
self.modules_to_add_later.append(path)
def activate(self):
if not self.started:
self.started = True
self.observer.start()
self.dirs = self.old_dirs
for dirname in self.dirs:
self.observer.schedule(self, dirname, recursive=False)
for module in self.modules_to_add_later:
self.add_module(module)
del self.modules_to_add_later[:]
def deactivate(self):
self.observer.unschedule_all()
self.old_dirs = self.dirs
self.dirs = defaultdict(set)
def on_any_event(self, event):
dirpath = os.path.dirname(event.src_path)
paths = [path + '.py' for path in self.dirs[dirpath]]
if event.src_path in paths:
self.on_change(event.src_path)
if __name__ == '__main__':
m = ModuleChangedEventHandler([])
m.add_module('./wdtest.py')
try:
while True:
time.sleep(1)
except KeyboardInterrupt:
m.observer.stop()
m.observer.join()