1
0
mirror of https://github.com/ihabunek/toot synced 2024-12-25 00:20:55 +01:00
Toot-Mastodon-CLI-TUI-clien.../toot/tui/entities.py

43 lines
1.3 KiB
Python
Raw Normal View History

2019-08-24 13:43:41 +02:00
from collections import namedtuple
2019-08-27 14:34:51 +02:00
from .utils import parse_datetime
2019-08-24 13:43:41 +02:00
Author = namedtuple("Author", ["account", "display_name"])
def get_author(data, instance):
# Show the author, not the persopn who reblogged
status = data["reblog"] or data
acct = status['account']['acct']
acct = acct if "@" in acct else "{}@{}".format(acct, instance)
return Author(acct, status['account']['display_name'])
class Status:
"""
A wrapper around the Status entity data fetched from Mastodon.
https://github.com/tootsuite/documentation/blob/master/Using-the-API/API.md#status
"""
def __init__(self, data, instance):
self.data = data
self.instance = instance
# TODO: make Status immutable?
self.id = self.data["id"]
2019-08-24 12:53:55 +02:00
self.display_name = self.data["account"]["display_name"]
self.account = self.get_account()
self.created_at = parse_datetime(data["created_at"])
2019-08-24 13:43:41 +02:00
self.author = get_author(data, instance)
2019-08-24 12:53:55 +02:00
self.favourited = data.get("favourited", False)
self.reblogged = data.get("reblogged", False)
def get_account(self):
acct = self.data['account']['acct']
return acct if "@" in acct else "{}@{}".format(acct, self.instance)
def __repr__(self):
return "<Status id={}>".format(self.id)