generate-json.py: Update for new structure

test/generate-json.py

@@ -15,7 +15,7 @@
 # PingMessage is generated by this project
 import sys
 sys.path.append("../src/protocol/python")
-import PingMessage as Message
+from pingmessage import *
 
 
 def generate_fake_data(message_format: str) -> bytearray:
@@ -130,7 +130,7 @@ def main():
         'device': args.device,
         'messages': []
     }
-    for message_id, ping_message in Message.payloadDict.items():
+    for message_id, ping_message in payload_dict.items():
         data = {
             'name': ping_message['name'],
             'id': message_id,
@@ -140,13 +140,13 @@ def main():
         var_payload = {}
         key = ''
         # Generate random text or array of bytes
-        if message_id in [Message.PING1D_ASCII_TEXT, Message.PING1D_NACK]:
-            if message_id == Message.PING1D_ASCII_TEXT:
+        if message_id in [PING1D_ASCII_TEXT, PING1D_NACK]:
+            if message_id == PING1D_ASCII_TEXT:
                 key = 'ascii_message'
-            elif message_id == Message.PING1D_NACK:
+            elif message_id == PING1D_NACK:
                 key = 'nack_message'
             var_payload[key] = b' !"#$%&\'()*+,-./0123456789:;<=>?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_`abcdefghijklmnopqrstuvwxyz{|}~\0'
-        elif message_id == Message.PING1D_PROFILE:
+        elif message_id == PING1D_PROFILE:
             var_payload['profile_data'] = [x for x in range(200)]
 
         # Create data with all bytes of the message