##################################
#                                #
# Last modified 2018/06/14       #
#                                #
# Georgi Marinov                 #
#                                # 
##################################

import sys
import string
from sets import Set
import time

def run():

    if len(sys.argv) < 4:
        print 'usage: python %s test.json test_suffix train_suffix train.json' % sys.argv[0]
        sys.exit(1)

    input = sys.argv[1]
    train_suffix = sys.argv[2]
    test_suffix = sys.argv[3]
    outfilename = sys.argv[4]

    outfile = open(outfilename, 'w')

    listoflines = open(input)
    for line in listoflines:
        if line.startswith('"intervals_file":'):
            newline = line.replace(train_suffix, test_suffix)
            outfile.write(newline)
        else:
            outfile.write(line)

    outfile.close()

run()