taxoniumtools 2.0.116__py3-none-any.whl → 2.0.117__py3-none-any.whl

taxoniumtools/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '2.0.116'
-__version_tuple__ = version_tuple = (2, 0, 116)
+__version__ = version = '2.0.117'
+__version_tuple__ = version_tuple = (2, 0, 117)

taxoniumtools/usher_to_taxonium.py

@@ -28,6 +28,7 @@ def do_processing(input_file,
                   chronumental_date_output=None,
                   chronumental_tree_output=None,
                   chronumental_reference_node=None,
+                  chronumental_add_inferred_date=None,
                   config_file=None,
                   title=None,
                   overlay_html=None,
@@ -82,7 +83,10 @@ def do_processing(input_file,
             metadata_file=metadata_file,
             chronumental_steps=chronumental_steps,
             chronumental_date_output=chronumental_date_output,
-            chronumental_tree_output=chronumental_tree_output)
+            chronumental_tree_output=chronumental_tree_output,
+            chronumental_add_inferred_date=chronumental_add_inferred_date,
+            metadata_dict=metadata_dict,
+            metadata_cols=metadata_cols)
 
     print("Ladderizing tree..")
     mat.tree.ladderize(ascending=False)
@@ -236,6 +240,12 @@ def get_parser():
         help=
         "A reference node to be used for Chronumental. This should be earlier in the outbreak and have a good defined date. If not set the oldest sample will be automatically picked by Chronumental.",
         default=None)
+    parser.add_argument(
+        "--chronumental_add_inferred_date",
+        type=str,
+        help=
+        "A new metadata-column-like name to be added for display with the value of Chronumental's inferred date for each sample.",
+        default=None)
     parser.add_argument(
         '-j',
         "--config_json",
@@ -304,26 +314,28 @@ def main():
     parser = get_parser()
 
     args = parser.parse_args()
-    do_processing(args.input,
-                  args.output,
-                  metadata_file=args.metadata,
-                  genbank_file=args.genbank,
-                  chronumental_enabled=args.chronumental,
-                  chronumental_steps=args.chronumental_steps,
-                  columns=args.columns,
-                  chronumental_date_output=args.chronumental_date_output,
-                  chronumental_tree_output=args.chronumental_tree_output,
-                  chronumental_reference_node=args.chronumental_reference_node,
-                  config_file=args.config_json,
-                  title=args.title,
-                  overlay_html=args.overlay_html,
-                  remove_after_pipe=args.remove_after_pipe,
-                  clade_types=args.clade_types,
-                  name_internal_nodes=args.name_internal_nodes,
-                  shear=args.shear,
-                  shear_threshold=args.shear_threshold,
-                  only_variable_sites=args.only_variable_sites,
-                  key_column=args.key_column)
+    do_processing(
+        args.input,
+        args.output,
+        metadata_file=args.metadata,
+        genbank_file=args.genbank,
+        chronumental_enabled=args.chronumental,
+        chronumental_steps=args.chronumental_steps,
+        columns=args.columns,
+        chronumental_date_output=args.chronumental_date_output,
+        chronumental_tree_output=args.chronumental_tree_output,
+        chronumental_reference_node=args.chronumental_reference_node,
+        chronumental_add_inferred_date=args.chronumental_add_inferred_date,
+        config_file=args.config_json,
+        title=args.title,
+        overlay_html=args.overlay_html,
+        remove_after_pipe=args.remove_after_pipe,
+        clade_types=args.clade_types,
+        name_internal_nodes=args.name_internal_nodes,
+        shear=args.shear,
+        shear_threshold=args.shear_threshold,
+        only_variable_sites=args.only_variable_sites,
+        key_column=args.key_column)
 
 
 if __name__ == "__main__":

taxoniumtools/utils.py

@@ -1,7 +1,7 @@
 from alive_progress import alive_it, alive_bar
 import pandas as pd
 import warnings
-import os, tempfile, sys
+import os, tempfile, sys, errno
 import treeswift
 import shutil
 from . import ushertools
@@ -34,7 +34,7 @@ def read_metadata(metadata_file, columns, key_column):
             )
 
         metadata_dict = metadata.to_dict("index")
