Merge pull request #117 from oz123/fix_116

Change how heuy_consumer is started
coleifer · Sep 11, 2015 · ff400a9 · ff400a9
2 parents 25d6492 + 2a160db
commit ff400a9
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/huey/bin/huey_consumer.py b/huey/bin/huey_consumer.py
@@ -91,7 +91,7 @@ def load_huey(path):
         raise
 
 
-if __name__ == '__main__':
+def consumer_main():
     parser = get_option_parser()
     options, args = parser.parse_args()
 
@@ -115,3 +115,7 @@ def load_huey(path):
         options.scheduler_interval,
         options.periodic_task_interval)
     consumer.run()
+
+
+if __name__ == '__main__':
+    consumer_main()
diff --git a/setup.py b/setup.py
@@ -24,5 +24,10 @@
         'Framework :: Django',
     ],
     test_suite='runtests.runtests',
+    entry_points={
+        'console_scripts': [
+            'huey_consumer = huey.bin.huey_consumer:consumer_main'
+            ]
+    }
     scripts = ['huey/bin/huey_consumer.py'],
 )