mirror of
https://github.com/alfem/telegram-download-daemon.git
synced 2025-03-02 00:56:20 +08:00
Temp directory var missing fixed
This commit is contained in:
parent
e8bdf2e673
commit
92aa4b81a2
1 changed files with 4 additions and 3 deletions
|
@ -29,7 +29,7 @@ TELEGRAM_DAEMON_SESSION_PATH = getenv("TELEGRAM_DAEMON_SESSION_PATH")
|
||||||
TELEGRAM_DAEMON_DEST=getenv("TELEGRAM_DAEMON_DEST", "/telegram-downloads")
|
TELEGRAM_DAEMON_DEST=getenv("TELEGRAM_DAEMON_DEST", "/telegram-downloads")
|
||||||
TELEGRAM_DAEMON_TEMP=getenv("TELEGRAM_DAEMON_TEMP", "")
|
TELEGRAM_DAEMON_TEMP=getenv("TELEGRAM_DAEMON_TEMP", "")
|
||||||
|
|
||||||
TELEGRAM_DAEMON_TEMP_SUFFIX=".tdd"
|
TELEGRAM_DAEMON_TEMP_SUFFIX="tdd"
|
||||||
|
|
||||||
parser = argparse.ArgumentParser(
|
parser = argparse.ArgumentParser(
|
||||||
description="Script to download files from Telegram Channel.")
|
description="Script to download files from Telegram Channel.")
|
||||||
|
@ -75,6 +75,7 @@ api_id = args.api_id
|
||||||
api_hash = args.api_hash
|
api_hash = args.api_hash
|
||||||
channel_id = args.channel
|
channel_id = args.channel
|
||||||
downloadFolder = args.dest
|
downloadFolder = args.dest
|
||||||
|
tempFolder = args.temp
|
||||||
worker_count = multiprocessing.cpu_count()
|
worker_count = multiprocessing.cpu_count()
|
||||||
|
|
||||||
# Edit these lines:
|
# Edit these lines:
|
||||||
|
@ -162,9 +163,9 @@ with TelegramClient(getSession(), api_id, api_hash,
|
||||||
|
|
||||||
download_callback = lambda received, total: set_progress(filename, received, total)
|
download_callback = lambda received, total: set_progress(filename, received, total)
|
||||||
|
|
||||||
await client.download_media(event.message, f"{downloadFolder}/{filename}.{TELEGRAM_DAEMON_TEMP_SUFFIX}", progress_callback = download_callback)
|
await client.download_media(event.message, f"{tempFolder}/{filename}.{TELEGRAM_DAEMON_TEMP_SUFFIX}", progress_callback = download_callback)
|
||||||
set_progress(filename, 1, 1)
|
set_progress(filename, 1, 1)
|
||||||
rename(f"{downloadFolder}/{filename}.{TELEGRAM_DAEMON_TEMP_SUFFIX}", f"{downloadFolder}/{filename}")
|
rename(f"{tempFolder}/{filename}.{TELEGRAM_DAEMON_TEMP_SUFFIX}", f"{downloadFolder}/{filename}")
|
||||||
await log_reply(event, f"{filename} ready")
|
await log_reply(event, f"{filename} ready")
|
||||||
|
|
||||||
queue.task_done()
|
queue.task_done()
|
||||||
|
|
Loading…
Reference in a new issue