--- Tools/scripts/h2py.py +++ Tools/scripts/h2py.py @@ -23,55 +23,56 @@ import sys, re, getopt, os -p_define = re.compile('^[\t ]*#[\t ]*define[\t ]+([a-zA-Z0-9_]+)[\t ]+') +p_define = re.compile(b'^[\t ]*#[\t ]*define[\t ]+([a-zA-Z0-9_]+)[\t ]+') p_macro = re.compile( - '^[\t ]*#[\t ]*define[\t ]+' - '([a-zA-Z0-9_]+)\(([_a-zA-Z][_a-zA-Z0-9]*)\)[\t ]+') + b'^[\t ]*#[\t ]*define[\t ]+' + b'([a-zA-Z0-9_]+)\(([_a-zA-Z][_a-zA-Z0-9]*)\)[\t ]+') -p_include = re.compile('^[\t ]*#[\t ]*include[\t ]+<([a-zA-Z0-9_/\.]+)') +p_include = re.compile(b'^[\t ]*#[\t ]*include[\t ]+<([a-zA-Z0-9_/\.]+)') -p_comment = re.compile(r'/\*([^*]+|\*+[^/])*(\*+/)?') -p_cpp_comment = re.compile('//.*') +p_comment = re.compile(br'/\*([^*]+|\*+[^/])*(\*+/)?') +p_cpp_comment = re.compile(b'//.*') ignores = [p_comment, p_cpp_comment] -p_char = re.compile(r"'(\\.[^\\]*|[^\\])'") +p_char = re.compile(br"'(\\.[^\\]*|[^\\])'") -p_hex = re.compile(r"0x([0-9a-fA-F]+)L?") +p_hex = re.compile(br"0x([0-9a-fA-F]+)L?") filedict = {} importable = {} try: - searchdirs=os.environ['include'].split(';') + searchdirs=os.environb[b'include'].split(b';') except KeyError: try: - searchdirs=os.environ['INCLUDE'].split(';') + searchdirs=os.environb[b'INCLUDE'].split(b';') except KeyError: - searchdirs=['/usr/include'] + searchdirs=[b'/usr/include'] def main(): global filedict opts, args = getopt.getopt(sys.argv[1:], 'i:') for o, a in opts: if o == '-i': - ignores.append(re.compile(a)) + ignores.append(re.compile(a.encode())) if not args: args = ['-'] for filename in args: if filename == '-': sys.stdout.write('# Generated by h2py from stdin\n') - process(sys.stdin, sys.stdout) + process(sys.stdin.buffer, sys.stdout.buffer) else: - fp = open(filename, 'r') + filename = filename.encode() + fp = open(filename, 'rb') outfile = os.path.basename(filename) - i = outfile.rfind('.') + i = outfile.rfind(b'.') if i > 0: outfile = outfile[:i] modname = outfile.upper() - outfile = modname + '.py' - outfp = open(outfile, 'w') - outfp.write('# Generated by h2py from %s\n' % filename) + outfile = modname + b'.py' + outfp = open(outfile, 'wb') + outfp.write(b'# Generated by h2py from ' + filename + b'\n') filedict = {} for dir in searchdirs: if filename[:len(dir)] == dir: @@ -85,9 +86,9 @@ def pytify(body): # replace ignored patterns by spaces for p in ignores: - body = p.sub(' ', body) + body = p.sub(b' ', body) # replace char literals by ord(...) - body = p_char.sub("ord('\\1')", body) + body = p_char.sub(b"ord('\\1')", body) # Compute negative hexadecimal constants start = 0 UMAX = 2*(sys.maxsize+1) @@ -98,7 +99,7 @@ val = int(body[slice(*m.span(1))], 16) if val > sys.maxsize: val -= UMAX - body = body[:s] + "(" + str(val) + ")" + body[e:] + body = body[:s] + b"(" + str(val).encode() + b")" + body[e:] start = s + 1 return body @@ -111,7 +112,7 @@ match = p_define.match(line) if match: # gobble up continuation lines - while line[-2:] == '\\\n': + while line[-2:] == b'\\\n': nextline = fp.readline() if not nextline: break lineno = lineno + 1 @@ -120,11 +121,11 @@ body = line[match.end():] body = pytify(body) ok = 0 - stmt = '%s = %s\n' % (name, body.strip()) + stmt = name + b' = ' + body.strip() + b'\n' try: exec(stmt, env) except: - sys.stderr.write('Skipping: %s' % stmt) + sys.stderr.buffer.write(b'Skipping: ' + stmt) else: outfp.write(stmt) match = p_macro.match(line) @@ -132,11 +133,11 @@ macro, arg = match.group(1, 2) body = line[match.end():] body = pytify(body) - stmt = 'def %s(%s): return %s\n' % (macro, arg, body) + stmt = b'def ' + macro + b'(' + arg + b'): return ' + body + b'\n' try: exec(stmt, env) except: - sys.stderr.write('Skipping: %s' % stmt) + sys.stderr.buffer.write(b'Skipping: ' + stmt) else: outfp.write(stmt) match = p_include.match(line) @@ -145,23 +146,24 @@ a, b = regs[1] filename = line[a:b] if filename in importable: - outfp.write('from %s import *\n' % importable[filename]) + outfp.write(b'from ' + importable[filename] + b' import *\n') elif filename not in filedict: filedict[filename] = None inclfp = None for dir in searchdirs: try: - inclfp = open(dir + '/' + filename) + inclfp = open(dir + b'/' + filename, 'rb') break except IOError: pass if inclfp: outfp.write( - '\n# Included from %s\n' % filename) + b'\n# Included from ' + filename + b'\n') process(inclfp, outfp, env) + inclfp.close() else: - sys.stderr.write('Warning - could not find file %s\n' % - filename) + sys.stderr.buffer.write(b'Warning - could not find file ' + + filename + b'\n') if __name__ == '__main__': main()