desktop_background.py

#!/usr/bin/env python

import logging
import importlib
from managers import bgmanager

logging.basicConfig()

logger = logging.getLogger('desktop_background:main')
logger.setLevel(logging.INFO)

__version__ = '1-flikr'


def get_plugin(name='earth'):
    return importlib.import_module('plugins.%s' % name)

plugin = get_plugin()


def _parse_arguments():
    import argparse
    parser = argparse.ArgumentParser()
    parser.add_argument('-s', '--source',
                        nargs='?', default='earth',
                        type=str, help='flikr earth')
    parser.add_argument('-v', '--verbose', action='count', default=0)

    args = parser.parse_args()
    log_level = logging.WARN - 10*args.verbose
    if log_level < 0:
        log_level = logging.DEBUG
    logger.setLevel(log_level)
    return args.source


def set_background():
    bgmanager.set_background(plugin.get_image())


if __name__ == '__main__':
    plugin_name = _parse_arguments() or 'earth'
    try:
        logger.info(
            "changing desktop background using '%s' plugin" % plugin_name)
        plugin = get_plugin(plugin_name)
        logger.debug("test debug")
    except ImportError as ex:
        logger.warning(
            "No '%s' plugin found, fallback to default: 'earth'" % plugin_name)
    set_background()