Commit 8d27c023 authored by Karol Latecki's avatar Karol Latecki Committed by Jim Harris
Browse files

scripts/rpc.py: move exit on error to rpc.py 



Moving exit in case of error to rpc.py
By using exceptions in  client.py it is easier to use it
as module in other scripts.

Change-Id: Ic9ecef481ef5b6a169e1293b3dcf859ffbbe0e89
Signed-off-by: default avatarKarol Latecki <karol.latecki@intel.com>
Reviewed-on: https://review.gerrithub.io/405739


Tested-by: default avatarSPDK Automated Test System <sys_sgsw@intel.com>
Reviewed-by: default avatarDaniel Verkamp <daniel.verkamp@intel.com>
Reviewed-by: default avatarJim Harris <james.r.harris@intel.com>
parent 3bce3519

scripts/rpc.py

+92 −1
Original line number Diff line number Diff line 
#!/usr/bin/env python



from rpc.client import print_dict

from rpc.client import print_dict, JSONRPCException



import argparse

import rpc
@@ -15,6 +15,16 @@ def print_array(a): 
    print(" ".join((quote(v) for v in a)))





def call_cmd(func):

    def rpc_cmd(*args, **kwargs):

        try:

            func(*args, **kwargs)

        except JSONRPCException as ex:

            print(ex.message)

            exit(1)

    return rpc_cmd





if __name__ == "__main__":

    parser = argparse.ArgumentParser(

        description='SPDK RPC command line interface')
@@ -30,12 +40,14 @@ if __name__ == "__main__": 
                        help='Verbose mode', action='store_true')

    subparsers = parser.add_subparsers(help='RPC methods')



    @call_cmd

    def get_rpc_methods(args):

        print_dict(rpc.get_rpc_methods(args.client, args))



    p = subparsers.add_parser('get_rpc_methods', help='Get list of supported RPC methods')

    p.set_defaults(func=get_rpc_methods)



    @call_cmd

    def save_config(args):

        rpc.save_config(args.client, args)
