Subversion Repositories SmartDukaan

#!/usr/bin/env python

'''
Created on 15-Jul-2010

@author: ashish
@author: Chandranshu
'''
import os
import sys
import optparse

if __name__ == '__main__' and __package__ is None:
    sys.path.insert(0, os.getcwd())

from thrift.transport import TSocket, TTransport
from thrift.protocol.TBinaryProtocol import TBinaryProtocolFactory
from thrift.server import TServer

from shop2020.utils.daemon import Daemon
from shop2020.config.client.ConfigClient import ConfigClient
from shop2020.helpers.impl.HelperServiceHandler import HelperServiceHandler
from shop2020.thriftpy.utils import HelperService

class HelperServer(Daemon):
    
    def __init__(self, logfile='/var/log/services/helper.log', pidfile='/tmp/helper-server.pid'):
        Daemon.__init__(self, pidfile, stdout=logfile, stderr=logfile)
        
    def run(self):
        #get the config client
        config_client = ConfigClient()
        host_name = config_client.get_property('helper_service_server_host')
        port = int(config_client.get_property('helper_service_server_port'))
        dbname = config_client.get_property('helper_service_dbname')
        db_hostname = config_client.get_property('helper_service_db_hostname')
        
        handler = HelperServiceHandler(dbname, db_hostname)
        processor = HelperService.Processor(handler)
        transport = TSocket.TServerSocket(port=port)
        tfactory = TTransport.TFramedTransportFactory()
        pfactory = TBinaryProtocolFactory()
        server = TServer.TThreadedServer(processor, transport, tfactory, pfactory)
        print "Starting HelperService at, port " + str(port) + " host " + host_name
        sys.stdout.flush()
        server.serve()
        print "Server functioning"

if __name__ == "__main__":
    parser = optparse.OptionParser()
    parser.add_option("-l", "--logfile", dest="logfile",
                      type="string",
                      help="Log all output to LOG_FILE",
                      )
    parser.add_option("-i", "--pidfile", dest="pidfile",
                      type="string",
                      help="Write the PID to pidfile")
    (options, args) = parser.parse_args()
    daemon = HelperServer(options.logfile, options.pidfile)
    if len(args) == 0:
        daemon.run()
    elif len(args) == 1:
        if 'start' == args[0]:
            daemon.start()
        elif 'stop' == args[0]:
            daemon.stop()
        elif 'restart' == args[0]:
            daemon.restart()
        else:
            print "Unknown command"
            sys.exit(2)
        sys.exit(0)
    else:
        print "usage: %s start|stop|restart" % sys.argv[0]
        sys.exit(2)