import sys import re import re f1 = open(sys.argv[1],'r') f2 = open(sys.argv[2],'r') ladder_file = open(sys.argv[3],'r') output = "" ladder = [] sktfile = [line.rstrip('\n').strip() for line in f1] tibfile = [line.rstrip('\n').strip() for line in f2] last_score = 0.5 def clean_num(string): string = re.sub("[^0-9, ]","",string) return int(string.split(',')[0]) for line in ladder_file: if len(line.split(':')) == 3: skt,tib,score = line.split(':') if re.search("[0-9]",skt) and re.search("[0-9]",tib): skt_num = clean_num(skt) tib_num = clean_num(tib) score = float(score) if score > 0.0: ladder.append([skt_num,tib_num,score]) last_skt = 0 last_tib = 0 for entry in ladder: output = output + ' +$+ '.join(sktfile[last_skt:entry[0]]) + "\n" output = output + "# " + ' +!+ '.join(tibfile[last_tib:entry[1]]) + "\n" #+ "\t" + " SCORE: " + str(entry[2]) + "\n" last_skt = entry[0] last_tib = entry[1] output = output + ' / '.join(sktfile[last_skt:-1]) + "\n" output = output + "# " + ' / '.join(tibfile[last_tib:-1]) + "\n" short_f1 = re.sub("\.tsv.*","",sys.argv[1]) short_f2 = re.sub(".*/","",sys.argv[2]) short_f2 = re.sub("\.tsv.*","",short_f2) with open(short_f1 + "_" + short_f2 + ".org", 'w') as file: file.write(output)