From fa370f8c52c24392378aeced3f17f540127b49c5 Mon Sep 17 00:00:00 2001
From: widdowquinn <leighton.pritchard@hutton.ac.uk>
Date: Wed, 21 Sep 2016 13:58:23 +0100
Subject: [PATCH] add pyani version to log output

---
 average_nucleotide_identity.py | 30 +++++++++++++++++-------------
 1 file changed, 17 insertions(+), 13 deletions(-)

diff --git a/average_nucleotide_identity.py b/average_nucleotide_identity.py
index a9cbc41f..c795d14c 100755
--- a/average_nucleotide_identity.py
+++ b/average_nucleotide_identity.py
@@ -170,6 +170,7 @@
 from pyani import run_multiprocessing as run_mp
 from pyani import run_sge
 from pyani.pyani_config import params_mpl, params_r, ALIGNDIR, FRAGSIZE
+from pyani import __version__ as VERSION
 
 
 # Process command-line arguments
@@ -580,19 +581,21 @@ def get_labels(filename):
     Input files should be formatted as <key>\t<label>, one pair per line.
     """
     labeldict = {}
-    with open(filename, 'rU') as fh:
-        count = 0
-        for line in fh.readlines():
-            count += 1
-            try:
-                key, label = line.strip().split('\t')
-            except ValueError:
-                logger.warning("Problem with class file: %s" % filename)
-                logger.warning("%d: %s" % (count, line.strip()))
-                logger.warning("(skipping line)")
-                continue
-            else:
-                labeldict[key] = label
+    if filename is not None:
+        logger.info("Reading labels from %s" % filename)
+        with open(filename, 'rU') as fh:
+            count = 0
+            for line in fh.readlines():
+                count += 1
+                try:
+                    key, label = line.strip().split('\t')
+                except ValueError:
+                    logger.warning("Problem with class file: %s" % filename)
+                    logger.warning("%d: %s" % (count, line.strip()))
+                    logger.warning("(skipping line)")
+                    continue
+                else:
+                    labeldict[key] = label
     return labeldict
                
 
@@ -725,6 +728,7 @@ def subsample_input(infiles):
     logger.addHandler(err_handler)
 
     # Report arguments, if verbose
+    logger.info("pyani version: %s" % VERSION)
     logger.info(args)
     logger.info("command-line: %s" % ' '.join(sys.argv))