Skip to content
Snippets Groups Projects
Commit de7171db authored by Matthew Krafczyk's avatar Matthew Krafczyk
Browse files

Use better scheme for getting root directory

parent 6c854246
No related branches found
No related tags found
No related merge requests found
......@@ -69,9 +69,13 @@ def set_stacktrace(flag):
def process_stacktrace(countback):
"""Gives file and line frame 'countback' frames from the bottom"""
st = traceback.extract_stack()
# First entry should be bin/spack. Use this to get spack 'root'.
# bin/spack is 9 characters, the length of the 'root' is then len-9.
root_len = len(st[0][0]) - 9
# All entries will be spack files based on how this function is called.
# We use commonprefix to find what the spack 'root' directory is.
file_list = []
for frame in st:
file_list.append(frame[0])
root_dir = os.path.commonprefix(file_list)
root_len = len(root_dir)
st_idx = len(st) - countback - 1
st_text = "%s:%i " % (st[st_idx][0][root_len:], st[st_idx][1])
return st_text
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment