79 lines
2.5 KiB
Python
Executable File
79 lines
2.5 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
# replaces a `include with the full include file.
|
|
# recursively replaces `include's until none are left
|
|
#
|
|
# args
|
|
# $1 - file to remove includes from
|
|
# $2 - file to write output to
|
|
# $3 - list of directories to search for includes in (note: NON-RECURSIVE must specify all dirs)
|
|
# includes are found relative to this path
|
|
# this is equivalent to something like +incdir+
|
|
|
|
import sys
|
|
import re
|
|
import os
|
|
import tempfile
|
|
import shutil
|
|
|
|
inVlog = sys.argv[1]
|
|
outVlog = sys.argv[2]
|
|
print("[INFO] Replaces includes from: " + str(inVlog))
|
|
|
|
if inVlog == outVlog:
|
|
sys.exit("[ERROR] The input and output file cannot be the same.")
|
|
|
|
# add directories to search list
|
|
incDirs = sys.argv[3:]
|
|
print("[INFO] Searching following dirs for includes: " + str(incDirs))
|
|
|
|
def process(inF, outF):
|
|
# open file
|
|
with open(inF, 'r') as inFile:
|
|
with open(outF, 'w') as outFile:
|
|
# for each include found, search through all dirs and replace if found, error if not
|
|
for num, line in enumerate(inFile, 1):
|
|
match = re.match(r"^ *`include +\"(.*)\"", line)
|
|
if match:
|
|
# search for include and replace
|
|
found = False
|
|
for d in incDirs:
|
|
potentialIncFileName = d + "/" + match.group(1)
|
|
if os.path.exists(potentialIncFileName):
|
|
found = True
|
|
with open(potentialIncFileName, 'r') as incFile:
|
|
for iline in incFile:
|
|
outFile.write(iline)
|
|
break
|
|
|
|
# must find something to include with
|
|
if not found:
|
|
sys.exit("[ERROR] Couldn't replace include \"" + str(match.group(1)) + "\" found on line " + str(num))
|
|
else:
|
|
outFile.write(line)
|
|
|
|
inF = inVlog
|
|
|
|
while True:
|
|
# create a copy of the input
|
|
fd, temp_path = tempfile.mkstemp()
|
|
shutil.copy2(inF, temp_path)
|
|
|
|
with open(temp_path, 'r') as inFile:
|
|
anyIncludes = False
|
|
for line in inFile:
|
|
match = re.match(r"^ *`include +\"(.*)\"", line)
|
|
if match:
|
|
anyIncludes = True
|
|
break
|
|
|
|
if anyIncludes:
|
|
process(temp_path, outVlog)
|
|
inF = outVlog
|
|
os.remove(temp_path)
|
|
else:
|
|
os.remove(temp_path)
|
|
break
|
|
|
|
print("[INFO] Success. Writing output to: " + str(outVlog))
|