From 8a339c721bc498c47165e56c725aa8426d818865 Mon Sep 17 00:00:00 2001 From: Ben Kurtovic Date: Thu, 4 May 2017 16:39:08 -0500 Subject: [PATCH] Fix spacing after banner. --- earwigbot/tasks/wikiproject_tagger.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/earwigbot/tasks/wikiproject_tagger.py b/earwigbot/tasks/wikiproject_tagger.py index 6cbcc4d..c366aec 100644 --- a/earwigbot/tasks/wikiproject_tagger.py +++ b/earwigbot/tasks/wikiproject_tagger.py @@ -397,7 +397,7 @@ class WikiProjectTagger(Task): def add_banner(self, code, banner): """Add *banner* to *code*, following template order conventions.""" predecessor = None - for template in code.ifilter_templates(): + for template in code.ifilter_templates(recursive=False): name = template.name.lower().replace("_", " ") for regex in self.TOP_TEMPS: if re.match(regex, name): @@ -412,7 +412,10 @@ class WikiProjectTagger(Task): self.logger.debug("Inserting banner after template") if not unicode(predecessor).endswith("\n"): banner = "\n" + banner - code.insert_after(predecessor, banner + "\n") + post = code.index(predecessor) + 1 + if len(code.nodes) > post and not code.get(post).startswith("\n"): + banner += "\n" + code.insert_after(predecessor, banner) else: self.logger.debug("Inserting banner at beginning") code.insert(0, banner + "\n")