diff --git a/bots/markpages.py b/bots/markpages.py index ca9d586..d20951f 100644 --- a/bots/markpages.py +++ b/bots/markpages.py @@ -88,6 +88,26 @@ class MarkPagesBot( CurrentPageBot ): # sets 'current_page' on each treat() """ self.gen = self.genFactory.getCombinedGenerator() + def redfam_talkpages_generator( self ): + """ + Wrappers the redfam.article_generator and + passes it to pagegenerators.PageWithTalkPageGenerator(). + Then it iterates over the generator and adds a reference to the + related redfam to each talkpage-object. + """ + + for redfam in self.redfams: + + # We need the talkpage (and only this) of each existing page + for talkpage in pagegenerators.PageWithTalkPageGenerator( + redfam.article_generator( filter_existing=True ), + return_talk_only=True ): + + # Add reference to redfam to talkpages + talkpage.redfam = redfam + + yield talkpage + def treat_page( self ): """ Handles work on current page