Scripts/vimm/vimm.py

120 lines
3.0 KiB
Python
Raw Normal View History

2023-12-02 23:40:27 -04:00
import json
import sys
import logging
import requests
from enum import Enum
from bs4 import BeautifulSoup
# User Agent
user_agent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:120.0) Gecko/20100101 Firefox/120.0"
# Referer
referer = "https://vimm.net/"
# Download URL
download_url2 = "https://download2.vimm.net/download/"
# Enum for type of download,
class WiiType(Enum):
wbfs = 0
rvz = 1
def get_media_id(page_content):
# Could be used later to be a method to extract the id from passed in json
pass
def download(url, filetype):
page_request = requests.get(url, headers={"User-Agent": user_agent})
page_content = BeautifulSoup(page_request.content, 'html.parser')
# Find div with the class "sectionTitle"
section_titles = page_content.find_all("div", class_="sectionTitle")
section_title = section_titles[0].text
# TODO: use match statement
if section_title == "Wii":
if filetype == "wbfs":
download_wii(page_content, WiiType.wbfs)
elif filetype == "rvz":
download_wii(page_content, WiiType.rvz)
def download_wii(page_content, filetype):
# Wish there was a better way to do this, manually parsing javascript for a variable
# Find script that contains 'var media = '
scripts = page_content.find_all("script")
script = str
# Find script that contains 'var media = '
for i in scripts:
if "var media = " in i.text:
script = i.string
media = list()
# This is the default usage, pycharm bug
# noinspection PyArgumentList
for lines in script.splitlines():
if "var media = " in lines:
media.append(lines)
for i in range(0, len(media)):
media[i] = media[i].replace(" var media = ", "")
media[i] = media[i].replace(";", "")
# Parse lines as json
media_json = list()
for i in media:
media_json.append(json.loads(i))
# Sort by version
media_json.sort(key=lambda x: x["Version"])
# Get ID of last entry
last_id = media_json[-1]["ID"]
logging.log(logging.INFO, "File ID of Download: " + str(last_id))
# Get the title of the last entry
filename = media_json[-1]["GoodTitle"].replace(".iso", ".7z")
logging.log(logging.INFO, "File Name of Download: " + filename)
# TODO: Parse json for file size
chunk_size = 1024 * 1024 * 10 # 10 MB
# Build request
request = requests.Session()
request.headers.update({"User-Agent": user_agent, "Referer": referer})
request.params.update({"mediaId": last_id})
if filetype == WiiType.rvz:
request.params.update({"alt": filetype.value})
r = request.get(download_url2, stream=True)
with open(filename, "wb") as file:
for chunk in r.iter_content(chunk_size=chunk_size):
if chunk:
file.write(chunk)
def main():
# Comment out to disable logging
logging.getLogger().setLevel(logging.INFO)
if len(sys.argv) != 3:
print("Usage: python vimm.py <url> <filetype>")
exit(1)
# Get arguments
url = sys.argv[1]
filetype = sys.argv[2]
download(url, filetype)
if __name__ == "__main__":
main()