matrix-appservice-kakaotalk/matrix_appservice_kakaotalk/kt/types/chat/attachment/media.py

106 lines
2.3 KiB
Python

# matrix-appservice-kakaotalk - A Matrix-KakaoTalk puppeting bridge.
# Copyright (C) 2022 Tulir Asokan, Andrew Ferrazzutti
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
from typing import Optional
from attr import dataclass
from . import Attachment
@dataclass
class MediaAttachment(Attachment):
# NOTE Added to cover Attachments that need a url but might not have a key
url: str
s: int
@dataclass
class MediaKeyAttachment(MediaAttachment):
k: str
@dataclass
class PhotoAttachment(MediaKeyAttachment):
# NOTE Changed superclass from Attachment
w: int
h: int
thumbnailUrl: str
thumbnailWidth: int
thumbnailHeight: int
cs: str
mt: str
@dataclass
class MultiPhotoAttachment(Attachment):
kl: list[str]
wl: list[int]
hl: list[int]
csl: list[str]
imageUrls: list[str]
thumbnailUrls: list[str]
thumbnailWidths: list[int]
thumbnailHeights: list[int]
sl: list[int] # NOTE Changed to a list
mtl: list[str] # NOTE Added
@dataclass
class VideoAttachment(MediaAttachment):
# NOTE Changed superclass from Attachment
tk: str
w: int
h: int
cs: str
d: int
@dataclass
class FileAttachment(MediaKeyAttachment):
# NOTE Changed superclass from Attachment
name: str
size: int
expire: int
cs: str
@dataclass
class AudioAttachment(Attachment):
url: str
d: int
expire: Optional[int] = None
s: Optional[int] = None # NOTE Optional for inbound
@dataclass
class LongTextAttachment(Attachment):
path: str
k: str
s: int
sd: bool
__all__ = [
"MediaAttachment",
"MediaKeyAttachment",
"PhotoAttachment",
"MultiPhotoAttachment",
"VideoAttachment",
"FileAttachment",
"AudioAttachment",
"LongTextAttachment",
]