Add support for backup cluster file

yuanx749 · Jun 26, 2024 · b5fa943 · b5fa943
1 parent 0f939af
commit b5fa943
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/pycdhit/__init__.py b/pycdhit/__init__.py
@@ -4,7 +4,7 @@
 from ._commands import *  # noqa: F403
 from ._io import *  # noqa: F403
 
-VERSION = "0.13.0"
+VERSION = "1.0.0"
 
 __all__ = [  # noqa: F405
     "CommandBase",

diff --git a/pycdhit/_io.py b/pycdhit/_io.py
@@ -70,13 +70,17 @@ def read_clstr(file: FilePath) -> pd.DataFrame:
     # refer to PrintInfo
     identifier, cluster, size, is_representative, identity = [], [], [], [], []
     coverage, strand = [], []  # distance is not used
+    with open(file) as f:
+        bak = f.read(1) != ">"
     with open(file) as f:
         for line in f:
             if line[0] == ">":
                 idx = int(re.search(r">Cluster (\d+)", line).group(1))
                 continue
-            cluster.append(idx)
             line = line.split()
+            if bak:
+                idx = int(line[0])
+            cluster.append(idx)
             size.append(int(re.search(r"(\d+)(aa|nt),", line[1]).group(1)))
             identifier.append(re.search(r">(.+)\.{3}", line[2]).group(1))
             if line[3] == "*":