scripts/qmp: Fix shebang and imports
[qemu.git] / scripts / qmp / qom-set
1 #!/usr/bin/env python3
2 ##
3 # QEMU Object Model test tools
4 #
5 # Copyright IBM, Corp. 2011
6 #
7 # Authors:
8 #  Anthony Liguori   <aliguori@us.ibm.com>
9 #
10 # This work is licensed under the terms of the GNU GPL, version 2 or later.  See
11 # the COPYING file in the top-level directory.
12 ##
13
14 import sys
15 import os
16
17 sys.path.append(os.path.join(os.path.dirname(__file__), '..', '..', 'python'))
18 from qemu.qmp import QEMUMonitorProtocol
19
20 cmd, args = sys.argv[0], sys.argv[1:]
21 socket_path = None
22 path = None
23 prop = None
24 value = None
25
26 def usage():
27     return '''environment variables:
28     QMP_SOCKET=<path | addr:port>
29 usage:
30     %s [-h] [-s <QMP socket path | addr:port>] <path>.<property> <value>
31 ''' % cmd
32
33 def usage_error(error_msg = "unspecified error"):
34     sys.stderr.write('%s\nERROR: %s\n' % (usage(), error_msg))
35     exit(1)
36
37 if len(args) > 0:
38     if args[0] == "-h":
39         print(usage())
40         exit(0);
41     elif args[0] == "-s":
42         try:
43             socket_path = args[1]
44         except:
45             usage_error("missing argument: QMP socket path or address");
46         args = args[2:]
47
48 if not socket_path:
49     if 'QMP_SOCKET' in os.environ:
50         socket_path = os.environ['QMP_SOCKET']
51     else:
52         usage_error("no QMP socket path or address given");
53
54 if len(args) > 1:
55     try:
56         path, prop = args[0].rsplit('.', 1)
57     except:
58         usage_error("invalid format for path/property/value")
59     value = args[1]
60 else:
61     usage_error("not enough arguments")
62
63 srv = QEMUMonitorProtocol(socket_path)
64 srv.connect()
65
66 print(srv.command('qom-set', path=path, property=prop, value=value))