Merge pull request #82 from gcarq/feature/handle-process-signals

handle SIGINT, SIGTERM and SIGABRT process signals

freqtrade/main.py

@@ -6,6 +6,7 @@ import time
 import traceback
 from datetime import datetime
 from typing import Dict, Optional
+from signal import signal, SIGINT, SIGABRT, SIGTERM
 
 from jsonschema import validate
 
@@ -223,6 +224,23 @@ def init(config: dict, db_url: Optional[str] = None) -> None:
     else:
         update_state(State.STOPPED)
 
+    # Register signal handlers
+    for sig in (SIGINT, SIGTERM, SIGABRT):
+        signal(sig, cleanup)
+
+
+def cleanup(*args, **kwargs) -> None:
+    """
+    Cleanup the application state und finish all pending tasks
+    :return: None
+    """
+    telegram.send_msg('*Status:* `Stopping trader...`')
+    logger.info('Stopping trader and cleaning up modules...')
+    update_state(State.STOPPED)
+    persistence.cleanup()
+    telegram.cleanup()
+    exit(0)
+
 
 def app(config: dict) -> None:
     """
@@ -251,10 +269,10 @@ def app(config: dict) -> None:
                 time.sleep(exchange.EXCHANGE.sleep_time)
             old_state = new_state
     except RuntimeError:
-        telegram.send_msg('*Status:* Got RuntimeError: ```\n{}\n```'.format(traceback.format_exc()))
+        telegram.send_msg(
+            '*Status:* Got RuntimeError:\n```\n{}\n```'.format(traceback.format_exc())
+        )
         logger.exception('RuntimeError. Trader stopped!')
-    finally:
-        telegram.send_msg('*Status:* `Trader has stopped`')
 
 
 def main():

freqtrade/persistence.py

@@ -37,6 +37,14 @@ def init(config: dict, db_url: Optional[str] = None) -> None:
     Base.metadata.create_all(engine)
 
 
+def cleanup() -> None:
+    """
+    Flushes all pending operations to disk.
+    :return: None
+    """
+    Trade.session.flush()
+
+
 class Trade(Base):
     __tablename__ = 'trades'
 

freqtrade/rpc/telegram.py

@@ -17,7 +17,7 @@ logging.getLogger('requests.packages.urllib3').setLevel(logging.INFO)
 logging.getLogger('telegram').setLevel(logging.INFO)
 logger = logging.getLogger(__name__)
 
-_updater = None
+_updater: Updater = None
 _CONF = {}
 
 
@@ -61,6 +61,14 @@ def init(config: dict) -> None:
     )
 
 
+def cleanup() -> None:
+    """
+    Stops all running telegram threads.
+    :return: None
+    """
+    _updater.stop()
+
+
 def authorized_only(command_handler: Callable[[Bot, Update], None]) -> Callable[..., Any]:
     """
     Decorator to check if the message comes from the correct chat_id