@@ -46,6 +58,7 @@ if __name__ == "__main__": 
    indent level is 2. If writing to file of filename is '-' then default is compact mode.""", type=int, default=2)

    p.set_defaults(func=save_config)



    @call_cmd

    def load_config(args):

        rpc.load_config(args.client, args)
@@ -55,6 +68,7 @@ if __name__ == "__main__": 
    p.set_defaults(func=load_config)



    # app

    @call_cmd

    def kill_instance(args):

        rpc.app.kill_instance(args.client, args)
@@ -62,6 +76,7 @@ if __name__ == "__main__": 
    p.add_argument('sig_name', help='signal will be sent to server.')

    p.set_defaults(func=kill_instance)



    @call_cmd

    def context_switch_monitor(args):

        print_dict(rpc.app.context_switch_monitor(args.client, args))
@@ -71,6 +86,7 @@ if __name__ == "__main__": 
    p.set_defaults(func=context_switch_monitor)



    # bdev

    @call_cmd

    def construct_malloc_bdev(args):

        print_array(rpc.bdev.construct_malloc_bdev(args.client, args))
@@ -83,6 +99,7 @@ if __name__ == "__main__": 
    p.add_argument('block_size', help='Block size for this bdev', type=int)

    p.set_defaults(func=construct_malloc_bdev)



    @call_cmd

    def construct_null_bdev(args):

        print_array(rpc.bdev.construct_null_bdev(args.client, args))
@@ -95,6 +112,7 @@ if __name__ == "__main__": 
    p.add_argument('block_size', help='Block size for this bdev', type=int)

    p.set_defaults(func=construct_null_bdev)



    @call_cmd

    def construct_aio_bdev(args):

        print_array(rpc.bdev.construct_aio_bdev(args.client, args))
@@ -105,6 +123,7 @@ if __name__ == "__main__": 
    p.add_argument('block_size', help='Block size for this bdev', type=int, default=argparse.SUPPRESS)

    p.set_defaults(func=construct_aio_bdev)



    @call_cmd

    def construct_nvme_bdev(args):

        print_array(rpc.bdev.construct_nvme_bdev(args.client, args))
@@ -122,6 +141,7 @@ if __name__ == "__main__": 
    p.add_argument('-n', '--subnqn', help='NVMe-oF target subnqn')

    p.set_defaults(func=construct_nvme_bdev)



    @call_cmd

    def construct_rbd_bdev(args):

        print_array(rpc.bdev.construct_rbd_bdev(args.client, args))
@@ -133,6 +153,7 @@ if __name__ == "__main__": 
    p.add_argument('block_size', help='rbd block size', type=int)

    p.set_defaults(func=construct_rbd_bdev)



    @call_cmd

    def construct_error_bdev(args):

        rpc.bdev.construct_error_bdev(args.client, args)
@@ -141,6 +162,7 @@ if __name__ == "__main__": 
    p.add_argument('base_name', help='base bdev name')

    p.set_defaults(func=construct_error_bdev)



    @call_cmd

    def construct_pmem_bdev(args):

        print_array(rpc.bdev.construct_pmem_bdev(args.client, args))
@@ -149,6 +171,7 @@ if __name__ == "__main__": 
    p.add_argument('-n', '--name', help='Block device name', required=True)

    p.set_defaults(func=construct_pmem_bdev)



    @call_cmd

    def get_bdevs(args):

        print_dict(rpc.bdev.get_bdevs(args.client, args))
@@ -157,6 +180,7 @@ if __name__ == "__main__": 
    p.add_argument('-b', '--name', help="Name of the Blockdev. Example: Nvme0n1", required=False)

    p.set_defaults(func=get_bdevs)



    @call_cmd

    def get_bdevs_config(args):

        print_dict(rpc.bdev.get_bdevs_config(args.client, args))
@@ -165,6 +189,7 @@ if __name__ == "__main__": 
    p.add_argument('-b', '--name', help="Name of the Blockdev. Example: Nvme0n1", required=False)

    p.set_defaults(func=get_bdevs_config)



    @call_cmd

    def delete_bdev(args):

        rpc.bdev.delete_bdev(args.client, args)
@@ -173,6 +198,7 @@ if __name__ == "__main__": 
        'bdev_name', help='Blockdev name to be deleted. Example: Malloc0.')

    p.set_defaults(func=delete_bdev)



    @call_cmd

    def bdev_inject_error(args):

        rpc.bdev.bdev_inject_error(args.client, args)
@@ -184,6 +210,7 @@ if __name__ == "__main__": 
        '-n', '--num', help='the number of commands you want to fail', type=int, default=1)

    p.set_defaults(func=bdev_inject_error)



    @call_cmd

    def apply_firmware(args):

        print_dict(rpc.bdev.apply_firmware(args.client, args))
@@ -193,6 +220,7 @@ if __name__ == "__main__": 
    p.set_defaults(func=apply_firmware)



    # iSCSI

    @call_cmd

    def get_portal_groups(args):

        print_dict(rpc.iscsi.get_portal_groups(args.client, args))
@@ -200,6 +228,7 @@ if __name__ == "__main__": 
        'get_portal_groups', help='Display current portal group configuration')

    p.set_defaults(func=get_portal_groups)



    @call_cmd

    def get_initiator_groups(args):

        print_dict(rpc.iscsi.get_initiator_groups(args.client, args))
@@ -207,12 +236,14 @@ if __name__ == "__main__": 
                              help='Display current initiator group configuration')

    p.set_defaults(func=get_initiator_groups)



    @call_cmd

    def get_target_nodes(args):

        print_dict(rpc.iscsi.get_target_nodes(args.client, args))



    p = subparsers.add_parser('get_target_nodes', help='Display target nodes')

    p.set_defaults(func=get_target_nodes)



    @call_cmd

    def construct_target_node(args):

        rpc.iscsi.construct_target_node(args.client, args)
@@ -246,6 +277,7 @@ if __name__ == "__main__": 
                   help='Data Digest should be required for this target node.', action='store_true')

    p.set_defaults(func=construct_target_node)



    @call_cmd

    def target_node_add_lun(args):

        rpc.iscsi.target_node_add_lun(args.client, args)
@@ -257,6 +289,7 @@ if __name__ == "__main__": 
    *** If LUN ID is omitted or -1, the lowest free one is assigned ***""", type=int, required=False)

    p.set_defaults(func=target_node_add_lun)



    @call_cmd

    def add_pg_ig_maps(args):

        rpc.iscsi.add_pg_ig_maps(args.client, args)
