from datetime import datetime


def parse_datetime(value):
    """Returns an aware datetime in local timezone"""
    return datetime.strptime(value, "%Y-%m-%dT%H:%M:%S.%f%z").astimezone()


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

        self.id = self.data["id"]
        self.account = self.get_account()
        self.created_at = parse_datetime(data["created_at"])

    def get_account(self):
        acct = self.data['account']['acct']
        return acct if "@" in acct else "{}@{}".format(acct, self.instance)