special case md5 prefixes, for prefetch

sourmash-bio · Jun 13, 2021 · e205e64 · e205e64
1 parent b57b2b3
commit e205e64
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/sourmash/index.py b/src/sourmash/index.py
@@ -509,6 +509,15 @@ def yield_fp():
                             print(f"({p} found!")
                             yield self.zf.open(zipinfo)
                             break
+        elif picklist and picklist.coltype == 'md5prefix8':
+            def yield_fp():
+                for zipinfo in self.zf.infolist():
+                    if zipinfo.filename.startswith('signatures/'):
+                        fn = zipinfo.filename[len('signatures/'):]
+                        prefix = fn[:8]
+                        if prefix in picklist.pickset:
+                            print(f"({prefix} q found)")
+                            yield self.zf.open(zipinfo)
         else:
             def yield_fp():
                 for zipinfo in self.zf.infolist():