diff --git a/bots/markpages.py b/bots/markpages.py index d20951f..b08776c 100644 --- a/bots/markpages.py +++ b/bots/markpages.py @@ -84,9 +84,14 @@ class MarkPagesBot( CurrentPageBot ): # sets 'current_page' on each treat() def build_generator( self ): """ - Builds generator + Builds generator to pass to super class """ - self.gen = self.genFactory.getCombinedGenerator() + # Add Talkpages to work on to generatorFactory + self.genFactory.gens.append( self.redfam_talkpages_generator() ) + + # Set generator to pass to super class + self.gen = pagegenerators.PreloadingGenerator( + self.genFactory.getCombinedGenerator() ) def redfam_talkpages_generator( self ): """