etwist/service.py
author Aurelien Campeas <aurelien.campeas@logilab.fr>
Wed, 21 Apr 2010 14:08:18 +0200
branchstable
changeset 5365 ca838c79af97
parent 5088 891cdb7d8cbb
child 5414 6eeec78c65f9
permissions -rw-r--r--
[service] do not bluntly override these things if they exist

import os

import win32serviceutil
import win32service
import win32event

from cubicweb.etwist.server import (CubicWebRootResource, reactor, server,
                                    parsePOSTData, channel)

from logging import getLogger, handlers
from cubicweb import set_log_methods
from cubicweb.cwconfig import CubicWebConfiguration as cwcfg

logger = getLogger('cubicweb.twisted')
logger.handlers = [handlers.NTEventLogHandler('cubicweb')]

if not os.environ.get('CW_INSTANCES_DIR'):
    os.environ['CW_INSTANCES_DIR'] = r'C:\etc\cubicweb.d'
if not os.environ.get('USERNAME'):
    os.environ['USERNAME'] = 'cubicweb'

class CWService(object, win32serviceutil.ServiceFramework):
    _svc_name_ = None
    _svc_display_name_ = None
    instance = None

    def __init__(self, *args, **kwargs):
        win32serviceutil.ServiceFramework.__init__(self, *args, **kwargs)
        cwcfg.load_cwctl_plugins()
        set_log_methods(CubicWebRootResource, logger)
        server.parsePOSTData = parsePOSTData

    def SvcStop(self):
        self.ReportServiceStatus(win32service.SERVICE_STOP_PENDING)
        logger.info('stopping %s service' % self.instance)
        reactor.stop()
        self.ReportServiceStatus(win32service.SERVICE_STOP_PENDING)

    def SvcDoRun(self):
        self.ReportServiceStatus(win32service.SERVICE_START_PENDING)
        logger = getLogger('cubicweb.twisted')
        logger.info('starting %s service' % self.instance)
        try:
            # create the site
            config = cwcfg.config_for(self.instance)
            root_resource = CubicWebRootResource(config, False)
            website = server.Site(root_resource)
            # serve it via standard HTTP on port set in the configuration
            port = config['port'] or 8080
            logger.info('listening on port %s' % port)
            reactor.listenTCP(port, channel.HTTPFactory(website))
            root_resource.init_publisher()
            root_resource.start_service()
            logger.info('instance started on %s', root_resource.base_url)
            self.ReportServiceStatus(win32service.SERVICE_RUNNING)
            reactor.run()
        except Exception, e:
            logger.error('service %s stopped (cause: %s)' % (self.instance, e))
            logger.exception('what happened ...')
        self.ReportServiceStatus(win32service.SERVICE_STOPPED)