Update protocol parsing logic and warn on invalid protocol

libnymea-modbus/tools/generate-connection.py

@@ -533,19 +533,11 @@ if 'blocks' in registerJson:
     validateBlocks(registerJson['blocks'])
 
 # Create classes depending on the protocol
-writeTcp = True
+writeTcp = protocol in ["TCP", "BOTH"]
-writeRtu = False
+writeRtu = protocol in ["RTU", "BOTH"]
-
+if not writeTcp and not writeRtu:
-if protocol == 'TCP':
+    logger.warning('Invalid protocol definition. Please use TCP, RTU or BOTH.')
-    writeTcp = True
+    exit(1)
-    writeRtu = False
-elif protocol == 'RTU':
-    writeTcp = False
-    writeRtu = True
-else:
-    # Any other value generates both classes
-    writeTcp = True
-    writeRtu = True
 
 headerFiles = []
 sourceFiles = []