@@ -269,6 +302,7 @@ if __name__ == "__main__": 
    *** The Portal/Initiator Groups must be precreated ***""")

    p.set_defaults(func=add_pg_ig_maps)



    @call_cmd

    def delete_pg_ig_maps(args):

        rpc.iscsi.delete_pg_ig_maps(args.client, args)
@@ -281,6 +315,7 @@ if __name__ == "__main__": 
    *** The Portal/Initiator Groups must be precreated ***""")

    p.set_defaults(func=delete_pg_ig_maps)



    @call_cmd

    def add_portal_group(args):

        rpc.iscsi.add_portal_group(args.client, args)
@@ -292,6 +327,7 @@ if __name__ == "__main__": 
    Example: '192.168.100.100:3260' '192.168.100.100:3261' '192.168.100.100:3262@0x1""")

    p.set_defaults(func=add_portal_group)



    @call_cmd

    def add_initiator_group(args):

        rpc.iscsi.add_initiator_group(args.client, args)
@@ -305,6 +341,7 @@ if __name__ == "__main__": 
    Example: '255.255.0.0 255.248.0.0' etc""")

    p.set_defaults(func=add_initiator_group)



    @call_cmd

    def add_initiators_to_initiator_group(args):

        rpc.iscsi.add_initiators_to_initiator_group(args.client, args)
@@ -318,6 +355,7 @@ if __name__ == "__main__": 
    This parameter can be omitted.  Example: '255.255.0.0 255.248.0.0' etc""", required=False)

    p.set_defaults(func=add_initiators_to_initiator_group)



    @call_cmd

    def delete_initiators_from_initiator_group(args):

        rpc.iscsi.delete_initiators_from_initiator_group(args.client, args)
