ssi-analysis-result-parsers 0.0.3__py3-none-any.whl → 0.0.5__py3-none-any.whl

ssi_analysis_result_parsers/Legionella_parser.py

@@ -25,8 +25,10 @@ import json  # for nicely printing json and yaml
 # import functions from core module (optional, but most likely needed).
 from ssi_analysis_result_parsers import (
     core,
+    blast_parser,
 )
-from .blast_parser import extract_presence_absence
+
+# from ssi_analysis_result_parsers.blast_parser import extract_presence_absence
 
 # Project specific libraries
 from pathlib import Path
@@ -39,11 +41,18 @@ def extract_legionella_sbt(legionella_sbt_results_tsv: Path) -> dict:
     Returns dictionary of results found in the Legionella SBT summary output
     """
     if os.path.exists(legionella_sbt_results_tsv):
-        df = pandas.read_csv(legionella_sbt_results_tsv, sep="\t")
-        df.set_index("sample", inplace=True, drop=True)
-        d = df.to_dict(orient="index")
-        fname = next(iter(d))
-        return d[fname]
+        try:
+            df = pandas.read_csv(legionella_sbt_results_tsv, sep="\t")
+            df.set_index("sample", inplace=True, drop=True)
+            d = df.to_dict(orient="index")
+            fname = next(iter(d))
+            return d[fname]
+        except pandas.errors.EmptyDataError:
+            print(
+                f"No Legionella SBT output empty at {legionella_sbt_results_tsv}",
+                file=sys.stderr,
+            )
+            return None
     else:
         print(
             f"No Legionella SBT output found at {legionella_sbt_results_tsv}",
@@ -56,7 +65,7 @@ def legionella_summary(legionella_sbt_results_tsv: Path, lag1_blast_tsv: Path) -
     sbt_results_dict = extract_legionella_sbt(
         legionella_sbt_results_tsv=legionella_sbt_results_tsv
     )
-    lag1_blast_dict = extract_presence_absence(
+    lag1_blast_dict = blast_parser.extract_presence_absence(
         blast_output_tsv=lag1_blast_tsv,
         hits_as_string=False,
         include_match_stats=False,
@@ -65,6 +74,8 @@ def legionella_summary(legionella_sbt_results_tsv: Path, lag1_blast_tsv: Path) -
     results_dict = core.update_results_dict(
         sbt_results_dict, lag1_blast_dict, old_duplicate_key_prefix="SBT: "
     )
+    if results_dict is None:
+        return {}
     return results_dict
 
 
@@ -133,22 +144,24 @@ class LegionellaResults(core.PipelineResults):
             results_dict[sample_name] = legionella_results
         return cls(results_dict)"""
 
+    @staticmethod
     def legionella_summary(
         legionella_sbt_results_tsv: Path, lag1_blast_tsv: Path
     ) -> dict:
         sbt_results_dict = extract_legionella_sbt(
             legionella_sbt_results_tsv=legionella_sbt_results_tsv
         )
