twblue/srcantiguo/sessions/mastodon/compose.py

# -*- coding: utf-8 -*-
import arrow
import languageHandler
from .  import utils, templates

def compose_post(post, db, settings, relative_times, show_screen_names, safe=True):
    if show_screen_names == False:
        user = utils.get_user_alias(post.account, settings)
    else:
        user = post.account.get("acct")
    original_date = arrow.get(post.created_at)
    if relative_times:
        ts = original_date.humanize(locale=languageHandler.curLang[:2])
    else:
        ts = original_date.shift(hours=db["utc_offset"]).format(_("dddd, MMMM D, YYYY H:m"), locale=languageHandler.curLang[:2])
    if post.reblog != None:
        text = _("Boosted from @{}: {}").format(post.reblog.account.acct, templates.process_text(post.reblog, safe=safe))
    else:
        text = templates.process_text(post, safe=safe)
    # Handle quoted posts
    if hasattr(post, 'quote') and post.quote != None and hasattr(post.quote, 'quoted_status') and post.quote.quoted_status != None:
        quoted_user = post.quote.quoted_status.account.acct
        quoted_text = templates.process_text(post.quote.quoted_status, safe=safe)
        text = text + " " + _("Quoting @{}: {}").format(quoted_user, quoted_text)
    filtered = utils.evaluate_filters(post=post, current_context="home")
    if filtered != None:
        text = _("hidden by filter {}").format(filtered)
    source = post.get("application", "")
    # "" means remote user, None for legacy apps so we should cover both sides.
    if source != None and source != "":
        source = source.get("name", "")
    else:
        source = ""
    return [user+", ", text, ts+", ", source]

def compose_user(user, db, settings, relative_times=True, show_screen_names=False, safe=False):
    original_date = arrow.get(user.created_at)
    if relative_times:
        ts = original_date.humanize(locale=languageHandler.curLang[:2])
    else:
        ts = original_date.shift(hours=db["utc_offset"]).format(_("dddd, MMMM D, YYYY H:m:s"), locale=languageHandler.curLang[:2])
    name = utils.get_user_alias(user, settings)
    return [_("%s (@%s). %s followers, %s following, %s posts. Joined %s") % (name, user.acct, user.followers_count, user.following_count,  user.statuses_count, ts)]

def compose_conversation(conversation, db, settings, relative_times, show_screen_names, safe=False):
    users = []
    for account in conversation.accounts:
        if account.display_name != "":
            users.append(utils.get_user_alias(account, settings))
        else:
            users.append(account.username)
    users = ", ".join(users)
    last_post = compose_post(conversation.last_status, db, settings, relative_times, show_screen_names)
    text = _("Last message from {}: {}").format(last_post[0], last_post[1])
    return [users, text, last_post[-2], last_post[-1]]

def compose_notification(notification, db, settings, relative_times, show_screen_names, safe=False):
    if show_screen_names == False:
        user = utils.get_user_alias(notification.account, settings)
    else:
        user = notification.account.get("acct")
    original_date = arrow.get(notification.created_at)
    if relative_times:
        ts = original_date.humanize(locale=languageHandler.curLang[:2])
    else:
        ts = original_date.shift(hours=db["utc_offset"]).format(_("dddd, MMMM D, YYYY H:m"), locale=languageHandler.curLang[:2])
    text = "Unknown: %r" % (notification)
    if notification.type == "status":
        text = _("{username} has posted: {status}").format(username=user, status=",".join(compose_post(notification.status, db, settings, relative_times, show_screen_names, safe=safe)))
    elif notification.type == "mention":
        text = _("{username} has mentioned you: {status}").format(username=user, status=",".join(compose_post(notification.status, db, settings, relative_times, show_screen_names, safe=safe)))
    elif notification.type == "reblog":
        text = _("{username} has boosted: {status}").format(username=user, status=",".join(compose_post(notification.status, db, settings, relative_times, show_screen_names, safe=safe)))
    elif notification.type == "favourite":
        text = _("{username} has added to favorites: {status}").format(username=user, status=",".join(compose_post(notification.status, db, settings, relative_times, show_screen_names, safe=safe)))
    elif notification.type == "follow":
        text = _("{username} has followed you.").format(username=user)
    elif notification.type == "admin.sign_up":
        text = _("{username} has joined the instance.").format(username=user)
    elif notification.type == "poll":
        text = _("A poll in which you have voted has expired: {status}").format(status=",".join(compose_post(notification.status, db, settings, relative_times, show_screen_names, safe=safe)))
    elif notification.type == "follow_request":
        text = _("{username} wants to follow you.").format(username=user)
    filtered = utils.evaluate_filters(post=notification, current_context="notifications")
    if filtered != None:
        text = _("hidden by filter {}").format(filtered)
    return [user, text, ts]

def compose_announcement(announcement, db, settings, relative_times, show_screen_names, safe=False):
    # Use the default template or a configured one if available
    template = settings.get("templates", {}).get("announcement", templates.announcement_default_template)
    text = templates.render_announcement(announcement, template, settings, relative_times, db["utc_offset"])
    return [text]