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()