twblue/src/sessions/twitter/compose.py

# -*- coding: utf-8 -*-
from __future__ import absolute_import
from __future__ import unicode_literals
from future import standard_library
standard_library.install_aliases()
from builtins import str
from builtins import chr
from builtins import range
import platform
system = platform.system()
from . import utils
import re
import html.entities
import time
import output
import languageHandler
import arrow
import logging
import config
from .long_tweets import twishort, tweets
log = logging.getLogger("compose")


def StripChars(s):
 """Converts any html entities in s to their unicode-decoded equivalents and returns a string."""
 entity_re = re.compile(r"&(#\d+|\w+);")
 def matchFunc(match):
  """Nested function to handle a match object.
 If we match &blah; and it's not found, &blah; will be returned.
 if we match #\d+, unichr(digits) will be returned.
 Else, a unicode string will be returned."""
  if match.group(1).startswith('#'): return chr(int(match.group(1)[1:]))
  replacement = html.entities.entitydefs.get(match.group(1), "&%s;" % match.group(1))
  return replacement
 return str(entity_re.sub(matchFunc, s))

chars = "abcdefghijklmnopqrstuvwxyz"

def compose_tweet(tweet, db, relative_times, show_screen_names=False, session=None):
 """ It receives a tweet and returns a list with the user, text for the tweet or message, date and the client where user is."""
 if system == "Windows":
  original_date = arrow.get(tweet.created_at, locale="en")
  if relative_times == True:
   ts = original_date.humanize(locale=languageHandler.curLang[:2])
  else:
   ts = original_date.shift(seconds=db["utc_offset"]).format(_(u"dddd, MMMM D, YYYY H:m:s"), locale=languageHandler.curLang[:2])
 else:
  ts = tweet.created_at
 if hasattr(tweet, "message"):
  value = "message"
 elif hasattr(tweet, "full_text"):
  value = "full_text"
 else:
  value = "text"
 if hasattr(tweet, "retweeted_status") and value != "message":
  text = StripChars(getattr(tweet.retweeted_status, value))
 else:
  text = StripChars(getattr(tweet, value))
 if show_screen_names:
  user = tweet.user.screen_name
 else:
  user = tweet.user.name
 source = re.sub(r"(?s)<.*?>", "", tweet.source)
 if hasattr(tweet, "retweeted_status"):
  if (hasattr(tweet, "message")) == False and tweet.retweeted_status.is_quote_status == False:
   text = "RT @%s: %s" % (tweet.retweeted_status.user.screen_name, text)
  elif tweet.retweeted_status.is_quote_status:
   text = "%s" % (text)
  else:
   text = "RT @%s: %s" % (tweet.retweeted_status.user.screen_name, text)
 if not hasattr(tweet, "message"):

  if hasattr(tweet, "retweeted_status"):
   text = utils.expand_urls(text, tweet.retweeted_status.entities)
  else:
   text = utils.expand_urls(text, tweet.entities)
  if config.app['app-settings']['handle_longtweets']: pass
 return [user+", ", text, ts+", ", source]

def compose_direct_message(item, db, relative_times, show_screen_names=False, session=None):
 # for a while this function will be together with compose_dm.
 # this one composes direct messages based on events (new API Endpoints).
 if system == "Windows":
   # Let's remove the last 3 digits in the timestamp string.
   # Twitter sends their "epoch" timestamp with 3 digits for milliseconds and arrow doesn't like it.
  original_date = arrow.get(int(item.created_timestamp[:-3]))
  if relative_times == True:
   ts = original_date.humanize(locale=languageHandler.curLang[:2])
  else:
   ts = original_date.shift(seconds=db["utc_offset"]).format(_(u"dddd, MMMM D, YYYY H:m:s"), locale=languageHandler.curLang[:2])
 else:
  ts = item.created_timestamp
 text = StripChars(item.message_create.message_data.text)
 source = "DM"
 sender = session.get_user(item.message_create.sender_id)
 if db["user_name"] == sender.screen_name:
  if show_screen_names:
   user = _(u"Dm to %s ") % (session.get_user(item.message_create.target.recipient_id).screen_name)
  else:
   user = _(u"Dm to %s ") % (session.get_user(item.message_create.target.recipient_id).name)
 else:
  if show_screen_names:
   user = sender.screen_name
  else:
   user = sender.name
 if text[-1] in chars: text=text+"."
 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):
 """ It receives a tweet and returns a list with the user, text for the tweet or message, date and the client where user is."""
 if hasattr(quoted_tweet, "retweeted_status"):
  if hasattr(quoted_tweet.retweeted_status, "full_text"):
   value = "full_text"
  else:
   value = "text"
  text = StripChars(getattr(quoted_tweet.retweeted_status, value))
 else:
  if hasattr(quoted_tweet, "full_text"):
   value = "full_text"
  else:
   value = "text"
  text = StripChars(getattr(quoted_tweet, value))
 if show_screen_names:
  quoting_user = quoted_tweet.user.screen_name
 else:
  quoting_user = quoted_tweet.user.name
 source = re.sub(r"(?s)<.*?>", "", quoted_tweet.source)
 if hasattr(quoted_tweet, "retweeted_status"):
  text = "rt @%s: %s" % (quoted_tweet.retweeted_status.user.screen_name, text)
 if text[-1] in chars: text=text+"."
 original_user = original_tweet.user.screen_name
 if hasattr(original_tweet, "message"):
  original_text = original_tweet.message
 elif hasattr(original_tweet, "full_text"):
  original_text = StripChars(original_tweet.full_text)
 else:
   original_text = StripChars(original_tweet.text)
 quoted_tweet.message = _(u"{0}. Quoted  tweet from @{1}: {2}").format( text, original_user, original_text)
 quoted_tweet = tweets.clear_url(quoted_tweet)
 quoted_tweet.entities["urls"].extend(original_tweet.entities["urls"])
 return quoted_tweet

def compose_followers_list(tweet, db, relative_times=True, show_screen_names=False, session=None):
 if system == "Windows":
  original_date = arrow.get(tweet.created_at, "ddd MMM D H:m:s Z YYYY", locale="en")
  if relative_times == True:
   ts = original_date.humanize(locale=languageHandler.curLang[:2])
  else:
   ts = original_date.shift(seconds=db["utc_offset"]).format(_(u"dddd, MMMM D, YYYY H:m:s"), locale=languageHandler.curLang[:2])
 else:
  ts = tweet.created_at
 if hasattr(tweet, "status"):
  if len(tweet.status) > 4 and system == "Windows":
   original_date2 = arrow.get(tweet.status.created_at, "ddd MMM D H:m:s Z YYYY", locale="en")
   if relative_times:
    ts2 = original_date2.humanize(locale=languageHandler.curLang[:2])
   else:
    ts2 = original_date2.shift(seconds=db["utc_offset"]).format(_(u"dddd, MMMM D, YYYY H:m:s"), locale=languageHandler.curLang[:2])
  else:
   ts2 = _("Unavailable")
 else:
  ts2 = _("Unavailable")
 return [_(u"%s (@%s). %s followers, %s friends, %s tweets. Last tweeted %s. Joined Twitter %s") % (tweet.name, tweet.screen_name, tweet.followers_count, tweet.friends_count,  tweet.statuses_count, ts2, ts)]

def compose_list(list):
 name = list.name
 if list.description == None: description = _(u"No description available")
 else: description = list.description
 user = list.user.name
 members = str(list.member_count)
 if list.mode == "private": status = _(u"private")
 else: status = _(u"public")
 return [name, description, user, members, status]