get_chat_audios.py

from pyrogram import Client, enums
import access_creds
import os
from pyrogram.errors.exceptions.bad_request_400 import PeerIdInvalid, UsernameInvalid, UsernameNotOccupied, ChatAdminRequired
import sys

app = Client('my_session', api_id=access_creds.api_id, api_hash=access_creds.api_hash)
audios_list = []
from_users_list = []

def Search_audios():
    try:
        chat_creds = input('Enter correct numeric chat ID or chat username: ')
        try:
            with app:
                print('\n\033[1;90mGetting information...\n\033[1;00m')
                for message in app.search_messages(chat_id=chat_creds, filter=enums.MessagesFilter.AUDIO):
                    print('\033[1;94m', message.date.strftime('%Y-%m-%d %H:%M:%S') + '\033[1;00m from user ID: \033[1;94m' + str(message.from_user.id) + '\033[00m')
                    from_users_list.append(message.date.strftime('%Y-%m-%d %H:%M:%S') + ' from user ID: ' + str(message.from_user.id))
                    audios_list.append(message.audio.file_id)
        except AttributeError:
            print('')
        with open("Telegram_scan_results.txt", 'a') as output:
            output.write('CHAT (id = {}) AUDIOS => \n'.format(chat_creds))
            for row in from_users_list:
                output.write(str(row) + '\n')

        if len(audios_list) == 0:
            print('There are no audios one can get.')
        else:
            print('\nTotal amount of audios is \033[1;96m{}\033[1;00m'.format(len(audios_list)))
            download_audios = input("Enter how many audios to download (press \033[1;93m<Enter>\033[1;00m to skip): ")
            
            if download_audios == '' or download_audios == '0':
                print('One can find scan results in \033[1;95m{}/Telegram_scan_results.txt\033[1;00m file'.format(os.getcwd()))
            else:
                print('\n\033[1;90mDownloading...\033[1;00m\n')
                try:
                    count = 0
                    with app:
                        while count < int(download_audios):
                            app.download_media(audios_list[count], file_name='{0}/audios/{1}/'.format(os.getcwd(), chat_creds))
                            count += 1
                    print('One can find audios in \033[1;95m{0}/audios/{1}\033[1;00m directory'.format(os.getcwd(), chat_creds))
                except (TypeError, ValueError):
                    print('\033[1;91mNot correct input!\033[1;00m\n')
                print('One can find scan results in \033[1;95m{}/Telegram_scan_results.txt\033[1;00m file'.format(os.getcwd()))
    except ChatAdminRequired:
        print('\033[1;93mChat Admin permissions required!\033[1;00m')
    except (PeerIdInvalid, UsernameInvalid, UsernameNotOccupied, IndexError, TypeError):
        print('\033[1;91mNot correct input or smth went wrong!\033[1;00m')
    except KeyboardInterrupt:
        sys.exit("\n")