Handle arguments uniformly (by using context.args)

freqtrade · Sep 2, 2019 · 3b15cce · 3b15cce
1 parent 8cad90f
commit 3b15cce
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/freqtrade/rpc/telegram.py b/freqtrade/rpc/telegram.py
@@ -407,7 +407,7 @@ def _forcesell(self, update: Update, context: CallbackContext) -> None:
         :return: None
         """
 
-        trade_id = update.message.text.replace('/forcesell', '').strip()
+        trade_id = context.args[0] if len(context.args) > 0 else None
         try:
             msg = self._rpc_forcesell(trade_id)
             self._send_msg('Forcesell Result: `{result}`'.format(**msg))
@@ -425,9 +425,8 @@ def _forcebuy(self, update: Update, context: CallbackContext) -> None:
         :return: None
         """
 
-        message = update.message.text.replace('/forcebuy', '').strip().split()
-        pair = message[0]
-        price = float(message[1]) if len(message) > 1 else None
+        pair = context.args[0]
+        price = float(context.args[1]) if len(context.args) > 1 else None
         try:
             self._rpc_forcebuy(pair, price)
         except RPCException as e: