yolando/music-iyoyah-bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix final

Browse Source
This commit is contained in:
YOLANDO
2026-03-25 16:05:26 +07:00
parent b5b6135cc3
commit c710a9bdd5
5 changed files with 105 additions and 174 deletions
Download Patch File Download Diff File Expand all files Collapse all files

134
yt_engine.py
View File

@@ -2,150 +2,94 @@ import yt_dlp
import os
import asyncio
# Buat folder penampungan sementara jika belum ada
DOWNLOAD_DIR = "downloads"
os.makedirs(DOWNLOAD_DIR, exist_ok=True)
def generate_netscape_cookies():
"""
Mengubah data tabel cookie (copy-paste dari tab Application browser)
menjadi format baku Netscape HTTP Cookie agar bisa dibaca yt-dlp.
"""
if not os.path.exists("raw_cookie.txt"):
print("⚠️ raw_cookie.txt tidak ditemukan. Lanjut tanpa cookies.")
return False
try:
with open("raw_cookie.txt", "r", encoding="utf-8") as f:
lines = f.readlines()
with open("cookies.txt", "w", encoding="utf-8") as f:
f.write("# Netscape HTTP Cookie File\n")
f.write("# This is a generated file! Do not edit.\n\n")
for line in lines:
# Pisahkan kolom berdasarkan tombol Tab (\t) bawaan copy-paste tabel
parts = line.split('\t')
# Pastikan baris tersebut valid (minimal punya Nama dan Value)
if len(parts) >= 2:
name = parts[0].strip()
value = parts[1].strip()
# Abaikan jika yang ter-copy adalah Header tabelnya ("Name", "Value")
if name.lower() == "name" or not name:
continue
# Format baku Netscape: domain, subdomains, path, secure, expiration, name, value
# Kita paksa domainnya .youtube.com agar yt-dlp bisa membacanya
name, value = parts[0].strip(), parts[1].strip()
if name.lower() == "name" or not name: continue
f.write(f".youtube.com\tTRUE\t/\tTRUE\t2147483647\t{name}\t{value}\n")
print("✅ Cookies berhasil di-generate dari raw_cookie.txt")
return True
except Exception as e:
print(f"❌ Error saat memproses cookies: {e}")
print(f"Error parsing cookies: {e}")
return False
def fetch_recommendations_sync(video_id: str, limit: int = 5):
ydl_opts_recs = {
'quiet': True,
'extract_flat': True,
'playlistend': limit + 2,
'extractor_args': {'youtube': ['player_client=android,ios']},
}
if os.path.exists("cookies.txt"):
ydl_opts_recs['cookiefile'] = 'cookies.txt'
try:
with yt_dlp.YoutubeDL(ydl_opts_recs) as ydl:
mix_url = f"https://www.youtube.com/watch?v={video_id}&list=RD{video_id}"
info = ydl.extract_info(mix_url, download=False)
recs = []
if 'entries' in info:
for entry in info['entries'][1:]:
if entry and entry.get('id') and entry.get('title'):
recs.append({'id': entry['id'], 'title': entry['title']})
return recs
except Exception as e:
return []
async def get_recommendations(video_id: str):
return await asyncio.to_thread(fetch_recommendations_sync, video_id)
def search_and_download(query_or_url: str):
"""
Fungsi utama untuk mencari dan mengunduh audio dari YouTube.
Berjalan secara sinkronus.
"""
# 1. Konversi cookie setiap kali mau download
has_cookies = generate_netscape_cookies()
# 2. Pengaturan yt-dlp
ydl_opts = {
'format': 'bestaudio/best/m4a/mp3',
# ❌ FFMPEG POSTPROCESSORS DIHAPUS ❌
# Langsung tembak format asli m4a (AAC) agar CPU tidak tersiksa
'format': 'bestaudio[ext=m4a]/bestaudio/best',
'outtmpl': f'{DOWNLOAD_DIR}/%(id)s.%(ext)s',
'max_filesize': 45000000, # Batas aman Telegram 45MB
'postprocessors': [{
'key': 'FFmpegExtractAudio',
'preferredcodec': 'mp3',
'preferredquality': '128',
}],
'max_filesize': 45000000,
'quiet': True,
'noplaylist': True,
'default_search': 'ytsearch1',
'js_runtimes': {'node': {}},
# 'extractor_args': {'youtube': ['player_client=android,ios']},
'extractor_args': {'youtube': ['player_client=android,ios']},
'js_runtimes': {'node': {}},
}
# 3. Sisipkan cookie jika berhasil digenerate
if has_cookies and os.path.exists("cookies.txt"):
ydl_opts['cookiefile'] = 'cookies.txt'
try:
with yt_dlp.YoutubeDL(ydl_opts) as ydl:
# Mulai proses ekstraksi & download
info = ydl.extract_info(query_or_url, download=True)
# Jika hasil berupa playlist/pencarian, ambil item pertama
if 'entries' in info:
info = info['entries'][0]
video_id = info.get('id')
title = info.get('title')
duration = info.get('duration')
file_path = f"{DOWNLOAD_DIR}/{video_id}.mp3"
# Ambil ekstensi aslinya (kemungkinan besar m4a)
ext = info.get('ext', 'm4a')
file_path = f"{DOWNLOAD_DIR}/{video_id}.{ext}"
return {
"status": "success",
"video_id": video_id,
"title": title,
"duration": duration,
"file_path": file_path if os.path.exists(file_path) else None
"file_path": file_path if os.path.exists(file_path) else None,
"ext": ext
}
except Exception as e:
return {"status": "error", "message": str(e)}
async def process_youtube_request(query: str):
"""
Bungkus (Wrapper) Async agar bot Telegram tidak nge-hang
saat menunggu proses download selesai.
"""
return await asyncio.to_thread(search_and_download, query)
# ==========================================
# FITUR TAMBAHAN: YOUTUBE MIX RECOMMENDATION
# ==========================================
def fetch_recommendations_sync(video_id: str, limit: int = 5):
"""
Mengambil rekomendasi lagu dari YouTube Mix algoritma (Vibes/Genre serupa)
"""
# Gunakan settingan ringan karena kita cuma mau ambil teks judul, BUKAN download
ydl_opts_recs = {
'quiet': True,
'extract_flat': True,
'playlistend': limit + 1, # +1 karena urutan pertama biasanya lagu aslinya
'extractor_args': {'youtube': ['player_client=android,ios']},
}
# Sisipkan cookie jika ada
if os.path.exists("cookies.txt"):
ydl_opts_recs['cookiefile'] = 'cookies.txt'
try:
with yt_dlp.YoutubeDL(ydl_opts_recs) as ydl:
# RD = Playlist YouTube Mix khusus untuk video tersebut
mix_url = f"https://www.youtube.com/watch?v={video_id}&list=RD{video_id}"
info = ydl.extract_info(mix_url, download=False)
recs = []
if 'entries' in info:
# Loop dan ambil lagunya (Lewati index 0 karena itu lagu yg sedang diputar)
for entry in info['entries'][1:]:
if entry and entry.get('id') and entry.get('title'):
recs.append({
'id': entry['id'],
'title': entry['title']
})
return recs
except Exception as e:
print(f"Gagal mengambil rekomendasi: {e}")
return []
async def get_recommendations(video_id: str):
"""Bungkus Async agar bot tidak hang"""
return await asyncio.to_thread(fetch_recommendations_sync, video_id)