#!/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.8' 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()