diff --git a/bots/reddiscparser.py b/bots/reddiscparser.py index 2a47642..818eb05 100644 --- a/bots/reddiscparser.py +++ b/bots/reddiscparser.py @@ -49,16 +49,27 @@ class DiscussionParserBot( onlyinclude_re = re.compile( jogobot.config["redundances"]["reddiscs_onlyinclude_re"] ) - def __init__( self, generator ): + def __init__( self, genFactory, **kwargs ): """ Constructor Parameters: - @param generator: The page generator that determines on which pages - to work. - @type generator: generator. + @param genFactory GenFactory with parsed pagegenerator args to + build generator + @type genFactory pagegenerators.GeneratorFactory + @param **kwargs Additional args + @type iterable """ + # Copy needed args + self.genFactory = genFactory + + # Build generator with genFactory + self.build_generator() + + # Run super class init with builded generator + super( DiscussionParserBot, self ).__init__(generator=self.gen) + def build_generator(self): """ Builds generator to work on, based on self.genFactory