Added an helper function for expanding t.co URLS for tweets

src/sessions/twitter/compose.py

@@ -69,17 +69,11 @@ def compose_tweet(tweet, db, relative_times, show_screen_names=False, session=No
   else:
    text = "RT @%s: %s" % (tweet["retweeted_status"]["user"]["screen_name"], text)
  if ("message" in tweet) == False:
-  urls = utils.find_urls_in_text(text)
+
   if "retweeted_status" in tweet:
-   for url in range(0, len(urls)):
-    try:
-     text = text.replace(urls[url], tweet["retweeted_status"]["entities"]["urls"][url]["expanded_url"])
-    except: pass
+   text = utils.expand_urls(text, tweet["retweeted_status"]["entities"])
   else:
-   for url in range(0, len(urls)):
-    try:
-     text = text.replace(urls[url], tweet["entities"]["urls"][url]["expanded_url"])
-    except: pass
+   text = utils.expand_urls(text, tweet["entities"])
   if config.app['app-settings']['handle_longtweets']: pass
  return [user+", ", text, ts+", ", source]
 
@@ -110,10 +104,7 @@ def compose_direct_message(item, db, relative_times, show_screen_names=False, se
   else:
    user = sender["name"]
  if text[-1] in chars: text=text+"."
- urls = utils.find_urls_in_text(text)
- for url in range(0, len(urls)):
-  try:  text = text.replace(urls[url], item["message_create"]["message_data"]["entities"]["urls"][url]["expanded_url"])
-  except IndexError: pass
+ text = utils.expand_urls(text, item["message_create"]["message_data"]["entities"])
  return [user+", ", text, ts+", ", source]
 
 def compose_quoted_tweet(quoted_tweet, original_tweet, show_screen_names=False, session=None):

src/sessions/twitter/session.py

@@ -359,10 +359,7 @@ class Session(base.baseSession):
    value = "full_text"
   else:
    value = "text"
-  urls = utils.find_urls_in_text(quoted_tweet[value])
-  for url in range(0, len(urls)):
-   try:  quoted_tweet[value] = quoted_tweet[value].replace(urls[url], quoted_tweet["entities"]["urls"][url]["expanded_url"])
-   except IndexError: pass
+  quoted_tweet[value] = utils.expand_urls(quoted_tweet[value], quoted_tweet["entities"])
   if "quoted_status" in quoted_tweet:
    original_tweet = quoted_tweet["quoted_status"]
   elif "retweeted_status" in quoted_tweet and "quoted_status" in quoted_tweet["retweeted_status"]:
@@ -370,17 +367,13 @@ class Session(base.baseSession):
   else:
    return quoted_tweet
   original_tweet = self.check_long_tweet(original_tweet)
-
   if "full_text" in original_tweet:
    value = "full_text"
   elif "message" in original_tweet:
    value = "message"
   else:
    value = "text"
-  urls = utils.find_urls_in_text(original_tweet[value])
-  for url in range(0, len(urls)):
-   try:  original_tweet[value] = original_tweet[value].replace(urls[url], original_tweet["entities"]["urls"][url]["expanded_url"])
-   except IndexError: pass
+  original_tweet[value] = utils.expand_urls(original_tweet[value], original_tweet["entities"])
   return compose.compose_quoted_tweet(quoted_tweet, original_tweet)
 
  def check_long_tweet(self, tweet):

src/sessions/twitter/utils.py

@@ -208,4 +208,12 @@ def twitter_error(error):
   msg = _(u"No status found with that ID")
  else:
   msg = _(u"Error code {0}").format(error.error_code,)
- output.speak(msg)
+ output.speak(msg)
+
+def expand_urls(text, entities):
+ """ Expand all URLS present in text with information found in entities"""
+ urls = find_urls_in_text(text)
+ for url in entities["urls"]:
+  if url["url"] in text:
+   text = text.replace(url["url"], url["expanded_url"])
+ return text