Merge pull request #3 from Cubiss/py

Fix bug: take only base name of a zip file when full path is supplied
XiaomiFirmwareUpdater · May 17, 2019 · 359f025 · 359f025
2 parents 6b9510c + 11071ee
commit 359f025
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/create_flashable_firmware.py b/create_flashable_firmware.py
@@ -166,6 +166,7 @@ def vendor_updater():
 
 def make_zip():
     rom, process = arg_parse()
+    rom = path.basename(rom)
     with open("out/META-INF/com/google/android/updater-script", 'r') as i:
         codename = str(i.readlines()[7].split('/', 3)[2]).split(':', 1)[0].replace('_', '-')
     print("Creating " + process + " zip from " + rom + " for " + codename)
@@ -214,5 +215,6 @@ def main():
     make_zip()
 
 
-arg_parse()
-main()
+if __name__ == '__main__':
+    arg_parse()
+    main()