Source code for fbchat._models._attachment

import attr
from . import Image
from .._common import attrs_default
from .. import _util

from typing import Optional, Sequence


@attrs_default
class Attachment:
    """Represents a Facebook attachment."""

    #: The attachment ID
    id = attr.ib(None, type=Optional[str])


@attrs_default
class UnsentMessage(Attachment):
    """Represents an unsent message attachment."""


@attrs_default
class ShareAttachment(Attachment):
    """Represents a shared item (e.g. URL) attachment."""

    #: ID of the author of the shared post
    author = attr.ib(None, type=Optional[str])
    #: Target URL
    url = attr.ib(None, type=Optional[str])
    #: Original URL if Facebook redirects the URL
    original_url = attr.ib(None, type=Optional[str])
    #: Title of the attachment
    title = attr.ib(None, type=Optional[str])
    #: Description of the attachment
    description = attr.ib(None, type=Optional[str])
    #: Name of the source
    source = attr.ib(None, type=Optional[str])
    #: The attached image
    image = attr.ib(None, type=Optional[Image])
    #: URL of the original image if Facebook uses ``safe_image``
    original_image_url = attr.ib(None, type=Optional[str])
    #: List of additional attachments
    attachments = attr.ib(factory=list, type=Sequence[Attachment])

    @classmethod
    def _from_graphql(cls, data):
        from . import _file

        image = None
        original_image_url = None
        media = data.get("media")
        if media and media.get("image"):
            image = Image._from_uri(media["image"])
            original_image_url = (
                _util.get_url_parameter(image.url, "url")
                if "/safe_image.php" in image.url
                else image.url
            )

        url = data.get("url")
        return cls(
            id=data.get("deduplication_key"),
            author=data["target"]["actors"][0]["id"]
            if data["target"].get("actors")
            else None,
            url=url,
            original_url=_util.get_url_parameter(url, "u")
            if "/l.php?u=" in url
            else url,
            title=data["title_with_entities"].get("text"),
            description=data["description"].get("text")
            if data.get("description")
            else None,
            source=data["source"].get("text") if data.get("source") else None,
            image=image,
            original_image_url=original_image_url,
            attachments=[
                _file.graphql_to_subattachment(attachment)
                for attachment in data.get("subattachments")
            ],
        )