aboutsummaryrefslogtreecommitdiffstats
path: root/domino_cli.py
blob: e1bbfbb8ec7707de5f797fb82965f010d658f6aa (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
#!/usr/bin/env python

#Copyright 2016 Open Platform for NFV Project, Inc. and its contributors
#   Licensed under the Apache License, Version 2.0 (the "License");
#   you may not use this file except in compliance with the License.
#   You may obtain a copy of the License at
#       http://www.apache.org/licenses/LICENSE-2.0
#   Unless required by applicable law or agreed to in writing, software
#   distributed under the License is distributed on an "AS IS" BASIS,
#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#   See the License for the specific language governing permissions and
#   limitations under the License.

import sys, glob, getopt
import getopt
import pkg_resources

try:
    dir = pkg_resources.resource_filename('lib','.')
except ImportError:
    dir = './lib'
sys.path.insert(0, glob.glob(dir)[0])

from dominoCLI import DominoClientCLI
from dominoCLI.ttypes import *
from dominoCLI.constants import *

from thrift import Thrift
from thrift.transport import TSocket
from thrift.transport import TTransport
from thrift.protocol import TBinaryProtocol

#Load configuration parameters
from domino_conf import *

def main():
  #cli_port = DOMINO_CLI_PORT
  if len(sys.argv) >= 2:
    cli_port = sys.argv[1]
  else:
    print 'domino-cli.py <cliport> ...'
    return 2
  try:
    # Make socket
    # NOTE that domino-cli.py and DominoClient.py are assumed to be run in the same machine
    transport = TSocket.TSocket('localhost', int(cli_port))
    # Buffering is critical. Raw sockets are very slow
    transport = TTransport.TBufferedTransport(transport)
    # Wrap in a protocol
    protocol = TBinaryProtocol.TBinaryProtocol(transport)

    # Create a client to use the protocol encoder
    client = DominoClientCLI.Client(protocol)

    # Connect!
    transport.open()

    CLImsg = CLIMessage()
    CLImsg.CLI_input = sys.argv[2:]
    CLIrespmsg = client.d_CLI(CLImsg)
    if CLIrespmsg.CLI_response is not None:
      print CLIrespmsg.CLI_response
  except Thrift.TException, tx:
    print '%s' % (tx.message)

if __name__ == "__main__":
  sys.exit(main())