From c4d8a95672e83d0b1b46551505a5238196eefef9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?GOLDERWEB=20=E2=80=93=20Jonathan=20Golder?= Date: Sun, 28 Aug 2016 18:13:27 +0200 Subject: [PATCH] Implement build_generator-method Build_generator will add the redfam_talkpages_generator to the genFactory, build a generator of the genFactory and sets self.gen which is used as generator for run() Related Task: [https://fs.golderweb.de/index.php?do=details&task_id=88 FS#88] --- bots/markpages.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) 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 ): """