From bb40d1ad38ad9754819f46ec7d025fab67c947c0 Mon Sep 17 00:00:00 2001
From: "Sean B. Palmer" <sean@miscoranda.com>
Date: Tue, 24 Jul 2012 09:11:05 +0100
Subject: Fixed seen.py apos problem, thus thanking Xarthisius

---
 modules/twitter.py | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

(limited to 'modules/twitter.py')

diff --git a/modules/twitter.py b/modules/twitter.py
index 4e7c134..afd1d5c 100755
--- a/modules/twitter.py
+++ b/modules/twitter.py
@@ -16,6 +16,8 @@ r_p = re.compile(r'(?ims)(<p class="js-tweet-text.*?</p>)')
 r_tag = re.compile(r'(?ims)<[^>]+>')
 r_anchor = re.compile(r'(?ims)(<a.*?</a>)')
 r_expanded = re.compile(r'(?ims)data-expanded-url=["\'](.*?)["\']')
+r_whiteline = re.compile(r'(?ims)[ \t]+[\r\n]+')
+r_breaks = re.compile(r'(?ims)[\r\n]+')
 
 def entity(*args, **kargs):
    return web.entity(*args, **kargs).encode('utf-8')
@@ -41,8 +43,8 @@ def read_tweet(url):
       text = expand(text)
       text = r_tag.sub('', text)
       text = text.strip()
-      text = text.replace('\r', '')
-      text = text.replace('\n', '')
+      text = r_whiteline.sub(' ', text)
+      text = r_breaks.sub(' ', text)
       return decode(text)
    return "Sorry, couldn't get a tweet from %s" % url
 
@@ -66,7 +68,11 @@ def id_tweet(tid):
    return "Sorry, couldn't get a tweet from %s" % link
 
 def twitter(phenny, input):
-   arg = input.group(2).strip()
+   arg = input.group(2)
+   if not arg:
+      return phenny.reply("Give me a link, a username, or a tweet id")
+
+   arg = arg.strip()
    if isinstance(arg, unicode):
       arg = arg.encode('utf-8')
 
-- 
cgit v1.2.3-1-g7c22