+
−# -*- coding: UTF-8 -*-
+
−# Copyright 2007 - 2010, Pascal Volk
+
−# See COPYING for distribution information.
+
−"""
+
−    VirtualMailManager.cli.main
+
−    ~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
−
+
−    VirtualMailManager's command line interface.
+
−"""
+
−
+
−from VirtualMailManager import ENCODING, errors
+
−from VirtualMailManager.config import ConfigValueError
+
−from VirtualMailManager.cli import w_err
+
−from VirtualMailManager.cli.handler import CliHandler
+
−from VirtualMailManager.constants import DATABASE_ERROR, EX_MISSING_ARGS, \
+
−     EX_SUCCESS, EX_UNKNOWN_COMMAND, EX_USER_INTERRUPT
+
−from VirtualMailManager.cli.subcommands import RunContext, cmd_map, usage
+
−
+
−
+
−_ = lambda msg: msg
+
−
+
−def _get_handler():
+
−    """Try to get a CliHandler. Exit the program when an error occurs."""
+
−    try:
+
−        handler = CliHandler()
+
−        handler.cfg_install()
+
−    except (errors.NotRootError, errors.PermissionError, errors.VMMError,
+
−            errors.ConfigError, ConfigValueError), err:
+
−        w_err(err.code, _(u'Error: %s') % err.msg)
+
−    else:
+
−        return handler
+
−
+
−
+
−def run(argv):
+
−    if len(argv) < 2:
+
−        usage(EX_MISSING_ARGS, _(u"You must specify a subcommand at least"))
+
−
+
−    sub_cmd = argv[1].lower()
+
−    if sub_cmd in cmd_map:
+
−        cmd_func = cmd_map[sub_cmd].func
+
−    else:
+
−        for cmd in cmd_map.itervalues():
+
−            if cmd.alias == sub_cmd:
+
−                cmd_func = cmd.func
+
−                sub_cmd = cmd.name
+
−                break
+
−        else:
+
−            usage(EX_UNKNOWN_COMMAND, _(u"Unknown subcommand: '%s'") % sub_cmd)
+
−
+
−    handler = _get_handler()
+
−    run_ctx = RunContext(argv, handler, sub_cmd)
+
−    try:
+
−        cmd_func(run_ctx)
+
−    except (EOFError, KeyboardInterrupt):
+
−        # TP: We have to cry, because root has killed/interrupted vmm
+
−        # with Ctrl+C or Ctrl+D.
+
−        w_err(EX_USER_INTERRUPT, '', _(u'Ouch!'), '')
+
−    except errors.VMMError, err:
+
−        if err.code != DATABASE_ERROR:
+
−            w_err(err.code, _(u'Error: %s') % err.msg)
+
−        w_err(err.code, unicode(err.msg, ENCODING, 'replace'))
+
−    if handler.has_warnings():
+
−        w_err(0, _(u'Warnings:'), *handler.get_warnings())
+
−    return EX_SUCCESS
+
−
+
−del _