loqusdb 2.7.16__py3-none-any.whl → 2.7.18__py3-none-any.whl

loqusdb/__init__.py

@@ -4,7 +4,7 @@ from pymongo import ASCENDING, IndexModel
 
 logger = logging.getLogger(__name__)
 
-__version__ = "2.7.16"
+__version__ = "2.7.18"
 
 INDEXES = {
     "variant": [

loqusdb/build_models/variant.py

@@ -148,6 +148,7 @@ def build_variant(
     case_id: Optional[str] = None,
     gq_threshold: Optional[int] = None,
     gq_qual: Optional[bool] = False,
+    ignore_gq_if_unset: Optional[bool] = False,
     genome_build: Optional[str] = None,
 ) -> Variant:
     """Return a Variant object
@@ -164,6 +165,8 @@ def build_variant(
         case_id(str): The case id
         gq_threshold(int): Genotype Quality threshold
         gq_qual(bool): Use variant.QUAL for quality instead of GQ
+        ignore_gq_if_unset(bool): Ignore GQ threshold check for variants that do not have GQ or QUAL set.
+        genome_build(str): Genome build. Ex. GRCh37 or GRCh38
 
     Return:
         formated_variant(models.Variant): A variant dictionary
@@ -201,14 +204,15 @@ def build_variant(
             ind_pos = ind_obj["ind_index"]
 
             if gq_qual:
-                gq = 0
+                gq = -1
                 if variant.QUAL:
                     gq = int(variant.QUAL)
 
             if not gq_qual:
                 gq = int(variant.gt_quals[ind_pos])
 
-            if gq_threshold and gq < gq_threshold:
+            # When gq is missing in FORMAT cyvcf2 assigns a score of -1
+            if (gq_threshold and 0 <= gq < gq_threshold) or (gq == -1 and not ignore_gq_if_unset):
                 continue
 
             genotype = GENOTYPE_MAP[variant.gt_types[ind_pos]]

loqusdb/commands/load.py

@@ -88,6 +88,20 @@ def validate_profile_threshold(ctx, param, value):
     callback=validate_profile_threshold,
     help="profile hamming distance to store similar individuals (0-1)",
 )
+@click.option(
+    "--snv-gq-only",
+    is_flag=True,
+    default=False,
+    show_default=True,
+    help="Apply GQ threshold only to SNV variants",
+)
+@click.option(
+    "--ignore-gq-if-unset",
+    is_flag=True,
+    default=False,
+    show_default=True,
+    help="Ignore GQ threshold if GQ (or the QUAL field for --qual-gq) is unset in VCF",
+)
 @click.pass_context
 def load(
     ctx,
@@ -104,6 +118,8 @@ def load(
     hard_threshold,
     soft_threshold,
     qual_gq,
+    snv_gq_only,
+    ignore_gq_if_unset,
 ):
     """Load the variants of a case
 
@@ -145,12 +161,14 @@ def load(
             skip_case_id=skip_case_id,
             case_id=case_id,
             gq_threshold=gq_threshold,
+            snv_gq_only=snv_gq_only,
             qual_gq=qual_gq,
             max_window=max_window,
             profile_file=variant_profile_path,
             hard_threshold=hard_threshold,
             soft_threshold=soft_threshold,
             genome_build=genome_build,
+            ignore_gq_if_unset=ignore_gq_if_unset,
         )
     except (SyntaxError, CaseError, IOError) as error:
         LOG.warning(error)

loqusdb/utils/load.py

@@ -31,6 +31,7 @@ def load_database(
     family_type="ped",
     skip_case_id=False,
     gq_threshold=None,
+    snv_gq_only=False,
     qual_gq=False,
     case_id=None,
     max_window=3000,
@@ -38,6 +39,7 @@ def load_database(
     hard_threshold=0.95,
     soft_threshold=0.9,
     genome_build=None,
+    ignore_gq_if_unset=False,
 ):
     """Load the database with a case and its variants
 
@@ -55,6 +57,8 @@ def load_database(
           check_profile(bool): Does profile check if True
           hard_threshold(float): Rejects load if hamming distance above this is found
           soft_threshold(float): Stores similar samples if hamming distance above this is found
+          genome_build(str): Store the genome version
+          ignore_gq_if_unset(str): Ignore the gq threhsold check for variants that do not have a GQ or QUAL set
 
     Returns:
           nr_inserted(int)
@@ -88,11 +92,13 @@ def load_database(
             adapter, profiles, hard_threshold=hard_threshold, soft_threshold=soft_threshold
         )
 
-    # If a gq threshold is used the variants needs to have GQ
+    # If a gq threshold is used the variants need to have GQ (only SNVs if snv_gq_only)
     for _vcf_file in vcf_files:
-        # Get a cyvcf2.VCF object
-        vcf = get_vcf(_vcf_file)
+        is_sv = _vcf_file == sv_file
+        if snv_gq_only and is_sv:
+            continue  # skip GQ check for SV VCF
 
+        vcf = get_vcf(_vcf_file)
         if gq_threshold and not vcf.contains("GQ") and not qual_gq:
             LOG.warning("Set gq-threshold to 0 or add info to vcf {0}".format(_vcf_file))
             raise SyntaxError("GQ is not defined in vcf header")
@@ -144,11 +150,12 @@ def load_database(
                 vcf_obj=vcf_obj,
                 case_obj=case_obj,
                 skip_case_id=skip_case_id,
-                gq_threshold=gq_threshold,
+                gq_threshold=gq_threshold if not snv_gq_only or variant_type == "snv" else None,
                 qual_gq=qual_gq,
                 max_window=max_window,
                 variant_type=variant_type,
                 genome_build=genome_build,
+                ignore_gq_if_unset=ignore_gq_if_unset,
             )
         except Exception as err:
             # If something went wrong do a rollback
@@ -196,18 +203,22 @@ def load_variants(
     max_window=3000,
     variant_type="snv",
     genome_build=None,
+    ignore_gq_if_unset=False,
 ):
     """Load variants for a family into the database.
 
     Args:
         adapter (loqusdb.plugins.Adapter): initialized plugin
+        vcf_obj(cyvcf2.VCF): Iterable with cyvcf2.Variant
         case_obj(Case): dict with case information
-        nr_variants(int)
         skip_case_id (bool): whether to include the case id on variant level
                              or not
         gq_threshold(int)
+        qual_gq(bool): whether to use QUAL instead of GQ
         max_window(int): Specify the max size for sv windows
         variant_type(str): 'sv' or 'snv'
+        genome_build(str): Genome version. Ex. GRCH37
+        ignore_gq_if_unset (bool): whether to add entries that have missing GQ or QUAL field
 
     Returns:
         nr_inserted(int)
@@ -226,7 +237,13 @@ def load_variants(
 
         variants = (
             build_variant(
-                variant, case_obj, case_id, gq_threshold, qual_gq, genome_build=genome_build
+                variant,
+                case_obj,
+                case_id,
+                gq_threshold,
+                qual_gq,
+                ignore_gq_if_unset,
+                genome_build=genome_build,
             )
             for variant in bar
         )

{loqusdb-2.7.16.dist-info → loqusdb-2.7.18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: loqusdb
-Version: 2.7.16
+Version: 2.7.18
 Summary: A simple observation count database
 License: MIT
 Author: Your Name

{loqusdb-2.7.16.dist-info → loqusdb-2.7.18.dist-info}/RECORD

@@ -1,16 +1,16 @@
-loqusdb/__init__.py,sha256=D1R2iXsppIzZJxfYNMSdRlrzsMKf-4z-tSMYAFWtEPw,1688
+loqusdb/__init__.py,sha256=r4jbFPCtsyXv8VehXkXxBx1HDPPeaidp1mmwpN_Ghrg,1688
 loqusdb/__main__.py,sha256=8FGKySAGaWSzAYMj6HRsxeyiME3V01Idt7HrmN7pSYY,397
 loqusdb/build_models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 loqusdb/build_models/case.py,sha256=AByutEYK2N3kS9JFvyZfPKNZdCpZHCSD0nNHAgaU1Cs,4127
 loqusdb/build_models/profile_variant.py,sha256=TbSxfVjESstS_FgbkOW4NQwMQVeTyhn9oc9yPZmDhzI,1021
-loqusdb/build_models/variant.py,sha256=2ate8viYhHK1yX7UvCzEPgfuuXwTgAApU_2E5jl1ZO4,6934
+loqusdb/build_models/variant.py,sha256=Vee9FVwCC03wo6u2uI7qH9rgfevndBf364YSmE3Js2k,7271
 loqusdb/commands/__init__.py,sha256=BXAN3UADgqPrkGczzjlLO9GyyQ96dnLnP7n92JlYHgo,603
 loqusdb/commands/annotate.py,sha256=748kImopE5WbaO1nuv3WUgIqezWFSsi7SBeWhOz26-s,1384
 loqusdb/commands/cli.py,sha256=wJD5S1BoCxtRTAobd1QtmQpzlngJg-mt1nsyD92fDD4,3176
 loqusdb/commands/delete.py,sha256=R6ysHKSMw1mmL4ZbktoUIKzdzDLQ3314YPYhIy1myic,1979
 loqusdb/commands/export.py,sha256=HKoRzUo_BHNOdw_TcKUId9TTowi8VJVGqnuDlK-FqFE,3531
 loqusdb/commands/identity.py,sha256=KLA9c8e6cJFDxtqIa1G6zdHTHK1sz2b3v1Utdtik_4k,787
-loqusdb/commands/load.py,sha256=sDGCOGbukdfebLn1iBGyWoI6t3tijLzywP1tg3WheQI,4431
+loqusdb/commands/load.py,sha256=QMUSzdaCi1lVC9F4u0xkAJtVz4N_tMPh3Nu6Obl9ky4,4903
 loqusdb/commands/load_profile.py,sha256=cflCbF9f77_HCH8xPnN8zSSocvIffRMnC2LPE0j7Xq8,3336
 loqusdb/commands/migrate.py,sha256=2C8YL-zVqnpnqg3JIyUr0rbVnb8-AGPVWNhicHnPKLo,667
 loqusdb/commands/restore.py,sha256=eqPX0yao0IAYS5SbjCdlsfSJRBbRByBLISUU2hTzqqs,1492
@@ -43,14 +43,14 @@ loqusdb/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 loqusdb/utils/annotate.py,sha256=cPNWlhsv6yoe3lxNfa9DytO5eACuM_mOJJw_mglVMN0,2646
 loqusdb/utils/case.py,sha256=aeTvyACJTDjzl-aOjAZaUzFMLisgFKMfcoXSvNAZz4s,2168
 loqusdb/utils/delete.py,sha256=-ddBM_QXKzlUN6egEJggKzXX1P-WEdi92HgaD1DJRtg,4843
-loqusdb/utils/load.py,sha256=ZdsnmHTVGOjwG24Fm5gcUu7tcLwYSqvh3W9j6ixqyF0,8239
+loqusdb/utils/load.py,sha256=LyYHPwaCVV-7edbaoPahah05yPfG9H4VCYSg1gmz_M0,9063
 loqusdb/utils/migrate.py,sha256=9Q6kdIi9TpFVzDYptlEE8RqPPS5wyzfM3F8egzmmBBk,1113
 loqusdb/utils/profiling.py,sha256=3OizF7CpYvSl9kyl2g4KGJxbIRUqWfmfLxn3843XYDk,9164
 loqusdb/utils/update.py,sha256=1edJG-u24FgOSxyXAQEiyTG4IyK-Uo3lSIl5qyzcXsI,4433
 loqusdb/utils/variant.py,sha256=U6nMZRUf5NDDQ74nG0HBCLMnFQVgFAT6eHll_F2uiwc,2087
 loqusdb/utils/vcf.py,sha256=ybmrTBEPYa0FbUXo8ttlwATk13RnKjX9eIDbRDwCiVE,5175
-loqusdb-2.7.16.dist-info/LICENSE,sha256=urpFcJXw3elN9kV2fFutc-lXegjuu2lqP_GSy8_CAbs,1054
-loqusdb-2.7.16.dist-info/METADATA,sha256=Aayv73fzmF_pCfXHNoRZEzen9CbZqZccc5pfwzMXaCE,5321
-loqusdb-2.7.16.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-loqusdb-2.7.16.dist-info/entry_points.txt,sha256=wFoWzEFjsSgXkj9FMQA8C9ihZoJ9R1XvbGuX9hEEI6E,52
-loqusdb-2.7.16.dist-info/RECORD,,
+loqusdb-2.7.18.dist-info/LICENSE,sha256=urpFcJXw3elN9kV2fFutc-lXegjuu2lqP_GSy8_CAbs,1054
+loqusdb-2.7.18.dist-info/METADATA,sha256=yTsm5y8i4u_chGfnAbMm5ZfjorJOVZQfl4qK9vdJDCk,5321
+loqusdb-2.7.18.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+loqusdb-2.7.18.dist-info/entry_points.txt,sha256=wFoWzEFjsSgXkj9FMQA8C9ihZoJ9R1XvbGuX9hEEI6E,52
+loqusdb-2.7.18.dist-info/RECORD,,

{loqusdb-2.7.16.dist-info → loqusdb-2.7.18.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.1.2
+Generator: poetry-core 2.1.3
 Root-Is-Purelib: true
 Tag: py3-none-any