nikita/mab
1
0
Fork 0
Code Issues 2 Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed downloader bugs related to temp file name

Browse Source
This commit is contained in:
nikita
2026-09-13 03:28:27 +03:00
parent 0e9b621895
commit 43b5d8e3a6
1 changed files with 3 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/mab/bot/_client_downloader.py
View File

@@ -2,10 +2,8 @@ import asyncio
import aiofiles import aiofiles
import logging import logging
import os import os
import time
import threading import threading
import traceback from uuid import uuid4
import random
from pathlib import Path from pathlib import Path
import hashlib import hashlib
@@ -36,7 +34,7 @@ class ClientDownloader:
config: MatrixBotConfig, config: MatrixBotConfig,
client: AsyncClient) -> None: client: AsyncClient) -> None:
""" """
Setup the sender. Setup the downloader.
Args: Args:
- config - config to use - config - config to use
@@ -218,14 +216,12 @@ class ClientDownloader:
if key is not None and iv is not None and sha256 is not None: if key is not None and iv is not None and sha256 is not None:
# on disk # on disk
if isinstance(result, Path): if isinstance(result, Path):
r = unpaddedbase64.encode_base64(random.randbytes(8))
temp_path = result.with_name( temp_path = result.with_name(
f".{result.name}.{int(time.time())}{r}.tmp" f".{result.name}.{uuid4().hex}.tmp"
) )
await self._decrypt_file( await self._decrypt_file(
result, temp_path, key, iv, sha256 result, temp_path, key, iv, sha256
) )
result.unlink()
temp_path.replace(result) temp_path.replace(result)
temp_path = None temp_path = None
# in memory # in memory