Fix cache updater

This commit is contained in:
2023-02-26 21:18:26 +01:00
parent 596429261d
commit 7d5faea45f

View File

@@ -1,5 +1,6 @@
import collections import collections
from datetime import timedelta from datetime import timedelta
from io import BytesIO
import logging import logging
import random import random
from tempfile import SpooledTemporaryFile from tempfile import SpooledTemporaryFile
@@ -86,7 +87,7 @@ async def cache_file(book: Book, file_type: str) -> Optional[CachedFile]:
response, client, filename = data response, client, filename = data
caption = get_caption(book) caption = get_caption(book)
temp_file = UploadFile(filename) temp_file = UploadFile(BytesIO(), filename=filename)
async for chunk in response.aiter_bytes(2048): async for chunk in response.aiter_bytes(2048):
await temp_file.write(chunk) await temp_file.write(chunk)
await temp_file.seek(0) await temp_file.seek(0)