aboutsummaryrefslogtreecommitdiffstats
path: root/qface/watch.py
blob: e5d86bae62ca9620470007dd1b51a128f3e776bb (plain)
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
from watchdog.events import FileSystemEventHandler
from watchdog.observers import Observer
import click
from path import Path
import time
from subprocess import call

"""
Provides an API to monitor the file system
"""


class RunScriptChangeHandler(FileSystemEventHandler):
    def __init__(self, args, cwd):
        super().__init__()
        self.args = args
        self.cwd = cwd
        self.is_running = False

    def on_modified(self, event):
        if event.is_directory:
            return
        self.run()

    def run(self):
        if self.is_running:
            return
        self.is_running = True
        call(self.args, cwd=self.cwd)
        self.is_running = False


def monitor(args, watch, cwd=Path.getcwd()):
    """
    reloads the script given by argv when src files changes
    """
    watch = watch if isinstance(watch, (list, tuple)) else [watch]
    watch = [Path(entry).expand().abspath() for entry in watch]
    event_handler = RunScriptChangeHandler(args, cwd)
    observer = Observer()
    for entry in watch:
        if entry.isfile():
            entry = entry.parent
        click.secho('watch recursive: {0}'.format(entry), fg='blue')
        observer.schedule(event_handler, entry, recursive=True)
    event_handler.run()  # run always once
    observer.start()

    try:
        while True:
            time.sleep(1)
    except KeyboardInterrupt:
        observer.stop()
    observer.join()