Migrate python2 scripts to python3

Tests Done: Boot check
Tracked-On: OAM-122386
Signed-off-by: Salini Venate <[email protected]>

bootloader_from_zip

@@ -83,12 +83,12 @@ def main(argv):
         sys.exit(1)
 
     if not OPTIONS.zipfile:
-        print "--zipfile is required"
+        print ("--zipfile is required")
         common.Usage(__doc__)
         sys.exit(1)
 
     tf = tempfile.NamedTemporaryFile()
-    tf.write("foo")
+    tf.write(b"foo")
     tf.flush()
 
     extra_files = OPTIONS.bootimage
@@ -106,9 +106,9 @@ if __name__ == '__main__':
     try:
         common.CloseInheritedPipes()
         main(sys.argv[1:])
-    except common.ExternalError, e:
+    except (common.ExternalError,e):
         print
-        print "   ERROR: %s" % (e,)
+        print ("   ERROR: %s" % (e,))
         print
         sys.exit(1)
 

create_gpt_image.py

@@ -17,7 +17,7 @@
 Script to create a GPT/UEFI image or to show information it contains.
 """
 
-from sys import version_info
+from sys import version_info, exit
 
 if version_info < (2, 7, 3):
     exit('Python version must be 2.7.3 or higher')
@@ -35,7 +35,7 @@
 if version_info < (3, 0, 1):
     from ConfigParser import SafeConfigParser, ParsingError, NoOptionError, NoSectionError
 else:
-    from configparser import SafeConfigParser, ParsingError, NoOptionError, NoSectionError
+    from configparser import ConfigParser, ParsingError, NoOptionError, NoSectionError
 from math import floor, log
 
 
@@ -135,8 +135,8 @@ def write(self, img_file, offset=0):
         Used to write MBR in an image file
         """
         self.raw = pack(MBRInfos._FMT, self.boot, self.os_type,
-                        self.lba_start, self.lba_size, '',
-                        MBRInfos._PART_ENTRY, '', self.sign)
+                        self.lba_start, self.lba_size, b'',
+                        MBRInfos._PART_ENTRY.encode('utf-8'), b'', self.sign.encode('utf-8'))
         img_file.seek(offset)
         img_file.write(self.raw)
 
@@ -284,30 +284,30 @@ def write(self, img_file, offset, block_size):
         """
         Used to write GPT header and backup in an image file
         """
