Migrate python2 scripts to python3

Tests Done: Boot check
Tracked-On: NA
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
 
 
@@ -93,7 +93,7 @@ def __repr__(self):
         if self.lba_size > 0:
             units = ('KBytes', 'MBytes', 'GBytes')
             index = int(floor(log(self.lba_size, 1024)))
-            computed_size = round(self.lba_size / (1024**index), 2)
+            computed_size = round(self.lba_size // (1024**index), 2)
             human_size = '{0} {1}'.format(computed_size, units[index])
         else:
             human_size = '0 Bytes'
@@ -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)
 
@@ -213,13 +213,13 @@ def __init__(self, img_size=2147483648, block_size=512, size=92):
         self.entry_size = 128
 
         # calculates the size of image in block
-        size_in_block = img_size / block_size
+        size_in_block = img_size // block_size
 
         # sets the lba backup at the value of first lba used by GPT backup
         self.lba_backup = size_in_block - 1
 
         # calculates the size of the partition table in block
-        table_size = (self.table_length * self.entry_size) / block_size
+        table_size = (self.table_length * self.entry_size) // block_size
 
         # sets the lba first at the first usable lba for a partition
         self.lba_first = table_size + 2
@@ -284,12 +284,12 @@ 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.raw = pack(GPTHeaderInfos._FMT, self.sign.encode('utf-8'), self.rev.encode('utf-8'),
                         self.size, 0, 1, self.lba_backup,
                         self.lba_first, self.lba_last, self.uuid,
                         2, self.table_length, self.entry_size, 0)
 
-        backup_raw = pack(GPTHeaderInfos._FMT, self.sign, self.rev,
+        backup_raw = pack(GPTHeaderInfos._FMT, self.sign.encode('utf-8'), self.rev.encode('utf-8'),
                           self.size, 0, self.lba_backup, 1,
                           self.lba_first, self.lba_last, self.uuid,
                           self.lba_start, self.table_length,
@@ -300,7 +300,7 @@ def write(self, img_file, offset, block_size):
         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
@@ -690,7 +690,7 @@ def _contruct_tlb_info(self, start_lba, cfg, block_size, parts):
                     readlen = cfg.getint(partname, 'len')
 
                 if readlen > 0:
-                    size = (readlen * 1024 * 1024) / block_size
+                    size = (readlen * 1024 * 1024) // block_size
                     start_lba = begin + size
                 else:
                     size = readlen
@@ -724,7 +724,7 @@ def _contruct_tlb_grp_info(self, start_lba, cfg, block_size, parts):
                     begin = start_lba
 
                 if readlen > 0:
-                    size = (readlen * 1024 * 1024) / block_size
+                    size = (readlen * 1024 * 1024) // block_size
                     start_lba = begin + size
                 else:
                     size = readlen
@@ -738,7 +738,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)
 
@@ -1035,7 +1035,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
@@ -1048,7 +1048,7 @@ def _write_partitions(self, img_file, tlb_infos, binaries_path):
             # checks if partition size is greather or equal to the binary file
             bin_size_in_bytes = stat(bin_path).st_size
             part_size_in_bytes = tlb_part.size * self.block_size
-            bin_size = bin_size_in_bytes / self.block_size
+            bin_size = bin_size_in_bytes // self.block_size
             if tlb_part.size < bin_size:
                 error('Size of binary file {0} ({1} Bytes) is greather than '
                       '{2} partition size ({3} Bytes)'.format(bin_path,
@@ -1081,7 +1081,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