WiiPy/modules/title.py

56 lines
2.0 KiB
Python

# "modules/title.py" from WiiPy by NinjaCheetah
# https://github.com/NinjaCheetah/WiiPy
import binascii
import re
from modules.core import fatal_error
def title_edit_ios(new_ios: str) -> str:
# Setting a new required IOS.
try:
new_ios = int(new_ios)
except ValueError:
fatal_error("The specified IOS is not valid! The new IOS should be a valid integer.")
if new_ios < 3 or new_ios > 255:
fatal_error("The specified IOS is not valid! The new IOS version should be between 3 and 255.")
new_ios_tid = f"00000001{new_ios:08X}"
return new_ios_tid
def title_edit_tid(tid: str, new_tid: str) -> str:
# Setting a new TID, only changing TID low since this expects a 4 character input with letters, numbers, and some
# symbols.
pattern = r"^[a-z0-9!@#$%^&*]{4}$"
if not re.fullmatch(pattern, new_tid, re.IGNORECASE):
fatal_error(f"The specified Title ID is not valid! The new Title ID should be 4 characters and only include "
f"letters, numbers, and the special characters \"!@#$%&*\".")
# Get the current TID high, because we want to preserve the title type while only changing the TID low.
tid_high = tid[:8]
new_tid = f"{tid_high}{str(binascii.hexlify(new_tid.encode()), 'ascii')}"
return new_tid
def title_edit_type(tid: str, new_type: str) -> str:
# Setting a new title type.
new_tid_high = None
match new_type:
case "System":
new_tid_high = "00000001"
case "Channel":
new_tid_high = "00010001"
case "SystemChannel":
new_tid_high = "00010002"
case "GameChannel":
new_tid_high = "00010004"
case "DLC":
new_tid_high = "00010005"
case "HiddenChannel":
new_tid_high = "00010008"
case _:
fatal_error("The specified type is not valid! The new type must be one of: System, Channel, "
"SystemChannel, GameChannel, DLC, HiddenChannel.")
tid_low = tid[8:]
new_tid = f"{new_tid_high}{tid_low}"
return new_tid