add entry point and signal handler

elastalert/elastalert.py

@@ -6,6 +6,8 @@
 import sys
 import time
 import traceback
+import os
+import signal
 from email.mime.text import MIMEText
 from smtplib import SMTP
 from smtplib import SMTPException
@@ -1259,7 +1261,19 @@ def next_alert_time(self, rule, name, timestamp):
         return timestamp + wait, exponent
 
 
-if __name__ == '__main__':
-    client = ElastAlerter(sys.argv[1:])
+def handle_signal(signal, frame):
+    elastalert_logger.info('SIGINT received, stopping ElastAlert...')
+    # use os._exit to exit immediately and avoid someone catching SystemExit
+    os._exit(0)
+
+
+def main(args=None):
+    signal.signal(signal.SIGINT, handle_signal)
+    if not args:
+        args = sys.argv[1:]
+    client = ElastAlerter(args)
     if not client.args.silence:
         client.start()
+
+if __name__ == '__main__':
+    sys.exit(main(sys.argv[1:]))

setup.py

@@ -17,7 +17,8 @@
     entry_points={
         'console_scripts': ['elastalert-create-index=elastalert.create_index:main',
                             'elastalert-test-rule=elastalert.test_rule:main',
-                            'elastalert-rule-from-kibana=elastalert.rule_from_kibana:main']},
+                            'elastalert-rule-from-kibana=elastalert.rule_from_kibana:main',
+                            'elastalert=elastalert.elastalert:main']},
     packages=find_packages(),
     package_data={'elastalert': ['schema.yaml']},
     install_requires=[