Update fetch_anime_and_play.py

web_programming/fetch_anime_and_play.py

@@ -1,5 +1,7 @@
+import os
+import subprocess
 import requests
-from bs4 import BeautifulSoup, NavigableString, Tag
+from bs4 import BeautifulSoup
 from fake_useragent import UserAgent
 
 BASE_URL = "https://ww1.gogoanime2.org"
@@ -39,7 +41,7 @@
 
     # get list of anime
     anime_ul = soup.find("ul", {"class": "items"})
     if anime_ul is None or isinstance(anime_ul, NavigableString):
         msg = f"Could not find and anime with name {anime_name}"
         raise ValueError(msg)
     anime_li = anime_ul.children
@@ -47,12 +49,12 @@
     # for each anime, insert to list. the name and url.
     anime_list = []
     for anime in anime_li:
         if isinstance(anime, Tag):
             anime_url = anime.find("a")
             if anime_url is None or isinstance(anime_url, NavigableString):
                 continue
             anime_title = anime.find("a")
             if anime_title is None or isinstance(anime_title, NavigableString):
                 continue
 
             anime_list.append({"title": anime_title["title"], "url": anime_url["href"]})
@@ -91,19 +93,19 @@
 
     # With this id. get the episode list.
     episode_page_ul = soup.find("ul", {"id": "episode_related"})
     if episode_page_ul is None or isinstance(episode_page_ul, NavigableString):
         msg = f"Could not find any anime eposiodes with name {anime_name}"
         raise ValueError(msg)
     episode_page_li = episode_page_ul.children
 
     episode_list = []
     for episode in episode_page_li:
         if isinstance(episode, Tag):
             url = episode.find("a")
             if url is None or isinstance(url, NavigableString):
                 continue
             title = episode.find("div", {"class": "name"})
             if title is None or isinstance(title, NavigableString):
                 continue
 
             episode_list.append(
@@ -154,6 +156,18 @@
     return [f"{BASE_URL}{episode_url}", f"{BASE_URL}{download_url}"]
 
 
+def download_video(download_url: str, output_filename: str) -> None:
+    """Download video using ffmpeg."""
+    command = ["ffmpeg", "-i", download_url, output_filename]
+    try:
+        subprocess.run(command, check=True)
+    except FileNotFoundError:
+        # If ffmpeg is not found, instruct the user on how to install it
+        print("Error: ffmpeg is not installed.")
+        print("Please install ffmpeg using the following command:")
+        print("brew install ffmpeg")  # This is the command for macOS
+
+
 if __name__ == "__main__":
     anime_name = input("Enter anime name: ").strip()
     anime_list = search_scraper(anime_name)
@@ -186,3 +200,21 @@
             episode_url, download_url = get_anime_episode(chosen_episode["url"])
             print(f"\nTo watch, ctrl+click on {episode_url}.")
             print(f"To download, ctrl+click on {download_url}.")
+
+            # Add an option to download or not
+            download_choice = (
+                input("\nDo you want to download this episode? (yes/no): ")
+                .strip()
+                .lower()
+            )
+            if download_choice in ["yes", "y"]:
+                output_filename = f"{chosen_anime['title']} - {chosen_episode['title']}.mp4"  # Change extension as needed
+                download_video(download_url, output_filename)
+                print(
+                    f"{chosen_episode['title']} has been downloaded as {output_filename}."
+                )
+            else:
+                print("Download skipped.")
+
+            # if error download please install ffmeg
+            # brew install ffmpeg for mac