Coverage for deluge/ui/ui : 20%

# -*- coding: utf-8 -*- 

# 

# Copyright (C) 2007 Andrew Resch <andrewresch@gmail.com> 

# 

# This file is part of Deluge and is licensed under GNU General Public License 3.0, or later, with 

# the additional special exception to link portions of this program with the OpenSSL library. 

# See LICENSE for more details. 

# 

from __future__ import print_function 

import logging 

import os 

import sys 

from optparse import OptionGroup, OptionParser 

import deluge.common 

import deluge.configmanager 

import deluge.log 

try: 

    from setproctitle import setproctitle 

except ImportError: 

exit    setproctitle = lambda t: None 

def version_callback(option, opt_str, value, parser): 

    print(os.path.basename(sys.argv[0]) + ": " + deluge.common.get_version()) 

    try: 

        from deluge._libtorrent import lt 

        print("libtorrent: %s" % lt.version) 

    except ImportError: 

        pass 

    raise SystemExit 

DEFAULT_PREFS = { 

    "default_ui": "gtk" 

} 

41if 'dev' not in deluge.common.get_version(): 

    import warnings 

    warnings.filterwarnings('ignore', category=DeprecationWarning, module='twisted') 

class _UI(object): 

    def __init__(self, name="gtk"): 

        self.__name = name 

        self.__parser = OptionParser(usage="%prog [options] [actions]") 

        self.__parser.add_option("-v", "--version", action="callback", callback=version_callback, 

                                 help="Show program's version number and exit") 

        group = OptionGroup(self.__parser, "Common Options") 

        group.add_option("-c", "--config", dest="config", help="Set the config folder location") 

        group.add_option("-l", "--logfile", dest="logfile", help="Output to designated logfile instead of stdout") 

        group.add_option("-L", "--loglevel", dest="loglevel", 

                         help="Set the log level: none, info, warning, error, critical, debug") 

        group.add_option("-q", "--quiet", dest="quiet", action="store_true", default=False, 

                         help="Sets the log level to 'none', this is the same as `-L none`") 

        group.add_option("-r", "--rotate-logs", help="Rotate logfiles.", action="store_true", default=False) 

        self.__parser.add_option_group(group) 

    @property 

    def name(self): 

        return self.__name 

    @property 

    def parser(self): 

        return self.__parser 

    @property 

    def options(self): 

        return self.__options 

    @property 

    def args(self): 

        return self.__args 

    def start(self): 

        # Make sure all arguments are unicode 

        argv = deluge.common.unicode_argv()[1:] 

        (self.__options, self.__args) = self.__parser.parse_args(argv) 

        if self.__options.quiet: 

            self.__options.loglevel = "none" 

        logfile_mode = 'w' 

        if self.__options.rotate_logs: 

            logfile_mode = 'a' 

        if self.__options.loglevel: 

            self.__options.loglevel = self.__options.loglevel.lower() 

        # Setup the logger 

        deluge.log.setup_logger(level=self.__options.loglevel, 

                                filename=self.__options.logfile, 

                                filemode=logfile_mode) 

        log = logging.getLogger(__name__) 

        if self.__options.config: 

            if not deluge.configmanager.set_config_dir(self.__options.config): 

                log.error("There was an error setting the config dir! Exiting..") 

                sys.exit(1) 

        # Setup gettext 

        deluge.common.setup_translations() 

        setproctitle("deluge-%s" % self.__name) 

        log.info("Deluge ui %s", deluge.common.get_version()) 

        log.debug("options: %s", self.__options) 

        log.debug("args: %s", self.__args) 

        log.info("Starting %s ui..", self.__name) 

class UI: 

    def __init__(self, options, args, ui_args): 

        import logging 

        log = logging.getLogger(__name__) 

        log.debug("UI init..") 

        # Setup gettext 

        deluge.common.setup_translations() 

        # Set the config directory 

        deluge.configmanager.set_config_dir(options.config) 

        config = deluge.configmanager.ConfigManager("ui.conf", DEFAULT_PREFS) 

        if not options.ui: 

            selected_ui = config["default_ui"] 

        else: 

            selected_ui = options.ui 

        setproctitle("deluge") 

        config.save() 

        del config 

        try: 

            if selected_ui == "gtk": 

                log.info("Starting GtkUI..") 

                from deluge.ui.gtkui.gtkui import GtkUI 

                GtkUI(args) 

            elif selected_ui == "web": 

                log.info("Starting WebUI..") 

                from deluge.ui.web.web import WebUI 

                WebUI(args) 

            elif selected_ui == "console": 

                log.info("Starting ConsoleUI..") 

                from deluge.ui.console.main import ConsoleUI 

                ConsoleUI(ui_args) 

        except ImportError as ex: 

            import sys 

            import traceback 

            error_type, error_value, tb = sys.exc_info() 

            stack = traceback.extract_tb(tb) 

            last_frame = stack[-1] 

            if last_frame[0] == __file__: 

                log.error("Unable to find the requested UI: %s.  Please select a different UI with the '-u' option \ 

                          or alternatively use the '-s' option to select a different default UI.", selected_ui) 

            else: 

                log.exception(ex) 

                log.error("There was an error whilst launching the request UI: %s", selected_ui) 

                log.error("Look at the traceback above for more information.") 

            sys.exit(1)