flexget.plugins.operate.max_reruns
Covered: 14 lines
Missed: 7 lines
Skipped 11 lines
Percent: 66 %
 1
import logging

 2
from flexget import validator

 3
from flexget.feed import Feed

 4
from flexget.plugin import register_plugin

 6
log = logging.getLogger('max_reruns')

 9
class MaxReRuns(object):

10
    """Overrides the maximum amount of re-runs allowed by a feed."""

12
    def __init__(self):

13
        self.default = Feed.max_reruns

15
    def validator(self):

16
        root = validator.factory('integer')

17
        return root

19
    def on_feed_start(self, feed, config):

20
        self.default = feed.max_reruns

21
        feed.max_reruns = config

22
        log.debug('changing max feed rerun variable to: %s' % config)

24
    def on_feed_exit(self, feed, config):

25
        log.debug('restoring max feed rerun variable to: %s' % self.default)

26
        feed.max_reruns = self.default

28
    on_feed_abort = on_feed_exit

31
register_plugin(MaxReRuns, 'max_reruns', api_ver=2)