-
Notifications
You must be signed in to change notification settings - Fork 2
/
telegram_bot.py
78 lines (69 loc) · 3.6 KB
/
telegram_bot.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
import telebot
import telegram_bot_config
import acrcloud_api
import parsing
# в файле telegram_bot_config.py - уникальный токен бота
bot = telebot.TeleBot(telegram_bot_config.token)
# декоратор функции (если приходит голосовое сообщение)
@bot.message_handler(content_types=["voice"])
def listener(message):
file_id = message.voice.file_id
file_info = bot.get_file(file_id)
file_path = file_info.file_path
downloaded_file = bot.download_file(file_path)
index = str(file_id)
inputFile = ("%s" % (telegram_bot_config.audio_dir))
try:
with open(inputFile, "wb") as new_file:
new_file.write(downloaded_file)
music_file_path = 'demo.ogg'
responce = acrcloud_api.get_responce(config=acrcloud_api.config, music_file_path=music_file_path,
start_seconds=3)
title, artist = acrcloud_api.parce_responce(responce)
if title is None and artist is None:
pattern_string = "Увы, я не нашел:("
bot.send_message(message.chat.id, pattern_string)
else:
yandex_music_ref = parsing.get_ref(title, artist)
if yandex_music_ref == -1: # если песня распозналась, но ее нет в Я.Музыке
pattern_string = "Это песня " + title + " исполнителя " + artist + "!\n"
else:
pattern_string = "Это песня " + title + " исполнителя " + artist + "!\n" + "Ссылка на Яндекс.Музыку " + yandex_music_ref
bot.send_message(message.chat.id, pattern_string)
except NameError:
pass
# если мы кидаем аудиофайл
@bot.message_handler(content_types=["audio"])
def listener(message):
file_id = message.audio.file_id
file_info = bot.get_file(file_id)
file_path = file_info.file_path
downloaded_file = bot.download_file(file_path)
index = str(file_id)
inputFile = ("%s" % (telegram_bot_config.audio_dir))
try:
with open(inputFile, "wb") as new_file:
new_file.write(downloaded_file)
music_file_path = 'demo.ogg'
responce = acrcloud_api.get_responce(config=acrcloud_api.config, music_file_path=music_file_path,
start_seconds=3)
title, artist = acrcloud_api.parce_responce(responce)
print(title, artist)
if title is None and artist is None:
pattern_string = "Увы, я не нашел:("
bot.send_message(message.chat.id, pattern_string)
else:
yandex_music_ref = parsing.get_ref(title, artist)
if yandex_music_ref == -1: # если песня распозналась, но ее нет в Я.Музыке
pattern_string = "Это песня " + title + " исполнителя " + artist + "!\n"
else:
pattern_string = "Это песня " + title + " исполнителя " + artist + "!\n" + "Ссылка на Яндекс.Музыку " + yandex_music_ref
bot.send_message(message.chat.id, pattern_string)
except NameError:
pass
# если просто пишем ему
@bot.message_handler(content_types=["text"])
def listener(message):
bot.send_message(message.chat.id, "Отправьте мне аудиофайл или запись песни :)")
if __name__ == '__main__':
bot.polling(none_stop=True)