-        lag1_blast_dict = extract_presence_absence(
+        lag1_blast_dict = blast_parser.extract_presence_absence(
             blast_output_tsv=lag1_blast_tsv,
             hits_as_string=False,
             include_match_stats=False,
             gene_names=["lag-1"],
         )
-        print(lag1_blast_dict)
         results_dict = core.update_results_dict(
             sbt_results_dict, lag1_blast_dict, old_duplicate_key_prefix="SBT: "
         )
+        if results_dict is None:
+            return {}
         return results_dict
 
     def __repr__(self):

ssi_analysis_result_parsers/__init__.py

@@ -1 +1 @@
-__version__ = "0.0.2"
+__version__ = "0.0.5"

ssi_analysis_result_parsers/blast_parser.py

@@ -51,21 +51,31 @@ def extract_presence_absence(
     if os.path.exists(blast_output_tsv):
         try:
             blast_df = pandas.read_csv(blast_output_tsv, sep="\t", header=None)
+            header_list = tsv_header.split(" ")
+            if len(header_list) == len(blast_df.columns):
+                blast_df.columns = tsv_header.split(" ")
+                blast_df["plen"] = blast_df["length"] / blast_df["qlen"] * 100
+                blast_df_unique = (
+                    blast_df.sort_values(by=["bitscore"], ascending=False)
+                    .groupby("qseqid")
+                    .first()
+                )
+                blast_df_filtered = blast_df_unique.query(
+                    "plen > @plen_threshold and pident > @pident_threshold"
+                )
+                blast_dict = dict(blast_df_filtered.to_dict(orient="index"))
+            else:
+                print(
+                    f"Failed to parse {blast_output_tsv}. Number of columns do not match length of provided header string",
+                    file=sys.stderr,
+                )
+                return None
 
-            blast_df.columns = tsv_header.split(" ")
-            blast_df["plen"] = blast_df["length"] / blast_df["qlen"] * 100
-            blast_df_unique = (
-                blast_df.sort_values(by=["bitscore"], ascending=False)
-                .groupby("qseqid")
-                .first()
-            )
-            blast_df_filtered = blast_df_unique.query(
-                "plen > @plen_threshold and pident > @pident_threshold"
-            )
-            blast_dict = dict(blast_df_filtered.to_dict(orient="index"))
         except pandas.errors.EmptyDataError:
             blast_dict = {}
             print(f"Blast output file {blast_output_tsv} empty. Assuming 0 blast hits.")
+        except Exception as e:
+            print(f"Error parsing blast: e")
         if hits_as_string:
 
             results = []

{ssi_analysis_result_parsers-0.0.3.dist-info → ssi_analysis_result_parsers-0.0.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ssi_analysis_result_parsers
-Version: 0.0.3
+Version: 0.0.5
 Summary: TODO
 Home-page: https://github.com/thej-ssi/ssi_analysis_result_parsers
 Author: Thor Bech Johannesen

{ssi_analysis_result_parsers-0.0.3.dist-info → ssi_analysis_result_parsers-0.0.5.dist-info}/RECORD

@@ -1,15 +1,16 @@
-ssi_analysis_result_parsers/Legionella_parser.py,sha256=zBsSYfuUj2uN_LRqgTpOwhFUQF2jTLNvVzKbh6OPNxE,8878
-ssi_analysis_result_parsers/__init__.py,sha256=QvlVh4JTl3JL7jQAja76yKtT-IvF4631ASjWY1wS6AQ,22
+ssi_analysis_result_parsers/Legionella_parser.py,sha256=8n15TEiOY1OxAe4VePVzaXz7WFen6U1y3pM5Vsjsg7U,9292
+ssi_analysis_result_parsers/__init__.py,sha256=S7u1lbuWmM3A3ajykBialmPoJUK6Jg-WmNqM-9OZFdk,22
 ssi_analysis_result_parsers/_modidx.py,sha256=JY_GM0tMojzTtX9O4D8as4k5a-sXqkxkb7ZUEPzhuMk,12232
-ssi_analysis_result_parsers/blast_parser.py,sha256=L7EdW2LUwSS2OQO7WZUAxP6whJXdvTILE2a3O59uv-s,7441
+ssi_analysis_result_parsers/blast_parser.py,sha256=EBqWlx8bDlaSzqAZomiUGnT2DGaaA-L7ukny7SEJbpk,7915
 ssi_analysis_result_parsers/core.py,sha256=6TGURv8spPdBpwKv6LvqvbVzJChdeHwsG3WQ6QLUuvE,12124
 ssi_analysis_result_parsers/hello_world.py,sha256=jpN94sqYuNHqUbUZMCJ35qGY5iLPB_emucgnDGDUk_U,1895
 ssi_analysis_result_parsers/some_string.py,sha256=JwmAXKbX_JgY8UGh4FAu5-7ZjezcAEhq4Q2B73pWp2M,923
 ssi_analysis_result_parsers/config/config.default.env,sha256=Zt6bfPbVV3rYCksoebX1ruAdFgeD9wqAnKDtswhtJJM,1390
 ssi_analysis_result_parsers/config/config.default.yaml,sha256=3qgUrUtQpxrzYv7WQaHsvz9dQB0RALKNU0idxv7oRqM,460
-ssi_analysis_result_parsers-0.0.3.dist-info/licenses/LICENSE,sha256=p6aTb6QIfqyZ2Uux2VjV4F2zthdUSHZOjB4mfwGc7fo,1094
+ssi_analysis_result_parsers-0.0.5.dist-info/licenses/LICENSE,sha256=p6aTb6QIfqyZ2Uux2VjV4F2zthdUSHZOjB4mfwGc7fo,1094
 test_input/.DS_Store,sha256=sdTEvl9DTKPHNPYYjMqDepX7q7ZETlonk21tGEuWLao,6148
-test_input/Legionella/batch_parser_file_paths.tsv,sha256=zls11lmEA5U89d8RsX6PR8M1zXNVimeL4raqdZ3ijvQ,210
+test_input/empty_file.txt,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+test_input/Legionella/batch_parser_file_paths.tsv,sha256=AikBS_Ez1xO3UrEQ19AY3z6drBDdMAiSGK66NLeyYj4,356
 test_input/Legionella/lag-1_blast.tsv,sha256=MN5QL_iBn9gQ8VTYEcTnT0JwKgpkD8G15-QFOrSWxkU,1133
 test_input/Legionella/lag-1_blast_2.tsv,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 test_input/Legionella/test.sbt.tsv,sha256=ibhaH3is2dxHaABPvR2QM2HAq9bKOs1AwOTmrwSrcd8,168
@@ -20,8 +21,8 @@ test_input/blast_parser/gene_presence_absence_test.tsv,sha256=qCvMkBC-1GuXx83RDh
 test_output/output_with_sample_name.tsv,sha256=NQG7WaxczuWCCsX2a9MUxCCYpbuAirz9gw08OLdEdUo,41
 test_output/test.tsv,sha256=6DGzarXMkUP03Z58vZimc-gu1K2k84zxZLWWF2HROCg,277
 test_output/test_batch_output.tsv,sha256=6DGzarXMkUP03Z58vZimc-gu1K2k84zxZLWWF2HROCg,277
-ssi_analysis_result_parsers-0.0.3.dist-info/METADATA,sha256=LPIINEBm-fhPx_sG1lS1xLf2ZIWtHXU97hFFW2M58fE,2765
-ssi_analysis_result_parsers-0.0.3.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-ssi_analysis_result_parsers-0.0.3.dist-info/entry_points.txt,sha256=eG2NzlNDoG__0PPHl3eoKK5EXIz02BGhRX-L2aWgKCY,447
-ssi_analysis_result_parsers-0.0.3.dist-info/top_level.txt,sha256=3q56bBc2Wv2a6ZQ1l_9m66vot2-Qu6tM9tDr3QQ8auM,81
-ssi_analysis_result_parsers-0.0.3.dist-info/RECORD,,
+ssi_analysis_result_parsers-0.0.5.dist-info/METADATA,sha256=BF-cuY_EJow8haoGw99WeGWAf_zWWQdWQ4OFu42NtcM,2765
+ssi_analysis_result_parsers-0.0.5.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+ssi_analysis_result_parsers-0.0.5.dist-info/entry_points.txt,sha256=eG2NzlNDoG__0PPHl3eoKK5EXIz02BGhRX-L2aWgKCY,447
+ssi_analysis_result_parsers-0.0.5.dist-info/top_level.txt,sha256=3q56bBc2Wv2a6ZQ1l_9m66vot2-Qu6tM9tDr3QQ8auM,81
+ssi_analysis_result_parsers-0.0.5.dist-info/RECORD,,

test_input/Legionella/batch_parser_file_paths.tsv

@@ -1,3 +1,5 @@
 sample_name	sbt_results	lag1_blast_results
 sample_1	test_input/Legionella/test.sbt.tsv	test_input/Legionella/lag-1_blast.tsv
 sample_2	test_input/Legionella/test2.sbt.tsv	test_input/Legionella/lag-1_blast_2.tsv
+sample_3	test_input/Legionella/test2.sbt.tsv	test_input/empty_file.txt
+sample_4	test_input/empty_file.txt	test_input/Legionella/lag-1_blast_2.tsv