From fe29eb47e7b554ebe605661a1500e6df7d66bb8a Mon Sep 17 00:00:00 2001 From: david Date: Thu, 14 Jan 2021 14:59:07 -0600 Subject: [PATCH] Add YTM, newer and better error handling --- Pipfile | 1 + music.py | 122 +++++++++++++++++++++++++++++++++++-------------------- 2 files changed, 78 insertions(+), 45 deletions(-) diff --git a/Pipfile b/Pipfile index 8a79d34..ea043e6 100644 --- a/Pipfile +++ b/Pipfile @@ -13,6 +13,7 @@ dpath = "*" gmusicapi = "*" pathlib = "*" uuid = "*" +ytmusicapi = "*" [requires] python_version = "3.6" diff --git a/music.py b/music.py index 3949bcb..60727cd 100755 --- a/music.py +++ b/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'
update = {html.escape(json.dumps(update.to_dict(), indent=2, ensure_ascii=False))}'
+		'
\n\n' + f'
context.chat_data = {html.escape(str(context.chat_data))}
\n\n' + f'
context.user_data = {html.escape(str(context.user_data))}
\n\n' + f'
{html.escape(tb_string)}
' + ) + + # 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 {update.effective_chat.id}.' + ) 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() \ No newline at end of file