From ace556e3bf36bb47b161f560f850ce375e49ef07 Mon Sep 17 00:00:00 2001
From: Janne Sinivirta <janne.sinivirta@gmail.com>
Date: Wed, 30 Aug 2017 21:19:14 +0300
Subject: [PATCH] move _should_stop from module global to class variable

---
 main.py         | 45 +++++++++++++++++++++++----------------------
 rpc/telegram.py |  4 ++--
 2 files changed, 25 insertions(+), 24 deletions(-)

diff --git a/main.py b/main.py
index 81f16607a..bef1753a0 100755
--- a/main.py
+++ b/main.py
@@ -27,27 +27,15 @@ CONFIG = get_conf()
 api_wrapper = get_exchange_api(CONFIG)
 
 
-@synchronized
-def get_instance(recreate=False):
-    """
-    Get the current instance of this thread. This is a singleton.
-    :param recreate: Must be True if you want to start the instance
-    :return: TradeThread instance
-    """
-    global _instance, _should_stop
-    if recreate and not _instance.is_alive():
-        logger.debug('Creating TradeThread instance')
-        _should_stop = False
-        _instance = TradeThread()
-    return _instance
-
-
-def stop_instance():
-    global _should_stop
-    _should_stop = True
-
-
 class TradeThread(threading.Thread):
+    def __init__(self):
+        super().__init__()
+        self._should_stop = False
+
+    def stop(self):
+        """ stops the trader thread """
+        self._should_stop = True
+
     def run(self):
         """
         Threaded main function
@@ -56,7 +44,7 @@ class TradeThread(threading.Thread):
         try:
             TelegramHandler.send_msg('*Status:* `trader started`')
             logger.info('Trader started')
-            while not _should_stop:
+            while not self._should_stop:
                 try:
                     self._process()
                 except (ConnectionError, JSONDecodeError, ValueError) as error:
@@ -113,7 +101,20 @@ class TradeThread(threading.Thread):
 
 # Initial stopped TradeThread instance
 _instance = TradeThread()
-_should_stop = False
+
+@synchronized
+def get_instance(recreate=False):
+    """
+    Get the current instance of this thread. This is a singleton.
+    :param recreate: Must be True if you want to start the instance
+    :return: TradeThread instance
+    """
+    global _instance
+    if recreate and not _instance.is_alive():
+        logger.debug('Creating TradeThread instance')
+        _should_stop = False
+        _instance = TradeThread()
+    return _instance
 
 
 def close_trade_if_fulfilled(trade):
diff --git a/rpc/telegram.py b/rpc/telegram.py
index 26aaa3005..838ddd527 100644
--- a/rpc/telegram.py
+++ b/rpc/telegram.py
@@ -174,10 +174,10 @@ class TelegramHandler(object):
         :param update: message update
         :return: None
         """
-        from main import get_instance, stop_instance
+        from main import get_instance
         if get_instance().is_alive():
             TelegramHandler.send_msg('`Stopping trader ...`', bot=bot)
-            stop_instance()
+            get_instance().stop()
         else:
             TelegramHandler.send_msg('*Status:* `already stopped`', bot=bot)