feat: 1-to-1 message render + web data-lake backend
This commit is contained in:
@@ -33,6 +33,11 @@ def self_destruct_ttl(message: Message) -> int | None:
|
||||
return getattr(obj, "ttl_seconds", None) if obj is not None else None
|
||||
|
||||
|
||||
def media_unique_id(message: Message) -> str | None:
|
||||
_, obj = media_object(message)
|
||||
return getattr(obj, "file_unique_id", None) if obj is not None else None
|
||||
|
||||
|
||||
async def capture_media( # noqa: PLR0913
|
||||
client: Client,
|
||||
message: Message,
|
||||
@@ -44,6 +49,7 @@ async def capture_media( # noqa: PLR0913
|
||||
kind, obj = media_object(message)
|
||||
if obj is None:
|
||||
return
|
||||
unique_id = getattr(obj, "file_unique_id", None)
|
||||
ttl = getattr(obj, "ttl_seconds", None)
|
||||
want = toggles.self_destruct_media if ttl else toggles.media
|
||||
file_size = getattr(obj, "file_size", None)
|
||||
@@ -51,12 +57,25 @@ async def capture_media( # noqa: PLR0913
|
||||
storage_key: str | None = None
|
||||
downloaded = False
|
||||
if want:
|
||||
buffer = await client.download_media(message, in_memory=True)
|
||||
if isinstance(buffer, BytesIO):
|
||||
data = buffer.getvalue()
|
||||
storage_key = ctx.storage.put(data)
|
||||
file_size = len(data)
|
||||
existing = await repository.current_media(
|
||||
ctx.pool, ctx.account_id, chat_id, message_id
|
||||
)
|
||||
if (
|
||||
existing is not None
|
||||
and existing["downloaded"]
|
||||
and existing["unique_id"] == unique_id
|
||||
and existing["storage_key"] is not None
|
||||
):
|
||||
storage_key = existing["storage_key"]
|
||||
file_size = existing["file_size"]
|
||||
downloaded = True
|
||||
else:
|
||||
buffer = await client.download_media(message, in_memory=True)
|
||||
if isinstance(buffer, BytesIO):
|
||||
data = buffer.getvalue()
|
||||
storage_key = ctx.storage.put(data)
|
||||
file_size = len(data)
|
||||
downloaded = True
|
||||
await repository.insert_media(
|
||||
ctx.pool,
|
||||
ctx.account_id,
|
||||
@@ -67,5 +86,6 @@ async def capture_media( # noqa: PLR0913
|
||||
file_size,
|
||||
mime,
|
||||
ttl,
|
||||
unique_id,
|
||||
downloaded=downloaded,
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user