Files
rdkit/Code/GraphMol/MolDraw2D/update_hash_codes.py
Ric 880a8e5725 Reformat Python code for 2023.03 release (#6294)
* run yapf

* run isort

---------

Co-authored-by: Greg Landrum <greg.landrum@gmail.com>
2023-04-28 06:53:56 +02:00

52 lines
1.4 KiB
Python

#!/usr/bin/env python
# Script to update the hash codes in the cpp codes.
# Takes 2 arguments, the output from running one of the test programs
# giving the values to update, and the corresponding cpp
# file. Output is to {file.cpp}.update.
# Note that the PNG hash codes are only for Freetype builds, so it's
# better to do the non-Freetype builds first, otherwise the PNG
# hashes will definitely be wrong.
import json
import re
import sys
updates = {}
with open(sys.argv[1], 'r') as f:
for line in f.readlines():
if line.startswith('file'):
line_bits = line.strip().split()
updates[(line_bits[1], line_bits[8])] = (line_bits[1], line_bits[4])
re_str = r'{"([\w.-]*.\w*)", (\d*U)},'
patt = re.compile(re_str)
with open(sys.argv[2], 'r') as f:
cpp_file = f.read()
hashes = patt.findall(cpp_file)
def hash_replace(match):
match_bits = match.group().split(',')
match_tuple = (match_bits[0][2:-1], match_bits[1][1:-1])
if match_tuple in updates:
rep_tuple = updates[match_tuple]
rep_str = f'{{"{rep_tuple[0]}", {rep_tuple[1]}}},'
print(f'replacing {match.group()} with {rep_str}')
return rep_str
else:
return match.group()
for hash in hashes:
print(hash)
if hash in updates:
print(hash, updates[hash])
new_cpp_file = patt.sub(hash_replace, cpp_file)
with open(f'{sys.argv[2]}.update', 'w') as f:
f.write(new_cpp_file)