mirror of
https://github.com/holll/teledown.git
synced 2025-01-08 11:37:45 +08:00
96 lines
3.7 KiB
Python
96 lines
3.7 KiB
Python
import os
|
||
import sys
|
||
from asyncio import CancelledError
|
||
from datetime import datetime
|
||
|
||
from telethon import TelegramClient
|
||
from telethon.errors import FileReferenceExpiredError
|
||
from telethon.tl.types import MessageMediaDocument, MessageMediaPhoto
|
||
|
||
from tools.tool import GetFileName, getHistoryMessage, GetChatId
|
||
from tools.tqdm import TqdmUpTo
|
||
|
||
|
||
# TODO 或许能加快下载速度的方法（https://gist.github.com/painor/7e74de80ae0c819d3e9abcf9989a8dd6）
|
||
# fileExist 检查文件是否存在（文件名和大小都相等），如果不存在重名文件加序号
|
||
def fileExist(file_path: str, file_size):
|
||
i = 2
|
||
ix = file_path.rfind('.', 1)
|
||
fileName = file_path[:ix]
|
||
fileType = file_path[ix:]
|
||
temp = file_path
|
||
while os.path.exists(temp):
|
||
if os.path.getsize(temp) == file_size:
|
||
return True, temp
|
||
temp = f'{fileName}({i}){fileType}'
|
||
i += 1
|
||
return False, temp
|
||
|
||
|
||
def GetFileSuffix(message) -> list:
|
||
mime_type = 'unknown/unknown'
|
||
if hasattr(message.media, 'document'):
|
||
mime_type = message.media.document.mime_type
|
||
elif hasattr(message.media, 'photo'):
|
||
mime_type = 'image/jpg'
|
||
return mime_type.split('/')
|
||
|
||
|
||
async def download_file(client: TelegramClient, channel_title, channel_id, message, old=False):
|
||
message_time = message.date
|
||
formatted_time = datetime.strftime(message_time, '%Y_%m')
|
||
|
||
file_name = GetFileName(message)
|
||
file_path = os.path.join(os.environ["save_path"], f'{channel_title}-{channel_id}', file_name)
|
||
file_size = message.file.size
|
||
ret, file_path = fileExist(file_path, file_size)
|
||
if not ret:
|
||
# 已经判断文件不存在，并且保证了文件名不重复
|
||
download_path = file_path + '.downloading'
|
||
print(f"开始下载：{file_name}")
|
||
try:
|
||
with TqdmUpTo(total=file_size, bar_format=TqdmUpTo.bar_format, desc=file_name[:10]) as bar:
|
||
await message.download_media(download_path, progress_callback=bar.update_to)
|
||
except CancelledError:
|
||
print("取消下载")
|
||
os.remove(download_path)
|
||
sys.exit()
|
||
# Todo 最新测试版telethon已经会捕获FileReferenceExpiredError了
|
||
except FileReferenceExpiredError:
|
||
if old:
|
||
print('重试失败，退出下载')
|
||
exit(1)
|
||
print('下载超时，重试中')
|
||
channelData = await client.get_entity(int(channel_id))
|
||
newMessages = client.iter_messages(entity=channelData, ids=message.id)
|
||
async for newMessage in newMessages:
|
||
await download_file(client, channel_title, channel_id, newMessage, old=True)
|
||
except Exception as e:
|
||
print("下载出错", e.__class__.__name__)
|
||
os.remove(download_path)
|
||
else:
|
||
os.rename(download_path, file_path)
|
||
else:
|
||
print(f"媒体已存在：{file_path}")
|
||
|
||
|
||
async def down_group(client: TelegramClient, chat_id, plus_func: str, from_user):
|
||
chat_id = await GetChatId(client, chat_id)
|
||
channel_title, messages = await getHistoryMessage(client, chat_id, plus_func, from_user=from_user) # messages是倒序的
|
||
async for message in messages:
|
||
"""转发消息
|
||
await message.forward_to('me')
|
||
"""
|
||
if message is None:
|
||
print('慢了一步，消息已被删除')
|
||
continue
|
||
'''
|
||
判定消息中是否存在媒体内容
|
||
MessageMediaDocument:文件、语音、视频
|
||
MessageMediaPhoto:图片
|
||
'''
|
||
if not isinstance(message.media, (MessageMediaDocument, MessageMediaPhoto)):
|
||
continue
|
||
await download_file(client, channel_title, chat_id, message)
|
||
print(channel_title, '全部下载完成')
|