flexget.plugins.filter_imdb_required
Covered: 13 lines
Missed: 9 lines
Skipped 9 lines
Percent: 59 %
 1
import logging
 2
from flexget.plugin import register_plugin, priority, get_plugin_by_name, PluginError
 4
log = logging.getLogger('imdb_required')
 7
class FilterImdbRequired(object):
 8
    """
 9
        Rejects entries without imdb url.
11
        Example:
13
        imdb_required: yes
14
    """
16
    def validator(self):
17
        from flexget import validator
18
        return validator.factory('boolean')
20
    @priority(32)
21
    def on_feed_filter(self, feed):
22
        for entry in feed.entries:
23
            try:
24
                get_plugin_by_name('imdb_lookup').instance.lookup(feed, entry)
25
            except PluginError:
26
                feed.reject(entry, 'imdb required')
27
            if not 'imdb_url' in entry:
28
                feed.reject(entry, 'imdb required')
30
register_plugin(FilterImdbRequired, 'imdb_required')