avoids crashing if OG annotation not in db. fixes #8

eggnogdb · Apr 18, 2017 · 237788e · 237788e
1 parent 1db6845
commit 237788e
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/eggnogmapper/annota.py b/eggnogmapper/annota.py
@@ -31,8 +31,12 @@ def get_og_annotations(ogname):
     cmd = 'SELECT level, nm, description, COG_categories FROM og WHERE og.og = "%s"' % ogname
     level, nm, desc, cat = None, None, None, None
     if db.execute(cmd):
-        level, nm, desc, cat = db.fetchone()
-        cat = re.sub(cog_cat_cleaner, '', cat)
+        try:
+            level, nm, desc, cat = db.fetchone()
+        except TypeError:                       # avoids crashing ig no annotation for OG viral
+            desc, cat = '', ''
+        else:
+            cat = re.sub(cog_cat_cleaner, '', cat)
     return level, nm, desc, cat