pyfasta split can handle > 52 splits

brentp · Feb 21, 2012 · 2464237 · 2464237
1 parent 4fa34b8
commit 2464237
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/CHANGELOG.txt b/CHANGELOG.txt
@@ -1,5 +1,9 @@
 Changes
 =======
+0.4.5
+-----
+pyfasta split can handle > 52 files. (thanks Devtulya)
+
 0.4.4
 -----
 fix pyfasta extract

diff --git a/pyfasta/split_fasta.py b/pyfasta/split_fasta.py
@@ -45,7 +45,8 @@ def newnames(oldname, n, kmers=None, overlap=None, header=None):
     if n == 1:
         names = [pattern % "split"]
     else:
-        names = [pattern % string.letters[i] for i in range(n)]
+        width = len(str(n))
+        names = [pattern % str(i).rjust(width, '0') for i in range(n)]
     print >>sys.stderr, "creating new files:"
     print >>sys.stderr, "\n".join(names)
     return names

diff --git a/setup.py b/setup.py
@@ -1,7 +1,7 @@
 from setuptools import setup, find_packages
 
 
-version = '0.4.4'
+version = '0.4.5'
 
 setup(name='pyfasta',
       version=version,