Add config flags for the scheduler and router

2 files changed, 22 insertions, 0 deletions

diff --git a/bin/emq-router b/bin/emq-router
index f10d8da..425a985 100755
--- a/bin/emq-router
+++ b/bin/emq-router
@@ -1,8 +1,20 @@
 #!/usr/bin/env python
 # -*- mode: python -*-
+import argparse
+
 from eventmq.log import setup_logger
 from eventmq.router import Router
 
 if __name__ == "__main__":
+        parser = argparse.ArgumentParser(
+            description='Route job messages to their appropriate queues')
+        parser.add_argument('--config', '-C', type=str, nargs='?',
+                        help='manually specify the location of eventmq.conf')
+
+        args = parser.parse_args()
+
+        if args.config:
+            conf.CONFIG_FILE = args.config
+
     r = Router()
     r.router_main()
diff --git a/bin/emq-scheduler b/bin/emq-scheduler
index 106b6bf..eae99cf 100755
--- a/bin/emq-scheduler
+++ b/bin/emq-scheduler
@@ -3,5 +3,15 @@
 from eventmq.scheduler import Scheduler
 
 if __name__ == "__main__":
+        parser = argparse.ArgumentParser(
+            description='Listen for scheduled jobs and run them')
+        parser.add_argument('--config', '-C', type=str, nargs='?',
+                        help='manually specify the location of eventmq.conf')
+
+        args = parser.parse_args()
+
+        if args.config:
+            conf.CONFIG_FILE = args.config
+
     s = Scheduler()
     s.scheduler_main()