aly / romdl.py

from cursesmenu import CursesMenu from cursesmenu.items import SubmenuItem, FunctionItem import requests import xmltodict from bs4 import BeautifulSoup as Soup from sys import stderr import os from urllib.parse import urlparse, unquote system_data = xmltodict.parse(requests.get("https://ia801407.us.archive.org/23/items/no-intro-rom-sets/no-intro-rom-sets_files.xml").text) archive_path = "https://ia801407.us.archive.org/view_archive.php?archive=/23/items/no-intro-rom-sets/" def download_file(url): if url[0] != "h": url = "https:" + url r = requests.get(url, stream=True) if r.status_code == 200: with open(os.path.basename(unquote(urlparse(url).path)), "wb") as f: for chunk in r: print("*") f.write(chunk) def search_submenu(filelist): term = input("Enter search term") results = [] for row in filelist.findAll("tr")[1:]: dest_fn = row.find("td") if ".zip" in dest_fn.text: if term.lower() in dest_fn.text.split(".zip")[0].lower(): results += (dest_fn.text.split(".zip")[0], dest_fn.find("a")["href"]) search_menu = CursesMenu("Search", "") for title, url in results: item = FunctionItem(title, download_file, [url]) search_menu.append_item(item) search_menu.show() def generate_submenu(filename): filelist = Soup(requests.get(archive_path + filename).text, "html.parser").find("table", {"class": "archext"}) system_menu = CursesMenu(filename, "Select a file to download") # system_menu.append_item(FunctionItem("Search...", search_submenu, [filelist])) for row in filelist.findAll("tr")[1:]: dest_fn = row.find("td") if ".zip" in dest_fn.text: item = FunctionItem(dest_fn.text.split(".zip")[0], download_file, [dest_fn.find("a")["href"]]) system_menu.append_item(item) system_menu.show() def main(): systems_menu = CursesMenu("System Menu", "Select the system to browse roms") for system in system_data["files"]["file"]: if system["@name"][-4:] == ".zip": item = FunctionItem(system["@name"], generate_submenu, [system["@name"]]) systems_menu.append_item(item) systems_menu.show() if __name__ == "__main__": main()