as-helios/affection-x-bot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fixed temp file record for media file paths

Browse source
This commit is contained in:
Alexander 2024-09-09 11:42:37 -04:00
parent b099fa7fec
commit 50e8257d9f
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
app/custom.py
View file

@ -100,7 +100,7 @@ async def find_text_and_download_media(bot, message):
# create a download progress file
os.makedirs(progress_folder := "{}/progress".format(os.getenv('DATA_FOLDER')), exist_ok=True)
media_progress_file = "{}/{}-{}.json".format(progress_folder, media_channel_id, message.id)
progress_data = {"last_update": time.time(), "percent": 0, "file_path": media_path, "cancelled": False, "id": None}
progress_data = {"last_update": time.time(), "percent": 0, "file_paths": [], "cancelled": False, "id": None}
open(media_progress_file, "w").write(json.dumps(progress_data))
media = []
@ -181,12 +181,13 @@ async def find_text_and_download_media(bot, message):
await bot.download_media(m, media_path, progress_callback=progress_callback)
# show it's completed
await bot.edit_message(progress_message, "Downloading from TG... 100%")
open(media_progress_file, "w").write(json.dumps(progress_data))
# store path to file in array
if os.path.exists(media_path):
media_paths.append(media_path)
progress_data['file_paths'] = media_paths
open(media_progress_file, "w").write(json.dumps(progress_data))
return text, media_paths