|
|
from flask import Flask, render_template, jsonify, request |
|
|
from ytmusicapi import YTMusic |
|
|
import os |
|
|
import logging |
|
|
import requests |
|
|
from datetime import datetime, timedelta |
|
|
import time |
|
|
import asyncio |
|
|
|
|
|
app = Flask(__name__) |
|
|
ytmusic = YTMusic() |
|
|
|
|
|
|
|
|
|
|
|
logging.basicConfig(level=logging.INFO) |
|
|
logger = logging.getLogger(__name__) |
|
|
|
|
|
@app.route('/') |
|
|
def index(): |
|
|
return render_template('index.html') |
|
|
|
|
|
@app.route('/search', methods=['POST']) |
|
|
def search(): |
|
|
query = request.json.get('query', '') |
|
|
search_results = ytmusic.search(query, filter="songs") |
|
|
return jsonify(search_results) |
|
|
|
|
|
@app.route('/searcht', methods=['POST']) |
|
|
def searcht(): |
|
|
query = request.json.get('query', '') |
|
|
search_results = ytmusic.search(query, filter="songs") |
|
|
first_song = next((song for song in search_results if 'videoId' in song and song['videoId']), {}) if search_results else {} |
|
|
return jsonify(first_song) |
|
|
|
|
|
async def get_track_download_url(track_id: str) -> str: |
|
|
apis = [ |
|
|
"https://cobalt-api.ayo.tf", |
|
|
"http://34.107.254.11" |
|
|
] |
|
|
|
|
|
for api_url in apis: |
|
|
try: |
|
|
logger.info(f"Attempting to get download URL from: {api_url}") |
|
|
y_url = f"https://youtu.be/{track_id}" |
|
|
response = requests.post(api_url, timeout=20, json={"url": y_url}, headers={"Accept": "application/json"}) |
|
|
logger.info(f"Response status: {response.status_code}") |
|
|
logger.info(f"Response content: {response.content}") |
|
|
|
|
|
if response.status_code == 200: |
|
|
download_url = response.json().get("url") |
|
|
if download_url: |
|
|
logger.info(f"Successfully obtained download URL from {api_url}") |
|
|
return download_url |
|
|
except requests.exceptions.RequestException as e: |
|
|
logger.warning(f"Failed to get download URL from {api_url}: {str(e)}") |
|
|
continue |
|
|
|
|
|
logger.error(f"No download URL found") |
|
|
raise Exception("Download URL not found") |
|
|
|
|
|
@app.route('/track_dl', methods=['POST']) |
|
|
async def track_dl(): |
|
|
track_id = request.json.get('track_id', '') |
|
|
dl_url = await get_track_download_url() |
|
|
result = {"url": dl_url} |
|
|
return jsonify(result) |
|
|
|
|
|
|
|
|
|
|
|
@app.route('/get_artist', methods=['GET']) |
|
|
def get_artist(): |
|
|
artist_id = request.args.get('id') |
|
|
artist_info = ytmusic.get_artist(artist_id) |
|
|
return jsonify(artist_info) |
|
|
|
|
|
@app.route('/get_album', methods=['GET']) |
|
|
def get_album(): |
|
|
album_id = request.args.get('id') |
|
|
album_info = ytmusic.get_album(album_id) |
|
|
return jsonify(album_info) |
|
|
|
|
|
@app.route('/get_song', methods=['GET']) |
|
|
def get_song(): |
|
|
song_id = request.args.get('id') |
|
|
song_info = ytmusic.get_song(song_id) |
|
|
return jsonify(song_info) |
|
|
|
|
|
if __name__ == '__main__': |
|
|
app.run(host='0.0.0.0', port=7860) |
|
|
|
