mirror of https://gitlab.com/Nanolx/qwad.git
193 lines
7.6 KiB
Python
193 lines
7.6 KiB
Python
#!/usr/bin/env python
|
|
#coding=utf-8
|
|
import sys, os
|
|
from optparse import OptionParser
|
|
from optparse import Option, OptionValueError
|
|
from PyQt4.QtGui import QApplication
|
|
from PyQt4.QtCore import QTranslator, QString, QLocale
|
|
from GUI.VenPri import MWQwad, nusDownloadingCLI, PackingCLI, UnpackingCLI, ShowTMD
|
|
from TitleIDs import TitleDict, IOSdict, swap_dic, ChannelCLIDict, ChannelJAPDict, ChannelPALDict, ChannelUSADict, ChannelJAPVerDict, ChannelPALVerDict, ChannelUSAVerDict
|
|
|
|
if sys.platform.startswith('linux'):
|
|
if os.getenv("QWAD_X_DIR"):
|
|
os.chdir(os.getenv("QWAD_X_DIR"))
|
|
else:
|
|
os.chdir(os.getenv("HOME"))
|
|
|
|
class MultipleOption(Option):
|
|
ACTIONS = Option.ACTIONS + ("extend",)
|
|
STORE_ACTIONS = Option.STORE_ACTIONS + ("extend",)
|
|
TYPED_ACTIONS = Option.TYPED_ACTIONS + ("extend",)
|
|
ALWAYS_TYPED_ACTIONS = Option.ALWAYS_TYPED_ACTIONS + ("extend",)
|
|
|
|
def take_action(self, action, dest, opt, value, values, parser):
|
|
if action == "extend":
|
|
values.ensure_value(dest, []).append(value)
|
|
else:
|
|
Option.take_action(self, action, dest, opt, value, values, parser)
|
|
|
|
VERSION = '0.9'
|
|
|
|
def opts():
|
|
description = """NUS-Downloader, WadManager and TMD-Viewer for Linux"""
|
|
parser = OptionParser(option_class=MultipleOption,
|
|
usage='usage: qwad [OPTIONS] ARGUMENT',
|
|
description=description)
|
|
parser.add_option('-d', '--download',
|
|
action="extend", type="string",
|
|
dest='download',
|
|
metavar='Arguments',
|
|
help="IOS Version Output Decrypt Pack")
|
|
parser.add_option('-u', "--unpack", dest="unpack", action="extend",
|
|
type="string", metavar='Arguments', help="unpack a WAD")
|
|
parser.add_option('-p', "--pack", dest="pack", action="extend",
|
|
type="string", metavar='Arguments', help="pack folder as WAD")
|
|
parser.add_option('-g', "--getversions", dest="getversions",
|
|
action="store_true", default=False, help="get available versions for IOS")
|
|
parser.add_option('-c', "--convert", dest="convert",
|
|
action="store_true", default=False, help="convert between IOSxx and hex-value")
|
|
parser.add_option('-t', "--tmdinfo", default=False, dest="tmdinfo", action="store_true",
|
|
help="Show infos provided by TMD file")
|
|
parser.add_option("-v", "--version", dest="version",
|
|
action="store_true", default=False, help="print version and exit")
|
|
|
|
options, args = parser.parse_args()
|
|
|
|
if options.version:
|
|
print "%s" % VERSION
|
|
sys.exit(0)
|
|
|
|
if options.download:
|
|
if args[0] is "0":
|
|
args[0] = None
|
|
if sys.platform.startswith('linux'):
|
|
if args[1].startswith('./'):
|
|
wadname = str(args[1][2:])
|
|
args[1] = os.getcwd() + '/' + wadname
|
|
elif not args[1].startswith('/'):
|
|
wadname = args[1]
|
|
args[1] = os.getcwd() + '/' + wadname
|
|
if "." in str(options.download[0]):
|
|
dld = ChannelCLIDict[str(options.download[0][:-4])]
|
|
reg = str(options.download[0][-3:])
|
|
if "pal" in reg or "PAL" in reg:
|
|
xarg = ChannelPALDict[str(dld)]
|
|
elif "usa" in reg or "USA" in reg:
|
|
xarg = ChannelUSADict[str(dld)]
|
|
elif "jap" in reg or "JAP" in reg:
|
|
xarg = ChannelJAPDict[str(dld)]
|
|
else:
|
|
print "specified region %s is invalid" % reg
|
|
sys.exit(1)
|
|
nusdow = nusDownloadingCLI(int(str(xarg).lower(),16), args[0], args[1], args[2], args[3])
|
|
elif "IOS" in str(options.download[0]) or "BC" in str(options.download[0]) or "MIOS" in str(options.download[0]):
|
|
xarg = TitleDict[str(options.download[0])]
|
|
nusdow = nusDownloadingCLI(int(str(xarg).lower(),16), args[0], args[1], args[2], args[3])
|
|
else:
|
|
nusdow = nusDownloadingCLI(int(str(options.download[0]).lower(),16), args[0], args[1], args[2], args[3])
|
|
nusdow.start()
|
|
sys.exit(0)
|
|
|
|
if options.getversions:
|
|
if "." in str(args[0]):
|
|
dld = ChannelCLIDict[str(args[0][:-4])]
|
|
reg = str(args[0][-3:])
|
|
if "pal" in reg or "PAL" in reg:
|
|
xarg = ChannelPALVerDict[str(dld)]
|
|
elif "usa" in reg or "USA" in reg:
|
|
xarg = ChannelUSAVerDict[str(dld)]
|
|
elif "jap" in reg or "JAP" in reg:
|
|
xarg = ChannelJAPVerDict[str(dld)]
|
|
else:
|
|
print "specified region %s is invalid" % reg
|
|
sys.exit(1)
|
|
print "Available Versions for %s: %s" % (dld, xarg)
|
|
elif "IOS" in str(args[0]) or "BC" in str(args[0]) or "MIOS" in str(args[0]):
|
|
print "Available Versions for %s: %s" % (str(args[0]), IOSdict[str(args[0])])
|
|
else:
|
|
NewDict = swap_dic(TitleDict)
|
|
xarg = NewDict[str(args[0])]
|
|
print "Available Versions for %s: %s" % (str(args[0]), IOSdict[str(xarg)])
|
|
sys.exit(0)
|
|
|
|
if options.convert:
|
|
if "." in str(args[0]):
|
|
dld = ChannelCLIDict[str(args[0][:-4])]
|
|
reg = str(args[0][-3:])
|
|
if "pal" in reg or "PAL" in reg:
|
|
xarg = ChannelPALDict[str(dld)]
|
|
elif "usa" in reg or "USA" in reg:
|
|
xarg = ChannelUSADict[str(dld)]
|
|
elif "jap" in reg or "JAP" in reg:
|
|
xarg = ChannelJAPDict[str(dld)]
|
|
else:
|
|
print "specified region %s is invalid" % reg
|
|
sys.exit(1)
|
|
print "%s == %s" % (dld, xarg)
|
|
elif "IOS" in str(args[0]) or "BC" in str(args[0]) or "MIOS" in str(args[0]):
|
|
print "%s == %s" % (str(args[0]), TitleDict[str(args[0])])
|
|
else:
|
|
NewDict = swap_dic(TitleDict)
|
|
xarg = NewDict[str(args[0])]
|
|
print "%s == %s" % (str(args[0]), xarg)
|
|
sys.exit(0)
|
|
|
|
if options.unpack:
|
|
if os.access(str(options.unpack[0]), os.R_OK):
|
|
wad = str(options.unpack[0])
|
|
else:
|
|
print "WAD file %s does not exist." % str(options.unpack[0])
|
|
sys.exit(1)
|
|
if os.access(str(args[0]), os.W_OK):
|
|
folder = str(args[0])
|
|
else:
|
|
os.mkdir(str(args[0]), 0755)
|
|
if os.access(str(args[0]), os.W_OK):
|
|
UnpackingCLI(wad, str(args[0])).start()
|
|
sys.exit(0)
|
|
else:
|
|
print "Output folder %s not writeable." % str(args[0])
|
|
sys.exit(1)
|
|
|
|
if options.pack:
|
|
if os.access(str(options.pack[0]), os.R_OK):
|
|
folder = str(options.pack[0])
|
|
else:
|
|
print "Input folder %s not readable." % str(options.pack[0])
|
|
sys.exit(1)
|
|
if os.access(os.path.dirname(str(args[0])), os.W_OK):
|
|
PackingCLI(folder, str(args[0])).start()
|
|
sys.exit(0)
|
|
else:
|
|
print "Output file %s can't be created." % str(args[0])
|
|
sys.exit(1)
|
|
|
|
if options.tmdinfo:
|
|
if os.access(str(args[0]), os.R_OK):
|
|
ShowTMD(str(args[0]))
|
|
sys.exit(0)
|
|
else:
|
|
print "Can't access %s." % str(args[0])
|
|
sys.exit(1)
|
|
|
|
def main():
|
|
# load our own translations
|
|
translator = QTranslator()
|
|
translator.load(QString("i18n/Qwad_%1").arg(QLocale.system().name()))
|
|
# load Qt translations
|
|
qttranslator = QTranslator()
|
|
qttranslator.load(QString("qt_%1").arg(QLocale.system().name()))
|
|
# misc stuff
|
|
Vapp = QApplication(sys.argv)
|
|
Vapp.setOrganizationName("Nanolx")
|
|
Vapp.setApplicationName("Qwad")
|
|
Vapp.installTranslator(translator)
|
|
Vapp.installTranslator(qttranslator)
|
|
VentanaP = MWQwad()
|
|
VentanaP.show()
|
|
sys.exit(Vapp.exec_())
|
|
|
|
if __name__ == "__main__":
|
|
opts()
|
|
main()
|