@@ -331,6 +369,7 @@ if __name__ == "__main__": 
    This parameter can be omitted.  Example: '255.255.0.0 255.248.0.0' etc""", required=False)

    p.set_defaults(func=delete_initiators_from_initiator_group)



    @call_cmd

    def delete_target_node(args):

        rpc.iscsi.delete_target_node(args.client, args)
@@ -340,6 +379,7 @@ if __name__ == "__main__": 
                   help='Target node name to be deleted. Example: iqn.2016-06.io.spdk:disk1.')

    p.set_defaults(func=delete_target_node)



    @call_cmd

    def delete_portal_group(args):

        rpc.iscsi.delete_portal_group(args.client, args)
@@ -349,6 +389,7 @@ if __name__ == "__main__": 
        'tag', help='Portal group tag (unique, integer > 0)', type=int)

    p.set_defaults(func=delete_portal_group)



    @call_cmd

    def delete_initiator_group(args):

        rpc.iscsi.delete_initiator_group(args.client, args)
@@ -358,6 +399,7 @@ if __name__ == "__main__": 
        'tag', help='Initiator group tag (unique, integer > 0)', type=int)

    p.set_defaults(func=delete_initiator_group)



    @call_cmd

    def get_iscsi_connections(args):

        print_dict(rpc.iscsi.get_iscsi_connections(args.client, args))
@@ -365,12 +407,14 @@ if __name__ == "__main__": 
                              help='Display iSCSI connections')

    p.set_defaults(func=get_iscsi_connections)



    @call_cmd

    def get_iscsi_global_params(args):

        print_dict(rpc.iscsi.get_iscsi_global_params(args.client, args))



    p = subparsers.add_parser('get_iscsi_global_params', help='Display iSCSI global parameters')

    p.set_defaults(func=get_iscsi_global_params)



    @call_cmd

    def get_scsi_devices(args):

        print_dict(rpc.iscsi.get_scsi_devices(args.client, args))
@@ -378,6 +422,7 @@ if __name__ == "__main__": 
    p.set_defaults(func=get_scsi_devices)



    # log

    @call_cmd

    def set_trace_flag(args):

        rpc.log.set_trace_flag(args.client, args)
@@ -386,6 +431,7 @@ if __name__ == "__main__": 
        'flag', help='trace mask we want to set. (for example "debug").')

    p.set_defaults(func=set_trace_flag)



    @call_cmd

    def clear_trace_flag(args):

        rpc.log.clear_trace_flag(args.client, args)
@@ -394,12 +440,14 @@ if __name__ == "__main__": 
        'flag', help='trace mask we want to clear. (for example "debug").')

    p.set_defaults(func=clear_trace_flag)



    @call_cmd

    def get_trace_flags(args):

        print_dict(rpc.log.get_trace_flags(args.client, args))



    p = subparsers.add_parser('get_trace_flags', help='get trace flags')

    p.set_defaults(func=get_trace_flags)



    @call_cmd

    def set_log_level(args):

        rpc.log.set_log_level(args.client, args)
@@ -407,12 +455,14 @@ if __name__ == "__main__": 
    p.add_argument('level', help='log level we want to set. (for example "DEBUG").')

    p.set_defaults(func=set_log_level)



    @call_cmd

    def get_log_level(args):

        print_dict(rpc.log.get_log_level(args.client, args))



    p = subparsers.add_parser('get_log_level', help='get log level')

    p.set_defaults(func=get_log_level)



    @call_cmd

    def set_log_print_level(args):

        rpc.log.set_log_print_level(args.client, args)
@@ -420,6 +470,7 @@ if __name__ == "__main__": 
    p.add_argument('level', help='log print level we want to set. (for example "DEBUG").')

    p.set_defaults(func=set_log_print_level)



    @call_cmd

    def get_log_print_level(args):

        print_dict(rpc.log.get_log_print_level(args.client, args))
@@ -427,6 +478,7 @@ if __name__ == "__main__": 
    p.set_defaults(func=get_log_print_level)



    # lvol

    @call_cmd

    def construct_lvol_store(args):

        print_array(rpc.lvol.construct_lvol_store(args.client, args))
@@ -436,6 +488,7 @@ if __name__ == "__main__": 
    p.add_argument('-c', '--cluster-sz', help='size of cluster (in bytes)', type=int, required=False)

    p.set_defaults(func=construct_lvol_store)



    @call_cmd

    def rename_lvol_store(args):

        rpc.lvol.rename_lvol_store(args.client, args)
@@ -444,6 +497,7 @@ if __name__ == "__main__": 
    p.add_argument('new_name', help='new name')

    p.set_defaults(func=rename_lvol_store)



    @call_cmd

    def construct_lvol_bdev(args):

        print_array(rpc.lvol.construct_lvol_bdev(args.client, args))
@@ -455,6 +509,7 @@ if __name__ == "__main__": 
    p.add_argument('size', help='size in MiB for this bdev', type=int)

    p.set_defaults(func=construct_lvol_bdev)



    @call_cmd

    def rename_lvol_bdev(args):

        rpc.lvol.rename_lvol_bdev(args.client, args)
@@ -472,6 +527,7 @@ if __name__ == "__main__": 
    # p.add_argument('size', help='new size in MiB for this bdev', type=int)

    # p.set_defaults(func=resize_lvol_bdev)



    @call_cmd

    def destroy_lvol_store(args):

        rpc.lvol.destroy_lvol_store(args.client, args)
@@ -480,6 +536,7 @@ if __name__ == "__main__": 
    p.add_argument('-l', '--lvs_name', help='lvol store name', required=False)

    p.set_defaults(func=destroy_lvol_store)



    @call_cmd

    def get_lvol_stores(args):

        print_dict(rpc.lvol.get_lvol_stores(args.client, args))
@@ -489,6 +546,7 @@ if __name__ == "__main__": 
    p.set_defaults(func=get_lvol_stores)



    # nbd

    @call_cmd

    def start_nbd_disk(args):

        rpc.nbd.start_nbd_disk(args.client,

                               bdev_name=args.bdev_name,
@@ -499,6 +557,7 @@ if __name__ == "__main__": 
    p.add_argument('nbd_device', help='Nbd device name to be assigned. Example: /dev/nbd0.')

    p.set_defaults(func=start_nbd_disk)



    @call_cmd

    def stop_nbd_disk(args):

        rpc.nbd.stop_nbd_disk(args.client,

                              nbd_device=args.nbd_device)
@@ -507,6 +566,7 @@ if __name__ == "__main__": 
    p.add_argument('nbd_device', help='Nbd device name to be stopped. Example: /dev/nbd0.')

    p.set_defaults(func=stop_nbd_disk)



    @call_cmd

    def get_nbd_disks(args):

        print_dict(rpc.nbd.get_nbd_disks(args.client,

                                         nbd_device=args.nbd_device))
@@ -516,6 +576,7 @@ if __name__ == "__main__": 
    p.set_defaults(func=get_nbd_disks)



    # net

    @call_cmd

    def add_ip_address(args):

        rpc.net.add_ip_address(args.client, args)
@@ -524,6 +585,7 @@ if __name__ == "__main__": 
    p.add_argument('ip_addr', help='ip address will be added.')

    p.set_defaults(func=add_ip_address)



    @call_cmd

    def delete_ip_address(args):

        rpc.net.delete_ip_address(args.client, args)
@@ -532,6 +594,7 @@ if __name__ == "__main__": 
    p.add_argument('ip_addr', help='ip address will be deleted.')

    p.set_defaults(func=delete_ip_address)



    @call_cmd

    def get_interfaces(args):

        print_dict(rpc.net.get_interfaces(args.client, args))
@@ -540,6 +603,7 @@ if __name__ == "__main__": 
    p.set_defaults(func=get_interfaces)



    # NVMe-oF

    @call_cmd

    def get_nvmf_subsystems(args):

        print_dict(rpc.nvmf.get_nvmf_subsystems(args.client, args))
@@ -547,6 +611,7 @@ if __name__ == "__main__": 
                              help='Display nvmf subsystems')

    p.set_defaults(func=get_nvmf_subsystems)



    @call_cmd

    def construct_nvmf_subsystem(args):

        rpc.nvmf.construct_nvmf_subsystem(args.client, args)
@@ -570,6 +635,7 @@ if __name__ == "__main__": 
                   type=int, default=0)

    p.set_defaults(func=construct_nvmf_subsystem)



    @call_cmd

    def delete_nvmf_subsystem(args):

        rpc.nvmf.delete_nvmf_subsystem(args.client, args)
@@ -579,6 +645,7 @@ if __name__ == "__main__": 
                   help='subsystem nqn to be deleted. Example: nqn.2016-06.io.spdk:cnode1.')

    p.set_defaults(func=delete_nvmf_subsystem)



    @call_cmd

    def nvmf_subsystem_add_listener(args):

        rpc.nvmf.nvmf_subsystem_add_listener(args.client, args)
@@ -590,6 +657,7 @@ if __name__ == "__main__": 
    p.add_argument('-s', '--trsvcid', help='NVMe-oF transport service id: e.g., a port number')

    p.set_defaults(func=nvmf_subsystem_add_listener)



    @call_cmd

    def nvmf_subsystem_remove_listener(args):

        rpc.nvmf.nvmf_subsystem_remove_listener(args.client, args)
@@ -601,6 +669,7 @@ if __name__ == "__main__": 
    p.add_argument('-s', '--trsvcid', help='NVMe-oF transport service id: e.g., a port number')

    p.set_defaults(func=nvmf_subsystem_remove_listener)



    @call_cmd

    def nvmf_subsystem_add_ns(args):

        rpc.nvmf.nvmf_subsystem_add_ns(args.client, args)
@@ -612,6 +681,7 @@ if __name__ == "__main__": 
    p.add_argument('-e', '--eui64', help='Namespace EUI-64 identifier (optional)')

    p.set_defaults(func=nvmf_subsystem_add_ns)



    @call_cmd

    def nvmf_subsystem_remove_ns(args):

        rpc.nvmf.nvmf_subsystem_remove_ns(args.client, args)
@@ -620,6 +690,7 @@ if __name__ == "__main__": 
    p.add_argument('nsid', help='The requested NSID', type=int)

    p.set_defaults(func=nvmf_subsystem_remove_ns)



    @call_cmd

    def nvmf_subsystem_add_host(args):

        rpc.nvmf.nvmf_subsystem_add_host(args.client, args)
@@ -628,6 +699,7 @@ if __name__ == "__main__": 
    p.add_argument('host', help='Host NQN to allow')

    p.set_defaults(func=nvmf_subsystem_add_host)



    @call_cmd

    def nvmf_subsystem_remove_host(args):

        rpc.nvmf.nvmf_subsystem_remove_host(args.client, args)
@@ -636,6 +708,7 @@ if __name__ == "__main__": 
    p.add_argument('host', help='Host NQN to remove')

    p.set_defaults(func=nvmf_subsystem_remove_host)



    @call_cmd

    def nvmf_subsystem_allow_any_host(args):

        rpc.nvmf.nvmf_subsystem_allow_any_host(args.client, args)
@@ -646,6 +719,7 @@ if __name__ == "__main__": 
    p.set_defaults(func=nvmf_subsystem_allow_any_host)



    # pmem

    @call_cmd

    def create_pmem_pool(args):

        rpc.pmem.create_pmem_pool(args.client, args)
@@ -655,6 +729,7 @@ if __name__ == "__main__": 
    p.add_argument('block_size', help='Block size for this pmem pool', type=int)

    p.set_defaults(func=create_pmem_pool)



    @call_cmd

    def pmem_pool_info(args):

        print_dict(rpc.pmem.pmem_pool_info(args.client, args))
@@ -662,6 +737,7 @@ if __name__ == "__main__": 
    p.add_argument('pmem_file', help='Path to pmemblk pool file')

    p.set_defaults(func=pmem_pool_info)



    @call_cmd

    def delete_pmem_pool(args):

        rpc.pmem.delete_pmem_pool(args.client, args)
@@ -670,6 +746,7 @@ if __name__ == "__main__": 
    p.set_defaults(func=delete_pmem_pool)



    # subsystem

    @call_cmd

    def get_subsystems(args):

        print_dict(rpc.subsystem.get_subsystems(args.client))
@@ -677,6 +754,7 @@ if __name__ == "__main__": 
    entry contain (unsorted) array of subsystems it depends on.""")

    p.set_defaults(func=get_subsystems)



    @call_cmd

    def get_subsystem_config(args):

        print_dict(rpc.subsystem.get_subsystem_config(args.client, args.name))
