主页 / 问题 / 1603976
Accepted
Funsy
Asked:2025-01-04 01:57:46 +0000 UTC2025-01-04 01:57:46 +0000 UTC

保存文件 - Telebot

  • 772

我决定在 Telebot 上为 Telegram 编写一个简单的机器人。我遇到了一个问题：我需要下载用户发送的所有文件和文本。我这样做：

from config import TOKEN
import telebot
import os
import time

bot = telebot.TeleBot(TOKEN)

@bot.message_handler(commands=['start'])
def start(message):
    bot.send_message(message.chat.id, 'Отправь своё (фото, видео, музыку, или текст)')

@bot.message_handler(content_types=['text', 'photo', 'video', 'audio'])
def download_data(message):
    user_directory = f'data/{message.from_user.username}'

    if not os.path.exists(user_directory):
        os.makedirs(user_directory)
    
    if message.photo:
        photo = message.photo[-1]
        file = bot.get_file(photo.file_id)
        downloaded_file = bot.download_file(file.file_path)
        filename = f'{user_directory}/photo_{int(time.time())}.jpg'

        with open(filename, 'wb') as f:
            f.write(downloaded_file)

    elif message.video:
        video = message.video
        file = bot.get_file(video.file_id)
        downloaded_file = bot.download_file(file.file_path)
        filename = f'{user_directory}/video_{int(time.time())}.mp4'

        with open(filename, 'wb') as f:
            f.write(downloaded_file)

    elif message.audio:
        audio = message.audio
        file = bot.get_file(audio.file_id)
        downloaded_file = bot.download_file(file.file_path)
        filename = f'{user_directory}/audio_{int(time.time())}.mp3'
        
        with open(filename, 'wb') as f:
            f.write(downloaded_file)

    elif message.text:
        filename = f'{user_directory}/text.txt'
        
        with open(filename, 'w') as f:
            f.write(message.text)

bot.infinity_polling()

发送大量图像、视频或音频时，会下载所发送内容的一小部分。我需要保存所有发送的文件。提前致谢！

python

1 个回答

  1. Best Answer

    您的问题是由于 @bot.message_handler 一次只处理一条消息。如果用户在一条消息中发送多个文件（例如，作为相册），Telegram 将它们作为一组发送，并且您的逻辑仅处理消息的一部分。要解决该问题，您需要处理该组中的所有文件。

    from config import TOKEN
import telebot
import os
import time

bot = telebot.TeleBot(TOKEN)

@bot.message_handler(commands=['start'])
def start(message):
    bot.send_message(message.chat.id, 'Отправь своё (фото, видео, музыку, или текст)')

@bot.message_handler(content_types=['photo', 'video', 'audio', 'text', 'document'])
def download_data(message):
    user_directory = f'data/{message.from_user.username}'
    if not os.path.exists(user_directory):
        os.makedirs(user_directory)

    if message.photo:  # Фото (может быть несколько в одном сообщении)
        for photo in message.photo:
            file = bot.get_file(photo.file_id)
            downloaded_file = bot.download_file(file.file_path)
            filename = f'{user_directory}/photo_{int(time.time())}.jpg'

            with open(filename, 'wb') as f:
                f.write(downloaded_file)

    elif message.video:  # Видео
        file = bot.get_file(message.video.file_id)
        downloaded_file = bot.download_file(file.file_path)
        filename = f'{user_directory}/video_{int(time.time())}.mp4'

        with open(filename, 'wb') as f:
            f.write(downloaded_file)

    elif message.audio:  # Аудио
        file = bot.get_file(message.audio.file_id)
        downloaded_file = bot.download_file(file.file_path)
        filename = f'{user_directory}/audio_{int(time.time())}.mp3'

        with open(filename, 'wb') as f:
            f.write(downloaded_file)

    elif message.document:  # Документы
        file = bot.get_file(message.document.file_id)
        downloaded_file = bot.download_file(file.file_path)
        filename = f'{user_directory}/{message.document.file_name}'

        with open(filename, 'wb') as f:
            f.write(downloaded_file)

    elif message.text:  # Текст
        filename = f'{user_directory}/text_{int(time.time())}.txt'

        with open(filename, 'w') as f:
            f.write(message.text)

# Обработка альбомов медиа (группы медиафайлов)
@bot.message_handler(content_types=['media_group'])
def download_media_group(message):
    user_directory = f'data/{message.from_user.username}'
    if not os.path.exists(user_directory):
        os.makedirs(user_directory)

    for media in message.media_group_id:
        if media.photo:  # Фото из альбома
            file = bot.get_file(media.photo[-1].file_id)
            downloaded_file = bot.download_file(file.file_path)
            filename = f'{user_directory}/photo_{int(time.time())}.jpg'

            with open(filename, 'wb') as f:
                f.write(downloaded_file)

        elif media.video:  # Видео из альбома
            file = bot.get_file(media.video.file_id)
            downloaded_file = bot.download_file(file.file_path)
            filename = f'{user_directory}/video_{int(time.time())}.mp4'

            with open(filename, 'wb') as f:
                f.write(downloaded_file)

# Запуск бота
bot.infinity_polling()
    • 0

相关问题