using watchdog to reload

thomasballinger · thomasballinger · commit 2077aa669e44 · 2014-07-31T19:54:03.000-04:00
diff --git a/bpython/curtsies.py b/bpython/curtsies.py
@@ -3,10 +3,10 @@
 import sys
 import code
 import logging
-import select
 from subprocess import Popen, PIPE
 from optparse import Option
 from itertools import izip
+from functools import wraps
 
 import curtsies
 import curtsies.window
@@ -19,6 +19,8 @@
 from bpython.translations import _
 from bpython.importcompletion import find_iterator
 
+import wdtest
+
 repl = None # global for `from bpython.curtsies import repl`
 #WARNING Will be a problem if more than one repl is ever instantiated this way
 
@@ -73,28 +75,38 @@ def mainloop(config, locals_, banner, interp=None, paste=None, interactive=True)
                 hide_cursor=False,
                 extra_bytes_callback=input_generator.unget_bytes) as window:
 
-            p = Popen(['python', '-m', 'bpython.curtsiesfrontend.beeper'], stdout=PIPE)
-
+            reload_requests = []
+            def request_reload():
+                reload_requests.append('reload!')
             refresh_requests = []
             def request_refresh():
                 refresh_requests.append(curtsies.events.RefreshRequestEvent())
+
+            watcher = wdtest.ModuleChangedEventHandler([], request_reload)
+
+            orig_import = __builtins__['__import__']
+            @wraps(orig_import)
+            def new_import(name, globals={}, locals={}, fromlist=[], level=-1):
+                m = orig_import(name, globals=globals, locals=locals, fromlist=fromlist)
+                watcher.add_module(m.__file__)
+                return m
+            __builtins__['__import__'] = new_import
+
             def event_or_refresh(timeout=None):
                 while True:
                     if refresh_requests:
                         yield refresh_requests.pop()
                     else:
-                        try:
-                            rs, ws, es = select.select([input_generator, p.stdout], [], [])
-                        except select.error as e:
-                            print e
-                            yield curtsies.events.SigIntEvent()
-                        else:
-                            for r in rs:
-                                if r is input_generator:
-                                    yield input_generator.send(timeout)
-                                else:
-                                    p.stdout.readline()
-                                    yield u'a'
+                        while True:
+                            if reload_requests:
+                                del reload_requests[:]
+                                e = '<F6>'
+                                print 'asdf'
+                            else:
+                                e = input_generator.send(.2)
+                            if e is not None:
+                                break
+                        yield e
 
             global repl # global for easy introspection `from bpython.curtsies import repl`
             with Repl(config=config,
diff --git a/wdtest.py b/wdtest.py
@@ -0,0 +1,49 @@
+import time
+import os
+from collections import defaultdict
+
+from bpython import importcompletion
+
+from watchdog.observers import Observer
+from watchdog.events import FileSystemEventHandler
+
+class ModuleChangedEventHandler(FileSystemEventHandler):
+    def __init__(self, paths, on_change):
+        self.dirs = defaultdict(list)
+        self.on_change = on_change
+        self.observer = Observer()
+        for path in paths:
+            self.add_module(path)
+        self.observer.start()
+
+    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)].append(path)
+
+    def on_any_event(self, event):
+        dirpath = os.path.dirname(event.src_path)
+        paths = [path + suffix for suffix in importcompletion.SUFFIXES
+                               for path in self.dirs[dirpath]]
+        if event.src_path in paths:
+            pass
+            #self.on_change()
+        self.on_change()
+
+if __name__ == '__main__':
+    m = ModuleChangedEventHandler([])
+    m.add_module('./wdtest.py')
+    try:
+        while True:
+            time.sleep(1)
+    except KeyboardInterrupt:
+        m.observer.stop()
+    m.observer.join()
+