@@ -685,6 +763,7 @@ if __name__ == "__main__": 
    p.set_defaults(func=get_subsystem_config)



    # vhost

    @call_cmd

    def set_vhost_controller_coalescing(args):

        rpc.vhost.set_vhost_controller_coalescing(args.client, args)
@@ -694,6 +773,7 @@ if __name__ == "__main__": 
    p.add_argument('iops_threshold', help='IOPS threshold when coalescing is enabled', type=int)

    p.set_defaults(func=set_vhost_controller_coalescing)



    @call_cmd

    def construct_vhost_scsi_controller(args):

        rpc.vhost.construct_vhost_scsi_controller(args.client, args)
@@ -703,6 +783,7 @@ if __name__ == "__main__": 
    p.add_argument('--cpumask', help='cpu mask for this controller')

    p.set_defaults(func=construct_vhost_scsi_controller)



    @call_cmd

    def add_vhost_scsi_lun(args):

        rpc.vhost.add_vhost_scsi_lun(args.client, args)
@@ -713,6 +794,7 @@ if __name__ == "__main__": 
    p.add_argument('bdev_name', help='bdev name')

    p.set_defaults(func=add_vhost_scsi_lun)



    @call_cmd

    def remove_vhost_scsi_target(args):

        rpc.vhost.remove_vhost_scsi_target(args.client, args)
