Implement watching extra files from CLI Pylons#18

j-carl · May 22, 2017 · 0e9ca5b · 0e9ca5b
1 parent 611a157
commit 0e9ca5b
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/src/hupper/cli.py b/src/hupper/cli.py
@@ -1,21 +1,26 @@
 from __future__ import print_function
 
 import argparse
+import runpy
 import sys
 
 from .reloader import start_reloader
 
 
 def main():
+    reloader = start_reloader("hupper.cli.main")
+
     parser = argparse.ArgumentParser()
     parser.add_argument("-m", dest="module", required=True)
+    parser.add_argument("-w", dest="watch", action="append")
 
     args, unknown_args = parser.parse_known_args()
 
     sys.argv[1:] = unknown_args
     sys.path.insert(0, "")
 
-    start_reloader(
-        "runpy.run_module",
-        worker_args=[args.module],
-        worker_kwargs={"alter_sys": True, "run_name": "__main__"})
+    reloader.watch_files(args.watch)
+
+    return runpy.run_module(
+        args.module,
+        **{"alter_sys": True, "run_name": "__main__"})