-        self.raw = pack(GPTHeaderInfos._FMT, self.sign, self.rev,
-                        self.size, 0, 1, self.lba_backup,
-                        self.lba_first, self.lba_last, self.uuid,
+        self.raw = pack(GPTHeaderInfos._FMT, self.sign.encode('utf-8'), self.rev.encode('utf-8'),
+                        self.size, 0, 1, int(self.lba_backup),
+                        int(self.lba_first), int(self.lba_last), self.uuid,
                         2, self.table_length, self.entry_size, 0)
 
-        backup_raw = pack(GPTHeaderInfos._FMT, self.sign, self.rev,
-                          self.size, 0, self.lba_backup, 1,
-                          self.lba_first, self.lba_last, self.uuid,
-                          self.lba_start, self.table_length,
+        backup_raw = pack(GPTHeaderInfos._FMT, self.sign.encode('utf-8'), self.rev.encode('utf-8'),
+                          self.size, 0, int(self.lba_backup), 1,
+                          int(self.lba_first), int(self.lba_last), self.uuid,
+                          int(self.lba_start), self.table_length,
                           self.entry_size, 0)
 
         # writes a new GPT header
         img_file.seek(offset)
         img_file.write(self.raw)
 
         # writes zero on unused blocks of GPT header
-        raw_stuffing = '\x00' * (block_size - len(self.raw))
+        raw_stuffing = b'\x00' * (block_size - len(self.raw))
         img_file.write(raw_stuffing)
 
         # saves the end of the GPT header
         gpt_header_end = img_file.tell()
 
         # writes a new GPT backup
-        backup_position = self.lba_backup * block_size
+        backup_position = int(self.lba_backup) * block_size
         img_file.seek(backup_position)
         img_file.write(backup_raw)
 
@@ -357,6 +357,7 @@ def write(self, img_file, offset, entry_size, tlb_infos, last_usable):
         """
         # erases the partition table entries
         self = []
+        offset=int(offset)
 
         # writes all new partition entries in GPT header
         current_offset = offset
@@ -371,7 +372,7 @@ def write(self, img_file, offset, entry_size, tlb_infos, last_usable):
         img_file.seek(offset)
         raw_entries_size = current_offset - offset
         raw_entries = img_file.read(raw_entries_size)
-        img_file.seek(last_usable + 1)
+        img_file.seek(int(last_usable) + 1)
         img_file.write(raw_entries)
 
         img_file.seek(current_offset)
@@ -738,7 +739,7 @@ def _read_ini(self, block_size):
         Used to read a INI TLB partition file
         """
         # sets a parser to read the INI TLB partition file
-        cfg = SafeConfigParser()
+        cfg = ConfigParser(strict=False)
         try:
             cfg.read(self.path)
 
@@ -965,7 +966,7 @@ def _write_crc(self, img_file):
         img_file.seek(2 * self.block_size)
         raw_table = img_file.read(self.gpt_header.table_length *
                                   self.gpt_header.entry_size)
-        img_file.seek((self.gpt_header.lba_backup - 32) * self.block_size)
+        img_file.seek((int(self.gpt_header.lba_backup) - 32) * self.block_size)
         raw_backup_table = img_file.read(self.gpt_header.table_length *
                                          self.gpt_header.entry_size)
 
@@ -1035,7 +1036,7 @@ def _write_partitions(self, img_file, tlb_infos, binaries_path):
             # no binary file used to build the partition or slot_b case
             label = tlb_part.label[0:]
             if bin_path == 'none' or label[len(label)-2:] == '_b':
-                line = '\0'
+                line = b'\0'
                 img_file.seek(offset)
                 img_file.write(line)
                 bin_size = 0
@@ -1081,7 +1082,7 @@ def write(self, tlb_infos, binaries_path):
 
             # fill output image header with 0x00: MBR size + GPT header size +
             # (partition table length * entry size)
-            zero = '\x00' * (2 * self.block_size +
+            zero = b'\x00' * (2 * self.block_size +
                              self.gpt_header.table_length *
                              self.gpt_header.entry_size)
             img_file.seek(0)

generate_factory_images

@@ -32,7 +32,7 @@ import os
 
 _FLASHALL_FILENAME = "flash-all.sh"
 # chmod (octal) -rwxr-x--x
-_PERMS = 0751
+_PERMS = 0o751
 _FLASH_HEADER = """#!/bin/bash
 
 # Copyright 2012 The Android Open Source Project
@@ -103,23 +103,23 @@ def ConvertToDOSFormat(filename):
 
 
 def AddFlashScript(filename, tar, commands, windows):
-    print "Archiving", filename
+    print ("Archiving", filename)
     tf = tempfile.NamedTemporaryFile(delete=False)
     if (windows):
-        tf.write(_WIN_FLASH_HEADER)
+        tf.write(_WIN_FLASH_HEADER.encode('utf-8'))
     else:
-        tf.write(_FLASH_HEADER)
+        tf.write(_FLASH_HEADER.encode('utf-8'))
 
     for c in commands:
         if windows:
-            tf.write(c.get_windows_command())
+            tf.write(c.get_windows_command().encode('utf-8'))
         else:
-            tf.write(c.get_linux_command())
+            tf.write(c.get_linux_command().encode('utf-8'))
 
     if (windows):
-        tf.write(_WIN_FLASH_FOOTER)
+        tf.write(_WIN_FLASH_FOOTER.encode('utf-8'))
     else:
-        tf.write(_FLASH_FOOTER)
+        tf.write(_FLASH_FOOTER.encode('utf-8'))
 
     tf.close()
     if (windows):
@@ -146,8 +146,8 @@ class CommandlineParser(ArgumentParser):
         self.description = __doc__
 
     def error(self, message):
-        print >>stderr, "ERROR: {}".format(message)
-        print >>stderr, "\n------\n"
+        print ("ERROR: {}".format(message), file=sys.stderr)
+        print ("\n------\n", file=sys.stderr)
         self.print_help()
         exit(2)
 
@@ -230,31 +230,31 @@ def main():
     archive_name = args.output
 
     # Create Archive
-    print "Creating archive: " + archive_name
+    print ("Creating archive: " + archive_name)
     tar = TarOpen(archive_name, "w:gz")
 
     for src_path, dst_path in files:
-        print "Archiving " + src_path
+        print ("Archiving " + src_path)
         RequireFile(src_path)
         tar.add(src_path, arcname=dst_path)
 
     # 'fastboot update' covers the additional AOSP pieces, add this to the
     # command list now
     commands.append(UpdateCommand(update_fn, True))
-    print "Archiving " + args.update_archive
+    print ("Archiving " + args.update_archive)
     RequireFile(args.update_archive)
     tar.add(args.update_archive, update_fn)
     AddFlashScript(_FLASHALL_FILENAME, tar, commands, windows=False)
     AddFlashScript(_WIN_FLASHALL_FILENAME, tar, commands, windows=True)
 
     tar.close()
 
-    print "Done."
+    print ("Done.")
 
 if __name__ == "__main__":
     try:
         exit(main())
-    except Usage, err:
-        print >>stderr, "ERROR: {}".format(err.msg)
-        print >>stderr, "       for help use --help"
+    except Usage as err:
+        print ("ERROR: {}".format(err.msg), file=sys.stderr)
+        print ("       for help use --help", file=sys.stderr)
         exit(2)

releasetools/bootloader_from_target_files

@@ -61,7 +61,7 @@ def main(argv):
         common.Usage(__doc__)
         sys.exit(1)
 
-    print "unzipping target-files..."
+    print ("unzipping target-files...")
     #OPTIONS.input_tmp = common.UnzipTemp(args[0])
     OPTIONS.input_tmp = args[0]
     #input_zip = zipfile.ZipFile(args[0], "r")
@@ -91,7 +91,7 @@ if __name__ == '__main__':
         main(sys.argv[1:])
     except common.ExternalError, e:
         print
-        print "   ERROR: %s" % (e,)
+        print ("   ERROR: %s" % (e,))
         print
         sys.exit(1)
     finally:

releasetools/flash_cmd_generator.py

@@ -330,7 +330,13 @@ def parse_config(ips, variant, platform):
     results_list = []
     for k,v in results.items():
         results_list.append((k,v))
-    flist = [f.rsplit(':', 1) for f in set(files)]
+    unique_files = []
+    for file in files:
+    # If the number is not already in the unique_numbers list, add it
+        if file not in unique_files:
+            unique_files.append(file)
+
+    flist = [f.rsplit(':', 1) for f in unique_files]
     return results_list, flist