scripts/qmp: Fix shebang and imports
[qemu.git] / scripts / qmp / qom-get
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
25 def usage():
26     return '''environment variables:
27     QMP_SOCKET=<path | addr:port>
28 usage:
29     %s [-h] [-s <QMP socket path | addr:port>] <path>.<property>
30 ''' % cmd
31
32 def usage_error(error_msg = "unspecified error"):
33     sys.stderr.write('%s\nERROR: %s\n' % (usage(), error_msg))
34     exit(1)
35
36 if len(args) > 0:
37     if args[0] == "-h":
38         print(usage())
39         exit(0);
40     elif args[0] == "-s":
41         try:
42             socket_path = args[1]
43         except:
44             usage_error("missing argument: QMP socket path or address");
45         args = args[2:]
46
47 if not socket_path:
48     if 'QMP_SOCKET' in os.environ:
49         socket_path = os.environ['QMP_SOCKET']
50     else:
51         usage_error("no QMP socket path or address given");
52
53 if len(args) > 0:
54     try:
55         path, prop = args[0].rsplit('.', 1)
56     except:
57         usage_error("invalid format for path/property/value")
58 else:
59     usage_error("not enough arguments")
60
61 srv = QEMUMonitorProtocol(socket_path)
62 srv.connect()
63
64 rsp = srv.command('qom-get', path=path, property=prop)
65 if type(rsp) == dict:
66     for i in rsp.keys():
67         print('%s: %s' % (i, rsp[i]))
68 else:
69     print(rsp)