from cubicweb.cwconfig import CubicWebConfiguration as cwcfg

from cubicweb.cwctl import CWCTL, InstanceCommand, init_cmdline_log_threshold

class PyramidStartHandler(InstanceCommand):

    """Start an interactive pyramid server.

    This command requires http://hg.logilab.org/review/pyramid_cubicweb/

    <instance>

      identifier of the instance to configure.

    """

    name = 'pyramid'

    options = (

        ("debug",

         {'short': 'D', 'action': 'store_true',

          'help': 'start server in debug mode.'}),

        ('loglevel',

         {'short': 'l', 'type': 'choice', 'metavar': '<log level>',

          'default': None, 'choices': ('debug', 'info', 'warning', 'error'),

          'help': 'debug if -D is set, error otherwise',

          }),

    )

    def pyramid_instance(self, appid):

        from pyramid_cubicweb import make_cubicweb_application

        from waitress import serve

        cwconfig = cwcfg.config_for(appid, debugmode=self['debug'])

        init_cmdline_log_threshold(cwconfig, self['loglevel'])

        host = cwconfig['interface']

        port = cwconfig['port'] or 8080

        pyramid_config = make_cubicweb_application(cwconfig)

        repo = cwconfig.repository()

        try:

            repo.start_looping_tasks()

            serve(pyramid_config.make_wsgi_app(), host=host, port=port)

        finally:

            repo.shutdown()

CWCTL.register(PyramidStartHandler)