diff options
author | numpor <numpor@yandex-team.ru> | 2022-02-10 16:51:42 +0300 |
---|---|---|
committer | Daniil Cherednik <dcherednik@yandex-team.ru> | 2022-02-10 16:51:42 +0300 |
commit | 54d3768621e7b304a1cccb26fb8e9bc4b87208f8 (patch) | |
tree | 4c0360fc8750b0d0604005914e78e96d9634a4e5 /build/scripts | |
parent | e826ebc0dee525ef263356dece4e470c3303cc7a (diff) | |
download | ydb-54d3768621e7b304a1cccb26fb8e9bc4b87208f8.tar.gz |
Restoring authorship annotation for <numpor@yandex-team.ru>. Commit 1 of 2.
Diffstat (limited to 'build/scripts')
-rw-r--r-- | build/scripts/decimal_md5.py | 136 |
1 files changed, 68 insertions, 68 deletions
diff --git a/build/scripts/decimal_md5.py b/build/scripts/decimal_md5.py index e70ca80a09..598fbf8401 100644 --- a/build/scripts/decimal_md5.py +++ b/build/scripts/decimal_md5.py @@ -1,79 +1,79 @@ -#!/usr/bin/env python -# -*- coding: utf-8 -*- - -import hashlib -import struct -import sys -import os -import argparse - - +#!/usr/bin/env python +# -*- coding: utf-8 -*- + +import hashlib +import struct +import sys +import os +import argparse + + def print_code(checksum, func_name): if len(func_name) == 0: # safe fallback for old ya.make files func_name = "DecimalMD5" print 'const char* ' + func_name + '() {return "' + checksum + '";}' - - -def ensure_paths_exist(paths): - bad_paths = sorted( - path for path in paths - if not os.path.exists(path) - ) - if bad_paths: - print >> sys.stderr, "decimal_md5 inputs do not exist:" - for path in bad_paths: - print >> sys.stderr, path - sys.exit(1) - - -def _update_digest_with_file_contents(digest, path, block_size=65535): - with open(path) as f: - while True: - block = f.read(block_size) - if not block: - break - digest.update(block) - - -def main(): - parser = argparse.ArgumentParser() - parser.add_argument("--fixed-output", help="don not calculate md5, use this value instead") - parser.add_argument("--lower-bits", help="use specified count of lower bits", type=int, default=32) - parser.add_argument("--source-root", help="arcadia source root") + + +def ensure_paths_exist(paths): + bad_paths = sorted( + path for path in paths + if not os.path.exists(path) + ) + if bad_paths: + print >> sys.stderr, "decimal_md5 inputs do not exist:" + for path in bad_paths: + print >> sys.stderr, path + sys.exit(1) + + +def _update_digest_with_file_contents(digest, path, block_size=65535): + with open(path) as f: + while True: + block = f.read(block_size) + if not block: + break + digest.update(block) + + +def main(): + parser = argparse.ArgumentParser() + parser.add_argument("--fixed-output", help="don not calculate md5, use this value instead") + parser.add_argument("--lower-bits", help="use specified count of lower bits", type=int, default=32) + parser.add_argument("--source-root", help="arcadia source root") parser.add_argument("--func-name", help="custom function name to be defined", default="DecimalMD5") - parser.add_argument("targets", nargs='*', default=['.']) - - args = parser.parse_args() - - abs_paths = [ + parser.add_argument("targets", nargs='*', default=['.']) + + args = parser.parse_args() + + abs_paths = [ os.path.join(args.source_root, target) - for target in args.targets - ] - ensure_paths_exist(abs_paths) - - if args.fixed_output: - try: + for target in args.targets + ] + ensure_paths_exist(abs_paths) + + if args.fixed_output: + try: bitmask = (1 << args.lower_bits) - 1 fmt = '{:0%dd}' % len(str(bitmask)) checksum = fmt.format(int(args.fixed_output) & bitmask) - except ValueError: - raise ValueError("decimal_md5: bad value passed via --fixed-output: %s" % args.fixed_output) + except ValueError: + raise ValueError("decimal_md5: bad value passed via --fixed-output: %s" % args.fixed_output) print_code(str(checksum), func_name=args.func_name) - return - - md5 = hashlib.md5() - for path in abs_paths: - _update_digest_with_file_contents(md5, path) - - md5_parts = struct.unpack('IIII', md5.digest()) - md5_int = sum(part << (32 * n) for n, part in enumerate(md5_parts)) - bitmask = (1 << args.lower_bits) - 1 - fmt = '{:0%dd}' % len(str(bitmask)) - - checksum_str = fmt.format(md5_int & bitmask) + return + + md5 = hashlib.md5() + for path in abs_paths: + _update_digest_with_file_contents(md5, path) + + md5_parts = struct.unpack('IIII', md5.digest()) + md5_int = sum(part << (32 * n) for n, part in enumerate(md5_parts)) + bitmask = (1 << args.lower_bits) - 1 + fmt = '{:0%dd}' % len(str(bitmask)) + + checksum_str = fmt.format(md5_int & bitmask) print_code(checksum_str, func_name=args.func_name) - - -if __name__ == "__main__": - main() - + + +if __name__ == "__main__": + main() + |