annotate validator.py @ 0:2626c5b4c665 draft

planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
author yating-l
date Mon, 19 Dec 2016 13:07:32 -0500
parents
children cf258ca024ff
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
1 import sys
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
2
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
3 """
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
4 Call checkAndFixBed, check the integrity of bed file. If the strand is not "+" or "-" truncate that line and report to users
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
5 create a column and move the score column to that column.
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
6 """
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
7 def checkAndFixBed(bedfile, revised_file):
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
8 # Store the lines that have been removed
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
9 removedLines = []
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
10 # Remove the lines with invalid strand, create a score column to store the original scores and change scores in the original score column all to 1000
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
11 with open(revised_file, 'w') as tmp:
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
12 with open(bedfile, 'r') as f:
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
13 lines = f.readlines()
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
14 i = 1
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
15 for line in lines:
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
16 fields = line.split()
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
17 strand = fields[5]
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
18 score = fields[4]
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
19 fields[4] = '1000'
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
20 fields.append(score)
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
21 if (strand == '+' or strand == '-'):
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
22 tmp.write('\t'.join(map(str, fields)))
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
23 tmp.write("\n")
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
24 else:
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
25 removedLines.append("line" + str(i) + ": " + line)
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
26 i = i+1
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
27
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
28 return removedLines
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
29
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
30 def main():
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
31 inputfile = str(sys.argv[1])
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
32 outputfile = str(sys.argv[2])
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
33 removed = checkAndFixBed(inputfile, outputfile)
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
34 if (removed != []):
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
35 print "\nRemoved invalid lines: \n"
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
36 print "\n".join(removed)
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
37
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
38 if __name__ == "__main__":
2626c5b4c665 planemo upload commit db634d933827ebc78981c7a0aa18205a85fa42e6-dirty
yating-l
parents:
diff changeset
39 main()