1
0
Fork 0
mirror of https://github.com/opentx/opentx.git synced 2025-07-16 12:55:12 +03:00
opentx/radio/util/include-guard.py
Sean Vig 4bd90d79d0 Use with open(...) as ...: context managers
Instead of manually opening and closes file objects, use the context
manager that will properly handle opening and closing files for us, even
in the event of some error
2015-12-13 18:57:36 -06:00

28 lines
917 B
Python
Executable file

#!/usr/bin/env python
from __future__ import print_function
import sys
import os
for filename in sys.argv[1:]:
with open(filename, "r") as f:
lines = f.readlines()
newguard = "_" + os.path.basename(filename).upper().replace(".", "_") + "_"
for i, line in enumerate(lines):
line = line.strip()
if line.startswith("#ifndef "):
guard = line[8:]
if lines[i + 1].strip() == "#define %s" % guard:
print(filename, ":", guard, "=>", newguard)
lines[i] = "#ifndef %s\n" % newguard
lines[i + 1] = "#define %s\n" % newguard
end = -1
while not lines[end].strip().startswith("#endif"):
end -= 1
lines[end] = "#endif // %s\n" % newguard
with open(filename, "w") as f:
f.write("".join(lines))
break