summaryrefslogtreecommitdiffstats
path: root/domino-cli.py
blob: d9245b35f8b028f9af116e613190c820dde6ab34 (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
#!/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

sys.path.insert(0, glob.glob('./lib')[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(argv, cli_port):
  #cli_port = DOMINO_CLI_PORT

  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 = argv
    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__":
   if len(sys.argv) >= 2:
     main(sys.argv[2:], sys.argv[1])
   else:
     print 'domino-cli.py <cliport> ...'
     sys.exit(2)