@@ -721,6 +803,7 @@ if __name__ == "__main__": 
    p.add_argument('scsi_target_num', help='scsi_target_num', type=int)

    p.set_defaults(func=remove_vhost_scsi_target)



    @call_cmd

    def construct_vhost_blk_controller(args):

        rpc.vhost.construct_vhost_blk_controller(args.client, args)
@@ -731,12 +814,14 @@ if __name__ == "__main__": 
    p.add_argument("-r", "--readonly", action='store_true', help='Set controller as read-only')

    p.set_defaults(func=construct_vhost_blk_controller)



    @call_cmd

    def get_vhost_controllers(args):

        print_dict(rpc.vhost.get_vhost_controllers(args.client, args))



    p = subparsers.add_parser('get_vhost_controllers', help='List vhost controllers')

    p.set_defaults(func=get_vhost_controllers)



    @call_cmd

    def remove_vhost_controller(args):

        rpc.vhost.remove_vhost_controller(args.client, args)
@@ -744,6 +829,7 @@ if __name__ == "__main__": 
    p.add_argument('ctrlr', help='controller name')

    p.set_defaults(func=remove_vhost_controller)



    @call_cmd

    def construct_virtio_user_scsi_bdev(args):

        print_dict(rpc.vhost.construct_virtio_user_scsi_bdev(args.client, args))
@@ -757,6 +843,7 @@ if __name__ == "__main__": 
    p.add_argument('--vq-size', help='Size of each queue', type=int)

    p.set_defaults(func=construct_virtio_user_scsi_bdev)



    @call_cmd

    def construct_virtio_pci_scsi_bdev(args):

        print_dict(rpc.vhost.construct_virtio_pci_scsi_bdev(args.client, args))
@@ -768,12 +855,14 @@ if __name__ == "__main__": 
    It will be inhereted by all created bdevs, which are named n the following format: <name>t<target_id>""")

    p.set_defaults(func=construct_virtio_pci_scsi_bdev)



    @call_cmd

    def get_virtio_scsi_devs(args):

        print_dict(rpc.vhost.get_virtio_scsi_devs(args.client, args))



    p = subparsers.add_parser('get_virtio_scsi_devs', help='List all Virtio-SCSI devices.')

    p.set_defaults(func=get_virtio_scsi_devs)



    @call_cmd

    def remove_virtio_scsi_bdev(args):

        rpc.vhost.remove_virtio_scsi_bdev(args.client, args)
@@ -782,6 +871,7 @@ if __name__ == "__main__": 
    p.add_argument('name', help='Virtio device name. E.g. VirtioUser0')

    p.set_defaults(func=remove_virtio_scsi_bdev)



    @call_cmd

    def construct_virtio_user_blk_bdev(args):

        print_dict(rpc.vhost.construct_virtio_user_blk_bdev(args.client, args))
@@ -792,6 +882,7 @@ if __name__ == "__main__": 
    p.add_argument('--vq-size', help='Size of each queue', type=int)

    p.set_defaults(func=construct_virtio_user_blk_bdev)



    @call_cmd

    def construct_virtio_pci_blk_bdev(args):

        print_dict(rpc.vhost.construct_virtio_pci_blk_bdev(args.client, args))

scripts/rpc/client.py

+16 −11
Original line number Diff line number Diff line
@@ -7,6 +7,11 @@ def print_dict(d): 
    print(json.dumps(d, indent=2))





class JSONRPCException(Exception):

    def __init__(self, message):

        self.message = message





class JSONRPCClient(object):

    def __init__(self, addr, port=None, verbose=False, timeout=60.0):

        self.verbose = verbose
@@ -68,21 +73,21 @@ class JSONRPCClient(object): 


        if not response:

            if method == "kill_instance":

                exit(0)

                return {}

            if closed:

                print("Connection closed with partial response:")

                msg = "Connection closed with partial response:"

            else:

                print("Timeout while waiting for response:")

            print(buf)

            exit(1)

                msg = "Timeout while waiting for response:"

            msg = "\n".join([msg, buf])

            raise JSONRPCException(msg)



        if 'error' in response:

            print("Got JSON-RPC error response")

            print("request:")

            print_dict(json.loads(reqstr))

            print("response:")

            print_dict(response['error'])

            exit(1)

            msg = "\n".join(["Got JSON-RPC error response",

                             "request:",

                             json.dumps(req, indent=2),

                             "response:",

                             json.dumps(response['error'], indent=2)])

            raise JSONRPCException(msg)



        if verbose:

            print("response:")