Changed order of arguments for query script

tests/query_device.py

@@ -60,18 +60,20 @@ while arg_index < len(argv):
             continue
     elif not action:
         action = argv[arg_index]
-    elif action == 'write' and value == None:
-        try:
-            value = int(argv[arg_index])
-        except ValueError:
-            print(f'Wrong format for argument "value"')
-            exit(-3)
     else:
-        # rest of the arguments should be register numbers/names
-        try:
-            register_number.append(int(argv[arg_index]))
-        except ValueError:
-            register_name.append(argv[arg_index])
+        # if action is write, last argument should be value
+        if action == 'write' and arg_index == len(argv) - 1:
+            try:
+                value = int(argv[arg_index])
+            except ValueError:
+                print(f'Wrong format for argument "value"')
+                exit(-3)
+        else:
+            # rest of the arguments should be register numbers/names
+            try:
+                register_number.append(int(argv[arg_index]))
+            except ValueError:
+                register_name.append(argv[arg_index])
     arg_index += 1