70 lines
2.3 KiB
Python
70 lines
2.3 KiB
Python
import logging
|
|
import time
|
|
import os
|
|
|
|
from Config import config
|
|
|
|
if config.debug and config.action == "main":
|
|
try:
|
|
import watchdog
|
|
import watchdog.observers
|
|
import watchdog.events
|
|
logging.debug("Watchdog fs listener detected, source code autoreload enabled")
|
|
enabled = True
|
|
except Exception as err:
|
|
logging.debug("Watchdog fs listener could not be loaded: %s" % err)
|
|
enabled = False
|
|
else:
|
|
enabled = False
|
|
|
|
|
|
class DebugReloader:
|
|
def __init__(self, paths=None):
|
|
if not paths:
|
|
paths = ["src", "plugins", config.data_dir + "/__plugins__"]
|
|
self.log = logging.getLogger("DebugReloader")
|
|
self.last_chaged = 0
|
|
self.callbacks = []
|
|
if enabled:
|
|
self.observer = watchdog.observers.Observer()
|
|
event_handler = watchdog.events.FileSystemEventHandler()
|
|
event_handler.on_modified = event_handler.on_deleted = self.onChanged
|
|
event_handler.on_created = event_handler.on_moved = self.onChanged
|
|
for path in paths:
|
|
if not os.path.isdir(path):
|
|
continue
|
|
self.log.debug("Adding autoreload: %s" % path)
|
|
self.observer.schedule(event_handler, path, recursive=True)
|
|
self.observer.start()
|
|
|
|
def addCallback(self, f):
|
|
self.callbacks.append(f)
|
|
|
|
def onChanged(self, evt):
|
|
path = evt.src_path
|
|
ext = path.rsplit(".", 1)[-1]
|
|
if ext not in ["py", "json"] or "Test" in path or time.time() - self.last_chaged < 1.0:
|
|
return False
|
|
self.last_chaged = time.time()
|
|
if os.path.isfile(path):
|
|
time_modified = os.path.getmtime(path)
|
|
else:
|
|
time_modified = 0
|
|
self.log.debug("File changed: %s reloading source code (modified %.3fs ago)" % (evt, time.time() - time_modified))
|
|
if time.time() - time_modified > 5: # Probably it's just an attribute change, ignore it
|
|
return False
|
|
|
|
time.sleep(0.1) # Wait for lock release
|
|
for callback in self.callbacks:
|
|
try:
|
|
callback()
|
|
except Exception as err:
|
|
self.log.exception(err)
|
|
|
|
def stop(self):
|
|
if enabled:
|
|
self.observer.stop()
|
|
self.log.debug("Stopped autoreload observer")
|
|
|
|
watcher = DebugReloader()
|