diff --git a/reddiscparser.py b/reddiscparser.py index 2d7164f..cd9cf29 100644 --- a/reddiscparser.py +++ b/reddiscparser.py @@ -106,7 +106,26 @@ class DiscussionParserBot( reddisc=red_page.page.title() ) ) -def main(*args): # noqa +def apply_conf_cat_generators( genFactory ): + """ + Builds generators for categories which are read from jogobot.config + + Parameters: + @param genFactory: The GeneratorFactory to which the builded generators + should be added. + @type genFactory: pagegenerators.GeneratorFactory + """ + # Create Generators for configured Categories + for category in jogobot.config["redundances"]["redpage_cats"]: + cgen = genFactory.getCategoryGen( + category, gen_func=pagegenerators.CategorizedPageGenerator) + + # If there is one, append to genFactory + if cgen: + genFactory.gens.append(cgen) + + +def main(*args): """ Process command line arguments and invoke bot. @@ -168,16 +187,7 @@ def main(*args): # noqa # Check wether there are generators waiting for factoring, if not # use configured categories if not genFactory.gens: - - # Create Generators for configured Categories - for category in jogobot.config["redundances"]["redpage_cats"]: - cgen = genFactory.getCategoryGen( - category, - gen_func=pagegenerators.CategorizedPageGenerator) - - # If there is one, append to genFactory - if cgen: - genFactory.gens.append(cgen) + apply_conf_cat_generators( genFactory ) # Create combined Generator (Union of all Generators) gen = genFactory.getCombinedGenerator()