-        metadata_cols = metadata.columns
+        metadata_cols = metadata.columns.tolist()
         del metadata
         print("Metadata loaded")
         return metadata_dict, metadata_cols
@@ -46,7 +46,8 @@ def read_metadata(metadata_file, columns, key_column):
 
 def do_chronumental(mat, chronumental_reference_node, metadata_file,
                     chronumental_steps, chronumental_date_output,
-                    chronumental_tree_output):
+                    chronumental_tree_output, chronumental_add_inferred_date,
+                    metadata_dict, metadata_cols):
     chronumental_is_available = os.system(
         "which chronumental > /dev/null") == 0
     if not chronumental_is_available:
@@ -93,6 +94,42 @@ def do_chronumental(mat, chronumental_reference_node, metadata_file,
         del time_tree
         del time_tree_iter
 
+        if chronumental_add_inferred_date:
+            print(
+                f"Adding chronumental inferred date as metadata-like item {chronumental_add_inferred_date}"
+            )
+            if chronumental_date_output:
+                date_output_filename = chronumental_date_output
+            else:
+                metadata_dir = os.path.dirname(metadata_file)
+                metadata_base = os.path.basename(metadata_file)
+                date_output_filename = f"chronumental_dates_{metadata_base}.tsv"
+                if metadata_dir:
+                    date_output_filename = metadata_dir + "/" + date_output_filename
+                if not os.path.exists(date_output_filename):
+                    raise FileNotFoundError(
+                        errno.ENOENT,
+                        f"Can't find default date output file in the expected location (try specifying a file name with --chronumental_date_output)",
+                        date_output_filename)
+            metadata_cols.append(chronumental_add_inferred_date)
+            inferred_dates = pd.read_csv(
+                date_output_filename,
+                sep="\t" if metadata_file.endswith(".tsv")
+                or metadata_file.endswith(".tsv.gz") else ",",
+                usecols=['strain', 'predicted_date'])
+            for idx, row in inferred_dates.iterrows():
+                node_name = row['strain']
+                inferred_date = row['predicted_date']
+                # Add inferred_date even if a node (e.g. internal node) has no metadata
+                if node_name not in metadata_dict:
+                    metadata_dict[node_name] = {
+                        col: ""
+                        for col in metadata_cols
+                    }
+                metadata_dict[node_name][
+                    chronumental_add_inferred_date] = inferred_date
+            del inferred_dates
+
 
 def set_x_coords(root, chronumental_enabled):
     """ Set x coordinates for the tree"""

{taxoniumtools-2.0.116.dist-info → taxoniumtools-2.0.117.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: taxoniumtools
-Version: 2.0.116
+Version: 2.0.117
 Summary: Generate files for taxonium
 Home-page: https://github.com/theosanderson/taxonium
 Author: Theo Sanderson
@@ -13,10 +13,10 @@ Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 Requires-Dist: treeswift
 Requires-Dist: alive-progress
-Requires-Dist: biopython <=1.81
+Requires-Dist: biopython<=1.81
 Requires-Dist: pandas
 Requires-Dist: google-api-python-client
-Requires-Dist: protobuf <4
+Requires-Dist: protobuf<4
 Requires-Dist: orjson
 Requires-Dist: psutil
 Requires-Dist: docker

taxoniumtools-2.0.117.dist-info/RECORD

@@ -0,0 +1,14 @@
+taxoniumtools/__init__.py,sha256=iizd2XLvtBHVDz6j82ZWJ2eojVvbTmtJCWQGk5MtqhE,51
+taxoniumtools/__main__.py,sha256=0-mrc1_4NaJur-6SQXV6vGrJFFDuRRFvqeQvnlOA5l4,25
+taxoniumtools/_version.py,sha256=iMMP4FrmJNydED5EK5ogxaESOZO_aSCQRTPLu1wUn8A,415
+taxoniumtools/newick_to_taxonium.py,sha256=05JMQgmYLEo4nZVGPB0afAlF2s93WE7Ka_jwd0kCw0A,6207
+taxoniumtools/parsimony_pb2.py,sha256=hE6jWYlgTpUbTS1uTjRKwVgPNO232QIu3LqjpJUlthU,15575
+taxoniumtools/usher_to_taxonium.py,sha256=fir7_p18f8CaFGX7H1tRTm5YSryMhmcuA_PlL3sY2m4,12295
+taxoniumtools/ushertools.py,sha256=duMipZuWfEtfOx-NosZe-6XSIL5YdKHGWKJbM9jM-sQ,16417
+taxoniumtools/utils.py,sha256=sNbadwK7CNLK-sQFf31Au8nG1rUmcrbGMKDSV7UJMJo,10477
+taxoniumtools/view_taxonium.py,sha256=LYWBhaXYrmOOrIHI2bBYhqNUw-HZRtYzdvMQb6ZombI,4640
+taxoniumtools-2.0.117.dist-info/METADATA,sha256=Ga_3Z1nv2drE6byceEMfHb_DgGQeSBH6HyGAXwvZtiU,2072
+taxoniumtools-2.0.117.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+taxoniumtools-2.0.117.dist-info/entry_points.txt,sha256=oIcDnAw81KjfG32-ePuBFvqS_wbLXWB8NmWo61o1LW0,183
+taxoniumtools-2.0.117.dist-info/top_level.txt,sha256=EEWGvODb1nR_CiPHUDxcfU4UzxtPnVN4yTywlT3Z0nE,14
+taxoniumtools-2.0.117.dist-info/RECORD,,

{taxoniumtools-2.0.116.dist-info → taxoniumtools-2.0.117.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (71.1.0)
+Generator: setuptools (75.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

taxoniumtools-2.0.116.dist-info/RECORD

@@ -1,14 +0,0 @@
-taxoniumtools/__init__.py,sha256=iizd2XLvtBHVDz6j82ZWJ2eojVvbTmtJCWQGk5MtqhE,51
-taxoniumtools/__main__.py,sha256=0-mrc1_4NaJur-6SQXV6vGrJFFDuRRFvqeQvnlOA5l4,25
-taxoniumtools/_version.py,sha256=syXcackkFxZ9RzJmN02y3UUyBKil9elFBwTnYNzz328,415
-taxoniumtools/newick_to_taxonium.py,sha256=05JMQgmYLEo4nZVGPB0afAlF2s93WE7Ka_jwd0kCw0A,6207
-taxoniumtools/parsimony_pb2.py,sha256=hE6jWYlgTpUbTS1uTjRKwVgPNO232QIu3LqjpJUlthU,15575
-taxoniumtools/usher_to_taxonium.py,sha256=RV_jUTHQlmqvIZw0TmbjYforUK0mLnN6u5VdIhgD0MA,11934
-taxoniumtools/ushertools.py,sha256=duMipZuWfEtfOx-NosZe-6XSIL5YdKHGWKJbM9jM-sQ,16417
-taxoniumtools/utils.py,sha256=2qDm4mbFZmOMsw3vLymlyHciHDSPS8LJycMJro2zTbU,8502
-taxoniumtools/view_taxonium.py,sha256=LYWBhaXYrmOOrIHI2bBYhqNUw-HZRtYzdvMQb6ZombI,4640
-taxoniumtools-2.0.116.dist-info/METADATA,sha256=LEoWTWcg70SD9CS7Dw3s_fuBQERljonrOVNxUmrWRk8,2074
-taxoniumtools-2.0.116.dist-info/WHEEL,sha256=Wyh-_nZ0DJYolHNn1_hMa4lM7uDedD_RGVwbmTjyItk,91
-taxoniumtools-2.0.116.dist-info/entry_points.txt,sha256=oIcDnAw81KjfG32-ePuBFvqS_wbLXWB8NmWo61o1LW0,183
-taxoniumtools-2.0.116.dist-info/top_level.txt,sha256=EEWGvODb1nR_CiPHUDxcfU4UzxtPnVN4yTywlT3Z0nE,14
-taxoniumtools-2.0.116.dist-info/RECORD,,