def ask_ramdisk(): ramdisks = ramdisk.list_ramdisks() if not ramdisks: raise Exception("No ramdisks found") ramdisks.sort() print("Which type of ramdisk do you have?") print("") counter = 0 for i in ramdisks: print("%i: %s" % (counter + 1, ramdisk.suffix.sub('', i))) counter += 1 print("") choice = input("Choice: ") try: choice = int(choice) choice -= 1 if choice < 0 or choice >= counter: raise Exception("Invalid choice") return (ramdisks, choice) except ValueError: raise Exception("Invalid choice")
def setdefaultvalues(self): # Auto patch by default self.rbautopatch.setChecked(True) self.patchtypetoggled() # Don't remove device checks self.cbdevicecheck.setChecked(False) # Boot image exists self.cbhasbootimg.setChecked(True) self.hasbootimagetoggled() # No loki self.cbloki.setChecked(False) # No patched init self.cbpatchedinit.setChecked(False) self.patchedinittoggled() # 'boot.img' self.lebootimg.setText("boot.img") self.enablebootimgcombobox(True) if not self.addonce: self.addonce = True self.presets = fileinfo.get_infos(config.get_device()) self.cmbpreset.addItem("Custom") for i in self.presets: self.cmbpreset.addItem(i[0][:-3]) for i in fileinfo.get_inits(): self.cmbinitfile.addItem(i) for i in ramdisk.list_ramdisks(): self.cmbramdisk.addItem(i[:-4]) self.autopatchers = autopatcher.get_auto_patchers() for i in self.autopatchers: self.cmbautopatchsel.addItem(i.name)
for i in sys.argv: if i == '--loki': loki = True elif i == '--askramdisk': askramdisk = True elif i == '--listramdisks': action = 'list' elif i.startswith('--'): print("Invalid argument: " + i) sys.exit(1) else: filename = i # List ramdisks if that was the action if action == 'list': for i in ramdisk.list_ramdisks(): print(i) sys.exit(1) if not filename: print("No file specified") sys.exit(1) file_info = fileinfo.FileInfo() file_info.set_filename(filename) file_info.set_device(config.get_device()) if not file_info.is_filetype_supported(): print('Unsupported file type') sys.exit(1)