Add YTM, newer and better error handling
This commit is contained in:
parent
78556ef6d2
commit
fe29eb47e7
1
Pipfile
1
Pipfile
@ -13,6 +13,7 @@ dpath = "*"
|
||||
gmusicapi = "*"
|
||||
pathlib = "*"
|
||||
uuid = "*"
|
||||
ytmusicapi = "*"
|
||||
|
||||
[requires]
|
||||
python_version = "3.6"
|
||||
|
||||
122
music.py
122
music.py
@ -1,12 +1,14 @@
|
||||
import logging
|
||||
from tbotconfig import *
|
||||
|
||||
# Telegram things:
|
||||
#from telegram import InlineKeyboardButton, InlineKeyboardMarkup
|
||||
from telegram import InlineKeyboardMarkup, InlineKeyboardButton, InlineQueryResultArticle, InputTextMessageContent, InlineQueryResult, InlineQueryResultAudio
|
||||
from telegram import Update, ParseMode, InlineKeyboardMarkup, InlineKeyboardButton, InlineQueryResultArticle, InputTextMessageContent, InlineQueryResult, InlineQueryResultAudio
|
||||
from telegram.ext import Updater, CommandHandler, MessageHandler, Filters, CallbackQueryHandler, ConversationHandler, InlineQueryHandler, ChosenInlineResultHandler, CallbackContext
|
||||
|
||||
# Other needful stuff
|
||||
import html
|
||||
import json
|
||||
import logging
|
||||
import traceback
|
||||
from uuid import uuid4
|
||||
import urllib.request
|
||||
import urllib
|
||||
@ -14,29 +16,22 @@ import re
|
||||
import spotipy
|
||||
import spotipy.oauth2
|
||||
import json
|
||||
import dpath
|
||||
from dpath import util
|
||||
import requests
|
||||
import time
|
||||
import os
|
||||
from pathlib import Path
|
||||
|
||||
from ytmusicapi import YTMusic
|
||||
ytmusic = YTMusic()
|
||||
|
||||
#Configure how many results to fetch
|
||||
num_results = 5
|
||||
|
||||
|
||||
# Enable logging
|
||||
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
||||
level=logging.INFO)
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
# Spotify:
|
||||
spotify_credentials = spotipy.oauth2.SpotifyClientCredentials(SPOTIPY_CLIENT_ID, SPOTIPY_CLIENT_SECRET)
|
||||
sp = spotipy.Spotify(client_credentials_manager=spotify_credentials)
|
||||
|
||||
|
||||
#
|
||||
# Actual bot stuffs
|
||||
#
|
||||
# Define a few command handlers. These usually take the two arguments bot and update. Error handlers also receive the raised TelegramError object in error.
|
||||
@ -56,26 +51,23 @@ def inlinequery(update, context: CallbackContext):
|
||||
|
||||
spotify_credentials = spotipy.oauth2.SpotifyClientCredentials(SPOTIPY_CLIENT_ID, SPOTIPY_CLIENT_SECRET)
|
||||
sp = spotipy.Spotify(client_credentials_manager=spotify_credentials)
|
||||
|
||||
if query:
|
||||
sp_info = sp.search(q=query, limit=num_results)
|
||||
|
||||
for i in range(num_results):
|
||||
try:
|
||||
dpath.util.get(sp_info, f"/tracks/items/[{i}]/name")
|
||||
except:
|
||||
continue
|
||||
else:
|
||||
sp_title = dpath.util.get(sp_info, f"/tracks/items/[{i}]/name")
|
||||
|
||||
sp_artist = dpath.util.get(sp_info, f"/tracks/items/[{i}]/artists/[0]/name")
|
||||
sp_albname = dpath.util.get(sp_info, f"/tracks/items/[{i}]/album/name")
|
||||
sp_albdate = dpath.util.get(sp_info, f"/tracks/items/[{i}]/album/release_date")
|
||||
sp_art = dpath.util.get(sp_info, f"/tracks/items/[{i}]/album/images/[2]/url")
|
||||
sp_audio = dpath.util.get(sp_info, f"/tracks/items/[{i}]/preview_url")
|
||||
sp_link = dpath.util.get(sp_info, f"/tracks/items/[{i}]/external_urls/spotify")
|
||||
sp_title = util.get(sp_info, f"/tracks/items/[{i}]/name")
|
||||
sp_artist = util.get(sp_info, f"/tracks/items/[{i}]/artists/[0]/name")
|
||||
sp_albname = util.get(sp_info, f"/tracks/items/[{i}]/album/name")
|
||||
sp_albdate = util.get(sp_info, f"/tracks/items/[{i}]/album/release_date")
|
||||
sp_art = util.get(sp_info, f"/tracks/items/[{i}]/album/images/[2]/url")
|
||||
sp_audio = util.get(sp_info, f"/tracks/items/[{i}]/preview_url")
|
||||
sp_link = util.get(sp_info, f"/tracks/items/[{i}]/external_urls/spotify")
|
||||
songlink = "https://song.link/{0}".format(sp_link)
|
||||
|
||||
reply_markup = InlineKeyboardMarkup([[InlineKeyboardButton("Spotify", url = sp_link), InlineKeyboardButton("More", url = songlink)]])
|
||||
yt_q = f"{sp_artist} - {sp_title}"
|
||||
yt_res = ytmusic.search(yt_q, 'songs', limit=1)
|
||||
yt_id = util.get(yt_res, f"[1]/videoId")
|
||||
yt_link = f"https://music.youtube.com/watch?v={yt_id}"
|
||||
reply_markup = InlineKeyboardMarkup([[InlineKeyboardButton("YouTube", url = yt_link), InlineKeyboardButton("Spotify", url = sp_link), InlineKeyboardButton("More", url = songlink)]])
|
||||
|
||||
description = f"By {sp_artist} on the album {sp_albname}, released {sp_albdate}"
|
||||
|
||||
@ -88,38 +80,78 @@ def inlinequery(update, context: CallbackContext):
|
||||
|
||||
update.inline_query.answer(results)
|
||||
|
||||
def error(update, context: CallbackContext):
|
||||
# Log Errors caused by Updates
|
||||
logger.warning(f"\nUpdate {update} caused error {context.error}\n")
|
||||
|
||||
|
||||
|
||||
|
||||
logging.basicConfig(
|
||||
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO
|
||||
)
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
# This can be your own ID, or one for a developer group/channel.
|
||||
# You can use the /start command of this bot to see your chat id.
|
||||
DEVELOPER_CHAT_ID = 175042676
|
||||
|
||||
|
||||
def error_handler(update: Update, context: CallbackContext) -> None:
|
||||
"""Log the error and send a telegram message to notify the developer."""
|
||||
# Log the error before we do anything else, so we can see it even if something breaks.
|
||||
logger.error(msg="Exception while handling an update:", exc_info=context.error)
|
||||
|
||||
# traceback.format_exception returns the usual python message about an exception, but as a
|
||||
# list of strings rather than a single string, so we have to join them together.
|
||||
tb_list = traceback.format_exception(None, context.error, context.error.__traceback__)
|
||||
tb_string = ''.join(tb_list)
|
||||
|
||||
# Build the message with some markup and additional information about what happened.
|
||||
# You might need to add some logic to deal with messages longer than the 4096 character limit.
|
||||
message = (
|
||||
f'An exception was raised while handling an update\n'
|
||||
f'<pre>update = {html.escape(json.dumps(update.to_dict(), indent=2, ensure_ascii=False))}'
|
||||
'</pre>\n\n'
|
||||
f'<pre>context.chat_data = {html.escape(str(context.chat_data))}</pre>\n\n'
|
||||
f'<pre>context.user_data = {html.escape(str(context.user_data))}</pre>\n\n'
|
||||
f'<pre>{html.escape(tb_string)}</pre>'
|
||||
)
|
||||
|
||||
# Finally, send the message
|
||||
context.bot.send_message(chat_id=DEVELOPER_CHAT_ID, text=message, parse_mode=ParseMode.HTML)
|
||||
|
||||
|
||||
def start(update: Update, context: CallbackContext) -> None:
|
||||
update.effective_message.reply_html(
|
||||
'Use /bad_command to cause an error.\n'
|
||||
f'Your chat id is <code>{update.effective_chat.id}</code>.'
|
||||
)
|
||||
|
||||
|
||||
def main():
|
||||
# Create the Updater and pass it your bot's token.
|
||||
# Make sure to set use_context=True to use the new context based callbacks
|
||||
# Post version 12 this will no longer be necessary
|
||||
updater = Updater(TBOT_TOKEN, use_context=True)
|
||||
updater = Updater(BOT_TOKEN, use_context=True)
|
||||
|
||||
# Get the dispatcher to register handlers
|
||||
bot = updater.dispatcher
|
||||
dispatcher = updater.dispatcher
|
||||
|
||||
# on different commands - answer in Telegram
|
||||
bot.add_handler(CommandHandler("start", start))
|
||||
bot.add_handler(CommandHandler("help", help))
|
||||
|
||||
# on noncommand i.e message - echo the message on Telegram
|
||||
bot.add_handler(InlineQueryHandler(inlinequery))
|
||||
|
||||
# log all errors
|
||||
bot.add_error_handler(error)
|
||||
# Register the commands...
|
||||
dispatcher.add_handler(CommandHandler('start', start))
|
||||
dispatcher.add_handler(CommandHandler("help", help))
|
||||
dispatcher.add_handler(InlineQueryHandler(inlinequery))
|
||||
|
||||
# ...and the error handler
|
||||
dispatcher.add_error_handler(error_handler)
|
||||
|
||||
# Start the Bot
|
||||
updater.start_polling()
|
||||
|
||||
# Block until the user presses Ctrl-C or the process receives SIGINT,
|
||||
# Run the bot until you press Ctrl-C or the process receives SIGINT,
|
||||
# SIGTERM or SIGABRT. This should be used most of the time, since
|
||||
# start_polling() is non-blocking and will stop the bot gracefully.
|
||||
updater.idle()
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
main()
|
||||
Loading…
Reference in New Issue
Block a user