Subversion Repositories SmartDukaan

'''

Created on 23-Feb-2010

@author: ashish

@author: rajveer

'''

import os

import sys

from getopt import getopt, GetoptError

from shop2020.config.impl.ConfigServerHandler import ConfigServerHandler

from thrift.protocol.TBinaryProtocol import TBinaryProtocolFactory

from shop2020.thriftpy.config import Configuration

from thrift.transport import TSocket, TTransport

from thrift.server import TServer

def main(argv):

    try:

        opts, args = getopt(argv,"p:f:e", ["port=", "file=", "env="])

    except GetoptError:

        usage()

        sys.exit(2)

    try:

        environment = os.environ["ENV_TYPE"]

    except KeyError:

        environment = "DEV"

    cfg_file_name = '/tmp/shop2020.cfg'

    port = 9999

    for opt,arg in opts:

        if opt in ('-p', "--port"):

            port = arg;

        elif opt in ('-f', "--file"):

            cfg_file_name = arg

        elif opt in ('-e', "--env"):

            environment = arg

    start_server(cfg_file_name, port, environment)

def start_server(filename, port, environment):

    handler = ConfigServerHandler(filename, environment)

    processor = Configuration.Processor(handler)

    transport = TSocket.TServerSocket(port)

    tfactory = TTransport.TFramedTransportFactory()

    pfactory = TBinaryProtocolFactory()

    server = TServer.TThreadedServer(processor, transport, tfactory, pfactory)

    print "Starting config server with config params, port "+ str(port)+" file "+ filename

    server.serve()

    print "Server functioning"

def usage():

    print 'Usage: ConfigServer.py -p <port_to_run> -f <config_file_path> -e <environment> or ConfigServer.py --port=<port_to_run> --file=<config_file_path> --env=<environment>' 

if __name__ == '__main__':

    main(sys.argv[1:])