# HG changeset patch
# User greg
# Date 1693322227 0
# Node ID 5aa8560d9d91edeb5db735247ba0d4bc98025e9f
# Parent  8fe04662585c96e6869f58606ad41d414df629d9
Uploaded

diff -r 8fe04662585c -r 5aa8560d9d91 lrn_risk.py
--- a/lrn_risk.py	Tue Aug 29 01:23:22 2023 +0000
+++ b/lrn_risk.py	Tue Aug 29 15:17:07 2023 +0000
@@ -133,11 +133,15 @@
                 freetext = '{0}/{1} ({2}%)'.format(pres, denom, perc)
             elif gtdb != '(Unknown Species)':
                 ann = 'NA'
-                denom = gtdbd[gtdb]
+                if gtdb in gtdbd.keys():
+                    denom = gtdbd[gtdb]
+                else:
+                    denom = 'NA' 
                 freetext = "*WARNING"
-                warnings.append("*WARNING: This gene has never been detected in this species! Interpret with caution!")
+                warnings.append("*WARNING: This gene has never been detected in this species and/or this species has not been included in the LRNRisk database! Interpret with caution!")
             else:
                 ann = 'NA'
+                denom = 'NA'
                 freetext = "**WARNING"
                 warnings.append("**WARNING: This genome belongs to an undescribed species. Interpret with caution!")
             finallines.append('%s\t%s\t%s' % (bv, ann, freetext))