Create main.py
This commit is contained in:
kromskii2 authored and GitHub committed
1 parent
3e722fb700
commit
15aedcfb0b
1 file changed
+164
@@ -0,0 +1,164 @@
|
||||
#!/usr/bin/env python3
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
╔═══════════════════════════════════════════════════════════════╗
|
||||
║ 🤖 YANDEX MUSIC DOWNLOADER v2.0 "MARK XLII" ║
|
||||
║ Designed by Pavel • Powered by J.A.R.V.I.S. core ║
|
||||
║ "Sometimes you gotta run before you can walk." — T. Stark ║
|
||||
╚═══════════════════════════════════════════════════════════════╝
|
||||
"""
|
||||
import asyncio
|
||||
import webbrowser
|
||||
import sys
|
||||
from pathlib import Path
|
||||
from yandex_music import ClientAsync
|
||||
|
||||
# ─────────────────────────────────────────────────────────────
|
||||
# 🎨 TONY STARK UI UTILS
|
||||
# ─────────────────────────────────────────────────────────────
|
||||
class StarkUI:
|
||||
"""Интерфейс в стиле голографического дисплея Тони Старка"""
|
||||
|
||||
COLORS = {
|
||||
'cyan': '\033[96m', 'blue': '\033[94m', 'green': '\033[92m',
|
||||
'yellow': '\033[93m', 'red': '\033[91m', 'bold': '\033[1m',
|
||||
'reset': '\033[0m'
|
||||
}
|
||||
|
||||
@staticmethod
|
||||
def banner():
|
||||
return f"""
|
||||
{StarkUI.COLORS['cyan']}
|
||||
╔═══════════════════════════════════════════════════════════════╗
|
||||
║ 🤖 YANDEX MUSIC DOWNLOADER v2.0 "MARK XLII" ║
|
||||
║ ⚡ Just another Tuesday in the workshop ║
|
||||
║ 🛠️ Powered by J.A.R.V.I.S. core • {StarkUI.COLORS['yellow']}Pavel Edition{StarkUI.COLORS['reset']}
|
||||
╚═══════════════════════════════════════════════════════════════╝
|
||||
{StarkUI.COLORS['reset']}"""
|
||||
|
||||
@staticmethod
|
||||
def status(msg, color='cyan'):
|
||||
print(f"{StarkUI.COLORS.get(color, '')}[✦] {msg}{StarkUI.COLORS['reset']}", flush=True)
|
||||
|
||||
@staticmethod
|
||||
def success(msg):
|
||||
print(f"{StarkUI.COLORS['green']}[✓] {msg}{StarkUI.COLORS['reset']}", flush=True)
|
||||
|
||||
@staticmethod
|
||||
def warning(msg):
|
||||
print(f"{StarkUI.COLORS['yellow']}[⚠] {msg}{StarkUI.COLORS['reset']}", flush=True)
|
||||
|
||||
@staticmethod
|
||||
def error(msg):
|
||||
print(f"{StarkUI.COLORS['red']}[✗] {msg}{StarkUI.COLORS['reset']}", flush=True)
|
||||
|
||||
@staticmethod
|
||||
def progress(current, total, status="Processing"):
|
||||
percent = (current / total) * 100
|
||||
bar = "█" * int(percent // 5) + "░" * (20 - int(percent // 5))
|
||||
print(f"\r{StarkUI.COLORS['blue']}[{bar}] {percent:.1f}% {status}{StarkUI.COLORS['reset']}", end="", flush=True)
|
||||
|
||||
# ─────────────────────────────────────────────────────────────
|
||||
# 🔗 КОНСТАНТЫ
|
||||
# ─────────────────────────────────────────────────────────────
|
||||
TOKEN_HELP_URL = "https://chromewebstore.google.com/detail/yandex-music-token/lcbjeookjibfhjjopieifgjnhlegmkib?hl=ru&utm_source=ext_sidebar"
|
||||
|
||||
# ─────────────────────────────────────────────────────────────
|
||||
# 🎵 CORE LOGIC
|
||||
# ─────────────────────────────────────────────────────────────
|
||||
async def download_track(client: ClientAsync, short_track, folder: Path):
|
||||
"""Загрузка трека. Быстро. Эффектно. Без компромиссов."""
|
||||
try:
|
||||
full_track = await short_track.fetch_track_async()
|
||||
artist = full_track.artists[0].name if full_track.artists else "Unknown Artist"
|
||||
title = full_track.title
|
||||
|
||||
# 🛡️ Фильтр "кривых" символов — Тони не любит баги в путях
|
||||
safe = lambda s: "".join(c for c in s if c not in r'<>:"/\|?*').strip()
|
||||
filename = folder / f"{safe(artist)} - {safe(title)}.mp3"
|
||||
|
||||
await full_track.download_async(filename=str(filename))
|
||||
return True, f"{artist} - {title}"
|
||||
except Exception as e:
|
||||
return False, str(e)
|
||||
|
||||
async def main(TOKEN, DOWNLOAD_DIR):
|
||||
StarkUI.status("Инициализация протоколов связи...", "blue")
|
||||
DOWNLOAD_DIR.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
client = await ClientAsync(TOKEN).init()
|
||||
StarkUI.success("Авторизация пройдена. Добро пожаловать, сэр.")
|
||||
|
||||
status = await client.account_status()
|
||||
StarkUI.status(f"Пользователь: {status.account.login}", "cyan")
|
||||
plus_status = "активен" if status.plus.has_plus else "не активен (качество будет стандартным)"
|
||||
StarkUI.status(f"Яндекс.Плюс: {plus_status}", "green" if status.plus.has_plus else "yellow")
|
||||
|
||||
# 🎛️ Настройки (можно вынести в конфиг позже)
|
||||
USE_LIKES_PLAYLIST = True
|
||||
USER_ID, PLAYLIST_KIND = "12345678", "567890"
|
||||
|
||||
StarkUI.status("Загрузка метаданных плейлиста...", "blue")
|
||||
if USE_LIKES_PLAYLIST:
|
||||
tracks_list = await client.users_likes_tracks()
|
||||
StarkUI.success("Плейлист 'Мне нравится' получен")
|
||||
else:
|
||||
playlist = await client.users_playlists(kind=PLAYLIST_KIND, user_id=USER_ID)
|
||||
if not playlist:
|
||||
StarkUI.error("Плейлист не найден. Проверьте координаты, сэр.")
|
||||
return
|
||||
tracks_list = playlist[0]
|
||||
|
||||
tracks = tracks_list.tracks if hasattr(tracks_list, 'tracks') else []
|
||||
StarkUI.status(f"Обнаружено треков: {len(tracks)}. Начинаем загрузку...", "cyan")
|
||||
print()
|
||||
|
||||
success, failed = 0, 0
|
||||
for i, short_track in enumerate(tracks, 1):
|
||||
StarkUI.progress(i, len(tracks), f"Track {i}/{len(tracks)}")
|
||||
ok, result = await download_track(client, short_track, DOWNLOAD_DIR)
|
||||
if ok:
|
||||
success += 1
|
||||
else:
|
||||
failed += 1
|
||||
StarkUI.warning(f"Пропуск: {result[:50]}...")
|
||||
await asyncio.sleep(0.25) # Не будим спящих драконов Яндекса
|
||||
|
||||
print()
|
||||
StarkUI.success(f"Миссия выполнена! ✅ Успешно: {success} | Пропущено: {failed}")
|
||||
StarkUI.status(f"Файлы сохранены в: {DOWNLOAD_DIR}", "green")
|
||||
await client.close()
|
||||
|
||||
# ─────────────────────────────────────────────────────────────
|
||||
# 🚀 ENTRY POINT
|
||||
# ─────────────────────────────────────────────────────────────
|
||||
if __name__ == "__main__":
|
||||
print(StarkUI.banner())
|
||||
|
||||
# 🌐 Открываем браузер для получения токена — Тони ценит ваше время
|
||||
StarkUI.status("Открываю панель получения токена в браузере...", "cyan")
|
||||
webbrowser.open(TOKEN_HELP_URL)
|
||||
print(f"{StarkUI.COLORS['yellow']}💡 Совет: скопируйте токен и вернитесь сюда. Я подожду.{StarkUI.COLORS['reset']}\n")
|
||||
|
||||
# 🔐 Ввод токена
|
||||
TOKEN = input(f"{StarkUI.COLORS['bold']}🔐 Введите токен доступа:{StarkUI.COLORS['reset']} ").strip()
|
||||
if not TOKEN or len(TOKEN) < 20:
|
||||
StarkUI.error("Токен выглядит подозрительно. Проверьте, сэр.")
|
||||
input(f"\n{StarkUI.COLORS['yellow']}Нажмите Enter для выхода...{StarkUI.COLORS['reset']}")
|
||||
sys.exit(1)
|
||||
|
||||
# 📁 Ввод пути
|
||||
folder_input = input(f"{StarkUI.COLORS['bold']}📁 Путь для сохранения (Enter для C:\\mus):{StarkUI.COLORS['reset']} ").strip()
|
||||
DOWNLOAD_DIR = Path(folder_input if folder_input else r"C:\mus")
|
||||
|
||||
# 🚀 Запуск
|
||||
try:
|
||||
asyncio.run(main(TOKEN, DOWNLOAD_DIR))
|
||||
except KeyboardInterrupt:
|
||||
print(f"\n{StarkUI.COLORS['yellow']}⏹️ Прервано по команде пользователя. Как скажете.{StarkUI.COLORS['reset']}")
|
||||
except Exception as e:
|
||||
StarkUI.error(f"Критический сбой: {e}")
|
||||
StarkUI.warning("Рекомендация: проверьте соединение и попробуйте снова, сэр.")
|
||||
finally:
|
||||
print(f"\n{StarkUI.COLORS['green']}🎯 Система готова к следующей операции.{StarkUI.COLORS['reset']}")
|
||||
input(f"{StarkUI.COLORS['bold']}Нажмите Enter для завершения...{StarkUI.COLORS['reset']}")
|
||||
Reference in new issue
Block a user