include diagnostics module when verbose logging enabled

1 files changed, 3 insertions, 1 deletions

diff --git a/src/pump_bot.py b/src/pump_bot.py
index 7bf9bd8..103f2f4 100755
--- a/src/pump_bot.py
+++ b/src/pump_bot.py
@@ -3,7 +3,7 @@ from enum import Enum
 from typing import Optional
 from telegram import ReplyKeyboardMarkup, User
 
-from home.config import config
+from home.config import config, is_development_mode
 from home.telegram import bot
 from home.telegram._botutil import user_any_name
 from home.relay.sunxi_h3_client import RelayClient
@@ -170,6 +170,8 @@ def markup(ctx: Optional[bot.Context]) -> Optional[ReplyKeyboardMarkup]:
 
 if __name__ == '__main__':
     mqtt = MqttNode(node_id=config.get('mqtt_water_relay.node_id'))
+    if is_development_mode():
+        add_mqtt_module('diagnostics')
     mqtt_relay_module = add_mqtt_module(mqtt, 'relay')
 
     mqtt.configure_tls()