Optimize code for mtk_gui

mtk_gui

@@ -446,13 +446,13 @@ class MainWindow(QMainWindow):
         return
 
     def on_eraseboot2(self):
-        self.eraseflash.eraseBoot2()
+        self.eraseflash.eraseFlash("boot2")
 
     def on_erasepreloader(self):
-        self.eraseflash.erasePreloader()
+        self.eraseflash.eraseFlash("boot1")
 
     def on_eraserpmb(self):
-        self.eraseflash.eraseRpmb()
+        self.eraseflash.eraseFlash("rpmb")
 
     def on_generatekeys(self):
         self.genkeys.generateKeys()

mtkclient/Library/Connection/usbscsi.py

@@ -8,7 +8,7 @@ from mtkclient.Library.Connection.usblib import Scsi
 def main():
     info = 'MassStorageBackdoor (c) B.Kerler 2019.'
     parser = argparse.ArgumentParser(description=info)
-    print("\n" + info + "\n\n")
+    print(f"\n{info}\n\n")
     parser.add_argument('-vid', metavar="<vid>", help='[Option] Specify vid, default=0x2e04)', default="0x2e04")
     parser.add_argument('-pid', metavar="<pid>", help='[Option] Specify pid, default=0xc025)', default="0xc025")
     parser.add_argument('-interface', metavar="<pid>", help='[Option] Specify interface number)', default="")

mtkclient/gui/eraseFlashPartitions.py

@@ -96,18 +96,6 @@ class EraseFlashWindow(QObject):
         thread.wait()
         self.enableButtonsSignal.emit()
 
-    def eraseBoot2(self):
-        self.eraseFlash("boot2")
-        return
-
-    def erasePreloader(self):
-        self.eraseFlash("boot1")
-        return
-
-    def eraseRpmb(self):
-        self.eraseFlash("rpmb")
-        return
-
     def erasePartitionAsync(self, toolkit, parameters):
         self.parent.timeEst.init()
         self.parent.timeEstTotal.init()