From 1bf475fa1a1f5281f01b1b6a8ab3104f7553c1e8 Mon Sep 17 00:00:00 2001
From: Matthias <xmatthias@outlook.com>
Date: Mon, 20 Jan 2020 20:24:40 +0100
Subject: [PATCH] Remove .get calls for dry_run - it's a mandatory property

---
 freqtrade/freqtradebot.py    | 5 ++---
 freqtrade/rpc/rpc.py         | 6 +++---
 freqtrade/rpc/rpc_manager.py | 2 +-
 tests/test_configuration.py  | 2 +-
 4 files changed, 7 insertions(+), 8 deletions(-)

diff --git a/freqtrade/freqtradebot.py b/freqtrade/freqtradebot.py
index e712892f1..acff2b2ad 100644
--- a/freqtrade/freqtradebot.py
+++ b/freqtrade/freqtradebot.py
@@ -63,8 +63,7 @@ class FreqtradeBot:
 
         self.exchange = ExchangeResolver.load_exchange(self.config['exchange']['name'], self.config)
 
-        persistence.init(self.config.get('db_url', None),
-                         clean_open_orders=self.config.get('dry_run', False))
+        persistence.init(self.config.get('db_url', None), clean_open_orders=self.config['dry_run'])
 
         self.wallets = Wallets(self.config, self.exchange)
 
@@ -930,7 +929,7 @@ class FreqtradeBot:
 
         # if stoploss is on exchange and we are on dry_run mode,
         # we consider the sell price stop price
-        if self.config.get('dry_run', False) and sell_type == 'stoploss' \
+        if self.config['dry_run'] and sell_type == 'stoploss' \
            and self.strategy.order_types['stoploss_on_exchange']:
             limit = trade.stop_loss
 
diff --git a/freqtrade/rpc/rpc.py b/freqtrade/rpc/rpc.py
index 0469d418d..d58b99f39 100644
--- a/freqtrade/rpc/rpc.py
+++ b/freqtrade/rpc/rpc.py
@@ -88,7 +88,7 @@ class RPC:
         """
         config = self._freqtrade.config
         val = {
-            'dry_run': config.get('dry_run', False),
+            'dry_run': config['dry_run'],
             'stake_currency': config['stake_currency'],
             'stake_amount': config['stake_amount'],
             'minimal_roi': config['minimal_roi'].copy(),
@@ -337,7 +337,7 @@ class RPC:
                 'stake': stake_currency,
             })
         if total == 0.0:
-            if self._freqtrade.config.get('dry_run', False):
+            if self._freqtrade.config['dry_run']:
                 raise RPCException('Running in Dry Run, balances are not available.')
             else:
                 raise RPCException('All balances are zero.')
@@ -351,7 +351,7 @@ class RPC:
             'symbol': symbol,
             'value': value,
             'stake': stake_currency,
-            'note': 'Simulated balances' if self._freqtrade.config.get('dry_run', False) else ''
+            'note': 'Simulated balances' if self._freqtrade.config['dry_run'] else ''
         }
 
     def _rpc_start(self) -> Dict[str, str]:
diff --git a/freqtrade/rpc/rpc_manager.py b/freqtrade/rpc/rpc_manager.py
index cb9e697e9..f687fe4d1 100644
--- a/freqtrade/rpc/rpc_manager.py
+++ b/freqtrade/rpc/rpc_manager.py
@@ -62,7 +62,7 @@ class RPCManager:
                 logger.error(f"Message type {msg['type']} not implemented by handler {mod.name}.")
 
     def startup_messages(self, config, pairlist) -> None:
-        if config.get('dry_run', False):
+        if config['dry_run']:
             self.send_msg({
                 'type': RPCMessageType.WARNING_NOTIFICATION,
                 'status': 'Dry run is enabled. All trades are simulated.'
diff --git a/tests/test_configuration.py b/tests/test_configuration.py
index a4d4c4abc..cbcd6416a 100644
--- a/tests/test_configuration.py
+++ b/tests/test_configuration.py
@@ -323,7 +323,7 @@ def test_load_dry_run(default_conf, mocker, config_value, expected, arglist) ->
     configuration = Configuration(Arguments(arglist).get_parsed_arg())
     validated_conf = configuration.load_config()
 
-    assert validated_conf.get('dry_run') is expected
+    assert validated_conf['dry_run'] is expected
 
 
 def test_load_custom_strategy(default_conf, mocker) -> None: