mirror of
https://github.com/SMGCommunity/Petari.git
synced 2024-11-24 05:59:43 +00:00
100 lines
2.7 KiB
Python
100 lines
2.7 KiB
Python
# setup.py -- by shibboleet
|
|
# Allows for the easy setup of RVL_SDK and its dependencies
|
|
|
|
import zipfile
|
|
import urllib.request
|
|
import os
|
|
import sys
|
|
import subprocess
|
|
import hashlib
|
|
|
|
def install(what):
|
|
subprocess.check_call([sys.executable, '-m', 'pip', 'install', what])
|
|
|
|
if not os.path.exists("baserom.dol"):
|
|
print("baserom.dol does not exist. Place a SMG1 dol named baserom.dol in the root.")
|
|
sys.exit(1)
|
|
else:
|
|
print("Found baserom.dol...")
|
|
|
|
try:
|
|
import capstone
|
|
print("Found capstone, continuing...")
|
|
except ModuleNotFoundError:
|
|
print("Module 'capstone' not installed. Installing...")
|
|
install("capstone")
|
|
|
|
try:
|
|
import elftools
|
|
print("Found elftools, continuing...")
|
|
except ModuleNotFoundError:
|
|
print("Module `elftools` not found. Installing...")
|
|
install("pyelftools")
|
|
|
|
try:
|
|
import ninja
|
|
print("Found ninja, continuing...")
|
|
except ModuleNotFoundError:
|
|
print("Module 'ninja' not found. Installing...")
|
|
install("ninja")
|
|
|
|
try:
|
|
import colorama
|
|
print("Found colorama, continuing...")
|
|
except ModuleNotFoundError:
|
|
print("Module 'colorama' not found. Installing...")
|
|
install("colorama")
|
|
|
|
try:
|
|
import pandas
|
|
print("Found pandas, continuing...")
|
|
except ModuleNotFoundError:
|
|
print("Module 'pandas' not found. Installing...")
|
|
install("pandas")
|
|
|
|
try:
|
|
import plotly
|
|
print("Found plotly, continuing...")
|
|
except ModuleNotFoundError:
|
|
print("Module 'plotly' not found. Installing...")
|
|
install("plotly")
|
|
|
|
try:
|
|
import git
|
|
print("Found GitPython, continuing...")
|
|
except ModuleNotFoundError:
|
|
print("Module 'GitPython' not installed. Installing...")
|
|
install("GitPython")
|
|
|
|
try:
|
|
import kaleido
|
|
print("Found kaleido, continuing...")
|
|
except ModuleNotFoundError:
|
|
print("Module 'kaleido' not installed. Installing...")
|
|
install("kaleido")
|
|
|
|
if not os.path.exists("Compilers"):
|
|
print("Compilers folder not found, downloading...")
|
|
|
|
with urllib.request.urlopen("http://shibbo.net/smg/Compilers.zip") as response, open("Compilers.zip", 'wb') as out:
|
|
data = response.read()
|
|
if hashlib.sha256(data).hexdigest().upper() != "12D439B707D7AECB1BCD5B321DA406310523357128ABADB1493BED212A4A225F":
|
|
print("Compilers.zip corrupt")
|
|
sys.exit(1)
|
|
out.write(data)
|
|
|
|
if os.path.exists("Compilers.zip"):
|
|
os.mkdir("Compilers")
|
|
print("Extracting file...")
|
|
with zipfile.ZipFile("Compilers.zip", "r") as zip:
|
|
zip.extractall("Compilers")
|
|
|
|
os.remove("Compilers.zip")
|
|
else:
|
|
print("Compilers.zip failed to download.")
|
|
sys.exit(1)
|
|
else:
|
|
print("Found compilers folder, continuing...")
|
|
|
|
print("Done.")
|
|
sys.exit(1) |