diff rename.py @ 2:8fdd3e06e1ec draft

planemo upload commit eb7c6866b218a41d9b22255ee6afb83551a5ff40
author yating-l
date Mon, 25 Jun 2018 15:29:18 -0400
parents 0d13e4410c3d
children 9529a207d704
line wrap: on
line diff
--- a/rename.py	Mon Jun 18 18:36:00 2018 -0400
+++ b/rename.py	Mon Jun 25 15:29:18 2018 -0400
@@ -3,10 +3,8 @@
 """
 import sys
 import csv
-from collections import OrderedDict
 
 def rename(inputfile, outputfile, writer):
-    namemap = OrderedDict()
     with open(outputfile, 'w') as out:
         with open(inputfile, 'r') as rf:
             lines = rf.readlines()
@@ -15,20 +13,17 @@
                 if ">" in line:
                     oldname = line[1:].rstrip()
                     newname = "scaffold_" + str(i)
-                    line = ">" + newname
+                    line = ">" + newname + "\n"
                     i = i+1
                     writer.writerow([oldname, newname])
-                #TODO: Add line breaks to chromosome sequences that are in a single line
-                out.write(line.rstrip() + "\n")
+                out.write(line)
 
 def main():
     inputfile = str(sys.argv[1])
     outputfile = str(sys.argv[2])
     indexfile = str(sys.argv[3])
     csvfile = open(indexfile, 'w')
-    fieldnames = ['Original sequence name', 'Renamed sequence name']
     writer = csv.writer(csvfile)
-    writer.writerow(fieldnames)
     rename(inputfile, outputfile, writer)
 
 if __name__ == "__main__":