Commit d76de30f authored by Kelvin Rodriguez's avatar Kelvin Rodriguez Committed by GitHub
Browse files

added data area change script (#207)

* added data area change scropt

* now featuring regex

* now featuring less whitespace
parent baff4431
Loading
Loading
Loading
Loading
+32 −0
Original line number Diff line number Diff line
#!/usr/bin/env python

import argparse
from glob import glob
from os import path
import re

parser = argparse.ArgumentParser()
parser.add_argument('mk_path', action='store', help='Path containing metakernel (*.tm) files.')
parser.add_argument('new_data_path', action='store', help='new data path containing kernel, this path will overwrite the current data path in the metakernels')
args = parser.parse_args()

mks = glob(path.join(args.mk_path,'*.tm'))

for mk in mks:
    with open(mk, 'r') as f:
        lines = f.readlines()

    lines_to_change = []
    for i,l in enumerate(lines):
        match = re.search('PATH_VALUES\s*=', l)
        if match:
            lines_to_change.append(i)

    if not lines_to_change:
        print(f'No path values in {mk}')

    for i in lines_to_change:
        lines[i] = f'      PATH_VALUES     = ( \'{new_data_dir}\' )\n'
    with open(mk, 'w') as f:
        f.write(''.join(lines))