tools/test/tofino-port-auto-setup-tool/thrift/gen-py/pltfm_pm_rpc/pltfm_pm_rpc-remote

#!/usr/bin/env python
#
# Autogenerated by Thrift Compiler (0.10.0)
#
# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
#
#  options string: py
#

import sys
import pprint
if sys.version_info[0] > 2:
    from urllib.parse import urlparse
else:
    from urlparse import urlparse
from thrift.transport import TTransport, TSocket, TSSLSocket, THttpClient
from thrift.protocol.TBinaryProtocol import TBinaryProtocol

from pltfm_pm_rpc import pltfm_pm_rpc
from pltfm_pm_rpc.ttypes import *

if len(sys.argv) <= 1 or sys.argv[1] == '--help':
    print('')
    print('Usage: ' + sys.argv[0] + ' [-h host[:port]] [-u url] [-f[ramed]] [-s[sl]] [-novalidate] [-ca_certs certs] [-keyfile keyfile] [-certfile certfile] function [arg1 [arg2...]]')
    print('')
    print('Functions:')
    print('  pltfm_pm_status_t pltfm_pm_port_add(pltfm_pm_device_t device, pltfm_pm_dev_port_t dev_port, pltfm_pm_port_speed_t ps, pltfm_pm_fec_type_t fec)')
    print('  pltfm_pm_status_t pltfm_pm_port_del(pltfm_pm_device_t device, pltfm_pm_dev_port_t dev_port)')
    print('  pltfm_pm_status_t pltfm_pm_port_enable(pltfm_pm_device_t device, pltfm_pm_dev_port_t dev_port)')
    print('  pltfm_pm_status_t pltfm_pm_port_dis(pltfm_pm_device_t device, pltfm_pm_dev_port_t dev_port)')
    print('  pltfm_pm_status_t pltfm_pm_switchd_port_cleanup(pltfm_pm_device_t device)')
    print('  pltfm_pm_oper_status_t pltfm_pm_port_oper_status_get(pltfm_pm_device_t device, pltfm_pm_dev_port_t dev_port)')
    print('  pltfm_pm_board_type_t pltfm_pm_board_type_get()')
    print('  pltfm_pm_status_t pltfm_pm_port_an_set(pltfm_pm_device_t device, pltfm_pm_dev_port_t dev_port, i32 an_flag)')
    print('  pltfm_pm_status_t pltfm_pm_serdes_lane_map_set(pltfm_pm_device_t device)')
    print('  pltfm_pm_status_t pltfm_pm_serdes_init(pltfm_pm_device_t device)')
    print('')
    sys.exit(0)

pp = pprint.PrettyPrinter(indent=2)
host = 'localhost'
port = 9090
uri = ''
framed = False
ssl = False
validate = True
ca_certs = None
keyfile = None
certfile = None
http = False
argi = 1

if sys.argv[argi] == '-h':
    parts = sys.argv[argi + 1].split(':')
    host = parts[0]
    if len(parts) > 1:
        port = int(parts[1])
    argi += 2

if sys.argv[argi] == '-u':
    url = urlparse(sys.argv[argi + 1])
    parts = url[1].split(':')
    host = parts[0]
    if len(parts) > 1:
        port = int(parts[1])
    else:
        port = 80
    uri = url[2]
    if url[4]:
        uri += '?%s' % url[4]
    http = True
    argi += 2

if sys.argv[argi] == '-f' or sys.argv[argi] == '-framed':
    framed = True
    argi += 1

if sys.argv[argi] == '-s' or sys.argv[argi] == '-ssl':
    ssl = True
    argi += 1

if sys.argv[argi] == '-novalidate':
    validate = False
    argi += 1

if sys.argv[argi] == '-ca_certs':
    ca_certs = sys.argv[argi+1]
    argi += 2

if sys.argv[argi] == '-keyfile':
    keyfile = sys.argv[argi+1]
    argi += 2

if sys.argv[argi] == '-certfile':
    certfile = sys.argv[argi+1]
    argi += 2

cmd = sys.argv[argi]
args = sys.argv[argi + 1:]

if http:
    transport = THttpClient.THttpClient(host, port, uri)
else:
    if ssl:
        socket = TSSLSocket.TSSLSocket(host, port, validate=validate, ca_certs=ca_certs, keyfile=keyfile, certfile=certfile)
    else:
        socket = TSocket.TSocket(host, port)
    if framed:
        transport = TTransport.TFramedTransport(socket)
    else:
        transport = TTransport.TBufferedTransport(socket)
protocol = TBinaryProtocol(transport)
client = pltfm_pm_rpc.Client(protocol)
transport.open()

if cmd == 'pltfm_pm_port_add':
    if len(args) != 4:
        print('pltfm_pm_port_add requires 4 args')
        sys.exit(1)
    pp.pprint(client.pltfm_pm_port_add(eval(args[0]), eval(args[1]), eval(args[2]), eval(args[3]),))

elif cmd == 'pltfm_pm_port_del':
    if len(args) != 2:
        print('pltfm_pm_port_del requires 2 args')
        sys.exit(1)
    pp.pprint(client.pltfm_pm_port_del(eval(args[0]), eval(args[1]),))

elif cmd == 'pltfm_pm_port_enable':
    if len(args) != 2:
        print('pltfm_pm_port_enable requires 2 args')
        sys.exit(1)
    pp.pprint(client.pltfm_pm_port_enable(eval(args[0]), eval(args[1]),))

elif cmd == 'pltfm_pm_port_dis':
    if len(args) != 2:
        print('pltfm_pm_port_dis requires 2 args')
        sys.exit(1)
    pp.pprint(client.pltfm_pm_port_dis(eval(args[0]), eval(args[1]),))

elif cmd == 'pltfm_pm_switchd_port_cleanup':
    if len(args) != 1:
        print('pltfm_pm_switchd_port_cleanup requires 1 args')
        sys.exit(1)
    pp.pprint(client.pltfm_pm_switchd_port_cleanup(eval(args[0]),))

elif cmd == 'pltfm_pm_port_oper_status_get':
    if len(args) != 2:
        print('pltfm_pm_port_oper_status_get requires 2 args')
        sys.exit(1)
    pp.pprint(client.pltfm_pm_port_oper_status_get(eval(args[0]), eval(args[1]),))

elif cmd == 'pltfm_pm_board_type_get':
    if len(args) != 0:
        print('pltfm_pm_board_type_get requires 0 args')
        sys.exit(1)
    pp.pprint(client.pltfm_pm_board_type_get())

elif cmd == 'pltfm_pm_port_an_set':
    if len(args) != 3:
        print('pltfm_pm_port_an_set requires 3 args')
        sys.exit(1)
    pp.pprint(client.pltfm_pm_port_an_set(eval(args[0]), eval(args[1]), eval(args[2]),))

elif cmd == 'pltfm_pm_serdes_lane_map_set':
    if len(args) != 1:
        print('pltfm_pm_serdes_lane_map_set requires 1 args')
        sys.exit(1)
    pp.pprint(client.pltfm_pm_serdes_lane_map_set(eval(args[0]),))

elif cmd == 'pltfm_pm_serdes_init':
    if len(args) != 1:
        print('pltfm_pm_serdes_init requires 1 args')
        sys.exit(1)
    pp.pprint(client.pltfm_pm_serdes_init(eval(args[0]),))

else:
    print('Unrecognized method %s' % cmd)
    sys.exit(1)

transport.close()