80 lines
2.1 KiB
Python
80 lines
2.1 KiB
Python
import attr
|
|
from ._attachment import Attachment
|
|
|
|
|
|
@attr.s
|
|
class QuickReply:
|
|
"""Represents a quick reply."""
|
|
|
|
#: Payload of the quick reply
|
|
payload = attr.ib(None)
|
|
#: External payload for responses
|
|
external_payload = attr.ib(None, init=False)
|
|
#: Additional data
|
|
data = attr.ib(None)
|
|
#: Whether it's a response for a quick reply
|
|
is_response = attr.ib(False)
|
|
|
|
|
|
@attr.s
|
|
class QuickReplyText(QuickReply):
|
|
"""Represents a text quick reply."""
|
|
|
|
#: Title of the quick reply
|
|
title = attr.ib(None)
|
|
#: URL of the quick reply image (optional)
|
|
image_url = attr.ib(None)
|
|
#: Type of the quick reply
|
|
_type = "text"
|
|
|
|
|
|
@attr.s
|
|
class QuickReplyLocation(QuickReply):
|
|
"""Represents a location quick reply (Doesn't work on mobile)."""
|
|
|
|
#: Type of the quick reply
|
|
_type = "location"
|
|
|
|
|
|
@attr.s
|
|
class QuickReplyPhoneNumber(QuickReply):
|
|
"""Represents a phone number quick reply (Doesn't work on mobile)."""
|
|
|
|
#: URL of the quick reply image (optional)
|
|
image_url = attr.ib(None)
|
|
#: Type of the quick reply
|
|
_type = "user_phone_number"
|
|
|
|
|
|
@attr.s
|
|
class QuickReplyEmail(QuickReply):
|
|
"""Represents an email quick reply (Doesn't work on mobile)."""
|
|
|
|
#: URL of the quick reply image (optional)
|
|
image_url = attr.ib(None)
|
|
#: Type of the quick reply
|
|
_type = "user_email"
|
|
|
|
|
|
def graphql_to_quick_reply(q, is_response=False):
|
|
data = dict()
|
|
_type = q.get("content_type").lower()
|
|
if q.get("payload"):
|
|
data["payload"] = q["payload"]
|
|
if q.get("data"):
|
|
data["data"] = q["data"]
|
|
if q.get("image_url") and _type is not QuickReplyLocation._type:
|
|
data["image_url"] = q["image_url"]
|
|
data["is_response"] = is_response
|
|
if _type == QuickReplyText._type:
|
|
if q.get("title") is not None:
|
|
data["title"] = q["title"]
|
|
rtn = QuickReplyText(**data)
|
|
elif _type == QuickReplyLocation._type:
|
|
rtn = QuickReplyLocation(**data)
|
|
elif _type == QuickReplyPhoneNumber._type:
|
|
rtn = QuickReplyPhoneNumber(**data)
|
|
elif _type == QuickReplyEmail._type:
|
|
rtn = QuickReplyEmail(**data)
|
|
return rtn
|