web_dat.py

import urllib.request
import re
import youtube_dl
import os
from rich.console import Console
from pl_song import GenFunc

class WebDat:

    def __init__(self) -> None:
        self.console = Console()
        self.main_funcs = GenFunc()


    def __search_input_constructor(self, search:str) -> int:
        """This function takes no argument and returns a search string."""

        if len(search.split("_")) != 1:
            search = search.split("_")
            return "+".join(search)
        
        else: return search


    def __search_yt(self, query:str) -> str:
        """This function searches that query on youtube and returns the url of top result"""
        if query != "":
            html = urllib.request.urlopen(f"https://www.youtube.com/results?search_query={query}")
            video_ids = re.findall(r"watch\?v=(\S{11})",html.read().decode())
            return f"https://youtube.com/watch?v={video_ids[0]}" # Returning top resulted video.


    def __download_mp_file(self, link:str) -> str:
        """This functin downloads the given videos's youtube link in .mp3 format"""

        # Changing pwd as we want to download music files in another repo not in current one.
        
        os.chdir(r"C:\Users\Aditya\Desktop\notcurr\bron")

        video_info = youtube_dl.YoutubeDL().extract_info(
            url = link, download = False
        )
        
        ydl_opts = {
                'format': 'bestaudio/best',
                'keepvideo': False,
                'outtmpl': f"{video_info['title']}.mp3"
        }
        with youtube_dl.YoutubeDL(ydl_opts) as ydl:
            ydl.download([video_info['webpage_url']])

        return f"{video_info['title']}.mp3".replace("|", "#")


    def __play_down_music(self, a:str, type:str, filename:str) -> None:
        """This function plays the song which is downloaded from youtube as per user's search"""

        if type == 'g': os.startfile(filename)
        else:
            self.main_funcs.play_song(filename)


    def search_n_play_song(self, type:str, search:str):
        """This function searches the song name and download and play it in terminal or through windows music player
        as per the type argument."""

        a = self.__search_input_constructor(search)         # Preparing input for search.
        self.main_funcs.animation()
        song = self.__download_mp_file(self.__search_yt(a)) # Downloading song.

        # Playing that song.
        if type == 't' or type == 'g':
            self.__play_down_music(a, type, song)
        else:
            raise Exception("type argument must be 't' or 'g'")