wallpapers.py

import os
import logging

from plugins import plugin_manager

class Wallpapers:

    def __init__(self, config):
        self.config = config

        self.wallpaper_plugins = plugin_manager['Source']
        self.plugin_index = 0

    def pop(self, count=1):
        if self.plugin_index >= len(self.wallpaper_plugins):
            self.plugin_index = 0

        wallpaper_plugin = self.wallpaper_plugins[self.plugin_index]

        self.plugin_index += 1

        wp = [w for w in wallpaper_plugin.pop(count)]

        if len(wp) == 0:
            raise RuntimeError("No wallpapers found, check configuration")

        return wp

    def count(self):
        return sum([q.count() for q in self.wallpaper_plugins])

    def wallpaper_complete(self):
        """ Calls methods of the same name in each wallpaper source plugin """
        for wpp in self.wallpaper_plugins:
            wpp.wallpaper_complete()