diff options
author | Doug Kerr <dek3rr@gmail.com> | 2022-04-15 15:54:30 -0400 |
---|---|---|
committer | Hauke Mehrtens <hauke@hauke-m.de> | 2022-04-16 14:53:17 +0200 |
commit | 6f692c9c497e5e60eb51da741b07ef1c10096355 (patch) | |
tree | 985be802ae7f7c6453ff55e9e431c039b6dfcd36 /scripts/cfe-partition-tag.py | |
parent | 0642a2166b9526ec01fe47b6aed1fd57a29e35f9 (diff) | |
download | upstream-6f692c9c497e5e60eb51da741b07ef1c10096355.tar.gz upstream-6f692c9c497e5e60eb51da741b07ef1c10096355.tar.bz2 upstream-6f692c9c497e5e60eb51da741b07ef1c10096355.zip |
scripts: format to black
clean up formatting with black using 80 character line limit
Signed-off-by: Doug Kerr <dek3rr@gmail.com>
Diffstat (limited to 'scripts/cfe-partition-tag.py')
-rwxr-xr-x | scripts/cfe-partition-tag.py | 173 |
1 files changed, 96 insertions, 77 deletions
diff --git a/scripts/cfe-partition-tag.py b/scripts/cfe-partition-tag.py index 875bfae85d..41495a9af0 100755 --- a/scripts/cfe-partition-tag.py +++ b/scripts/cfe-partition-tag.py @@ -25,96 +25,115 @@ PART_VERSION_SIZE = 21 def auto_int(x): - return int(x, 0) + return int(x, 0) + def str_to_bytes_pad(string, size): - str_bytes = string.encode() - num_bytes = len(str_bytes) - if (num_bytes >= size): - str_bytes = str_bytes[:size - 1] + '\0'.encode() - else: - str_bytes += '\0'.encode() * (size - num_bytes) - return str_bytes + str_bytes = string.encode() + num_bytes = len(str_bytes) + if num_bytes >= size: + str_bytes = str_bytes[: size - 1] + "\0".encode() + else: + str_bytes += "\0".encode() * (size - num_bytes) + return str_bytes + def create_tag(args, in_bytes, size): # JAM CRC32 is bitwise not and unsigned - crc = (~binascii.crc32(in_bytes) & 0xFFFFFFFF) + crc = ~binascii.crc32(in_bytes) & 0xFFFFFFFF + + tag = bytearray() + tag += struct.pack(">I", args.part_id) + tag += struct.pack(">I", size) + tag += struct.pack(">H", args.part_flags) + tag += str_to_bytes_pad(args.part_name, PART_NAME_SIZE) + tag += str_to_bytes_pad(args.part_version, PART_VERSION_SIZE) + tag += struct.pack(">I", crc) - tag = bytearray() - tag += struct.pack('>I', args.part_id) - tag += struct.pack('>I', size) - tag += struct.pack('>H', args.part_flags) - tag += str_to_bytes_pad(args.part_name, PART_NAME_SIZE) - tag += str_to_bytes_pad(args.part_version, PART_VERSION_SIZE) - tag += struct.pack('>I', crc) + return tag - return tag def create_output(args): - in_st = os.stat(args.input_file) - in_size = in_st.st_size + in_st = os.stat(args.input_file) + in_size = in_st.st_size - in_f = open(args.input_file, 'r+b') - in_bytes = in_f.read(in_size) - in_f.close() + in_f = open(args.input_file, "r+b") + in_bytes = in_f.read(in_size) + in_f.close() - tag = create_tag(args, in_bytes, in_size) + tag = create_tag(args, in_bytes, in_size) + + out_f = open(args.output_file, "w+b") + out_f.write(tag) + out_f.close() - out_f = open(args.output_file, 'w+b') - out_f.write(tag) - out_f.close() def main(): - global args - - parser = argparse.ArgumentParser(description='') - - parser.add_argument('--flags', - dest='part_flags', - action='store', - type=auto_int, - help='Partition Flags') - - parser.add_argument('--id', - dest='part_id', - action='store', - type=auto_int, - help='Partition ID') - - parser.add_argument('--input-file', - dest='input_file', - action='store', - type=str, - help='Input file') - - parser.add_argument('--output-file', - dest='output_file', - action='store', - type=str, - help='Output file') - - parser.add_argument('--name', - dest='part_name', - action='store', - type=str, - help='Partition Name') - - parser.add_argument('--version', - dest='part_version', - action='store', - type=str, - help='Partition Version') - - args = parser.parse_args() - - if ((not args.part_flags) or - (not args.part_id) or - (not args.input_file) or - (not args.output_file) or - (not args.part_name) or - (not args.part_version)): - parser.print_help() - else: - create_output(args) + global args + + parser = argparse.ArgumentParser(description="") + + parser.add_argument( + "--flags", + dest="part_flags", + action="store", + type=auto_int, + help="Partition Flags", + ) + + parser.add_argument( + "--id", + dest="part_id", + action="store", + type=auto_int, + help="Partition ID", + ) + + parser.add_argument( + "--input-file", + dest="input_file", + action="store", + type=str, + help="Input file", + ) + + parser.add_argument( + "--output-file", + dest="output_file", + action="store", + type=str, + help="Output file", + ) + + parser.add_argument( + "--name", + dest="part_name", + action="store", + type=str, + help="Partition Name", + ) + + parser.add_argument( + "--version", + dest="part_version", + action="store", + type=str, + help="Partition Version", + ) + + args = parser.parse_args() + + if ( + (not args.part_flags) + or (not args.part_id) + or (not args.input_file) + or (not args.output_file) + or (not args.part_name) + or (not args.part_version) + ): + parser.print_help() + else: + create_output(args) + main() |