From 7d52d4558a94d1cce2e6a0e15764b9a6be3c2ffa Mon Sep 17 00:00:00 2001
From: Ben Kurtovic <ben.kurtovic@verizon.net>
Date: Thu, 26 Jul 2012 21:11:23 -0400
Subject: [PATCH] Some updates to !lag.

---
 earwigbot/commands/lag.py          | 22 +++++++++++++++++-----
 earwigbot/wiki/copyvios/parsers.py |  2 +-
 2 files changed, 18 insertions(+), 6 deletions(-)

diff --git a/earwigbot/commands/lag.py b/earwigbot/commands/lag.py
index 8435336..1e5697c 100644
--- a/earwigbot/commands/lag.py
+++ b/earwigbot/commands/lag.py
@@ -32,12 +32,24 @@ class Lag(Command):
         site = self.get_site(data)
         if not site:
             return
-
-        msg = "\x0302{0}\x0F: Toolserver replag is {1}; database maxlag is {2}."
-        replag, maxlag = site.get_replag(), site.get_maxlag()
-        msg = msg.format(site.name, self.format(replag), self.format(maxlag))
+        if self.command == "replag":
+            base = "\x0302{0}\x0F: {1}."
+            msg = base.format(site.name, self.get_replag(site))
+        elif self.command == "maxlag":
+            base = "\x0302{0}\x0F: {1}."
+            msg = base.format(site.name, self.get_maxlag(site).capitalize())
+        else:
+            base = "\x0302{0}\x0F: {1}; {2}."
+            msg = base.format(site.name, self.get_replag(site),
+                              self.get_maxlag(site))
         self.reply(data, msg)
 
+    def get_replag(self, site):
+        return "Toolserver replag is {0}".format(self.time(site.get_replag()))
+
+    def get_maxlag(self, site):
+        return "database maxlag is {0}".format(self.time(site.get_maxlag()))
+
     def get_site(self, data):
         if data.kwargs and "project" in data.kwargs and "lang" in data.kwargs:
             project, lang = data.kwargs["project"], data.kwargs["lang"]
@@ -76,7 +88,7 @@ class Lag(Command):
                 return
         return site
 
-    def format(self, seconds):
+    def time(self, seconds):
         parts = [("year", 31536000), ("day", 86400), ("hour", 3600),
                  ("minute", 60), ("second", 1)]
         msg = []
diff --git a/earwigbot/wiki/copyvios/parsers.py b/earwigbot/wiki/copyvios/parsers.py
index fab867d..0df925b 100644
--- a/earwigbot/wiki/copyvios/parsers.py
+++ b/earwigbot/wiki/copyvios/parsers.py
@@ -71,7 +71,7 @@ class ArticleTextParser(BaseTextParser):
         """
         wikicode = mwparserfromhell.parse(self.text)
         clean = wikicode.strip_code(normalize=True, collapse=True)
-        self.clean = clean.replace("\n\n", "\n")  # Collapse extra newlines.
+        self.clean = clean.replace("\n\n", "\n")  # Collapse extra newlines
         return self.clean
 
     def chunk(self, nltk_dir, max_chunks, max_query=256):