[WIP] Add support for watching for changes and re-generating

exhibition/command.py

@@ -26,6 +26,7 @@
 import logging
 
 import click
+from watchdog.observers import Observer
 
 from . import __version__, config, utils
 
@@ -58,15 +59,26 @@ def gen():
 @exhibition.command(short_help="Serve site locally")
 @click.option("-s", "--server", default="localhost", help="Hostname to serve the site at.")
 @click.option("-p", "--port", default=8000, type=int, help="Port to serve the site at.")
-def serve(server, port):
+@click.option("--watch/--no-watch", default=True, help="Enables/Disables watching for changes.")
+def serve(server, port, watch):
     """
     Serve files from deploy_path as a webserver would
     """
     settings = config.Config.from_path(config.SITE_YAML_PATH)
     server_address = (server, port)
     httpd, thread = utils.serve(settings, server_address)
 
+    print("Watch status: %s" % watch)
+    if watch:
+        utils.gen(settings)
+        observer = Observer()
+        handler = utils.FileSystemChangeHandler(settings)
+        observer.schedule(handler, settings["content_path"], recursive=True)
+        observer.schedule(handler, settings["templates"], recursive=True)
+        observer.start()
+
     try:
         thread.join()
     except (KeyboardInterrupt, SystemExit):
         httpd.shutdown()
+        observer.join()

exhibition/utils.py

@@ -25,6 +25,8 @@
 import shutil
 import threading
 
+import watchdog
+
 from .node import Node
 
 logger = logging.getLogger("exhibition")
@@ -44,6 +46,21 @@ def gen(settings):
         item.render()
 
 
+class FileSystemChangeHandler(watchdog.events.FileSystemEventHandler):
+    """ Handler for changes on the file system """
+    settings = None
+
+    def __init__(self, settings, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.settings = settings
+
+    def on_any_event(self, event):
+        """ An event has happened, re-generate the site """
+        super().on_any_event(event)
+        print("Regenerating website")
+        gen(self.settings)
+
+
 class ExhibitionBaseHTTPRequestHandler(SimpleHTTPRequestHandler):
     def _sanitise_path(self, path):
         """ Strip leading and trailing / as well as base_url, if preset """

setup.py

@@ -40,6 +40,7 @@
         "ruamel.yaml",
         "typogrify",
         "pypandoc",
+        "watchdog",
     ],
     extras_require={
         "docs": [