loqusdb 2.7.17__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.17"
+__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

@@ -95,6 +95,13 @@ def validate_profile_threshold(ctx, param, value):
     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,
@@ -112,6 +119,7 @@ def load(
     soft_threshold,
     qual_gq,
     snv_gq_only,
+    ignore_gq_if_unset,
 ):
     """Load the variants of a case
 
@@ -160,6 +168,7 @@ def load(
             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

@@ -39,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
 
@@ -56,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)
@@ -152,6 +155,7 @@ def load_database(
                 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
@@ -199,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)
@@ -229,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.17.dist-info → loqusdb-2.7.18.dist-info}/METADATA

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

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

@@ -1,16 +1,16 @@
-loqusdb/__init__.py,sha256=OKW6E-XujZmZRrXRSOInDNnGqybJIkbeIBvnFcWqhuo,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=6rrt_XIdUHhNc-HisMRS7G-bWkgYhqggDjbEQubYmUQ,4635
+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=1UVB5jUy1DESZdWnRrMcyfFiUgqmj9l-Bm0DZOHetfg,8428
+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.17.dist-info/LICENSE,sha256=urpFcJXw3elN9kV2fFutc-lXegjuu2lqP_GSy8_CAbs,1054
-loqusdb-2.7.17.dist-info/METADATA,sha256=WH2QGz7PhKif97YZxHnOOxmiaN96ZAd_zGCfUnm2GGU,5321
-loqusdb-2.7.17.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-loqusdb-2.7.17.dist-info/entry_points.txt,sha256=wFoWzEFjsSgXkj9FMQA8C9ihZoJ9R1XvbGuX9hEEI6E,52
-loqusdb-2.7.17.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,,