Add version tracking

Closes #16

Author: jeromekelleher

bio2zarr/__init__.py

@@ -1,2 +1 @@
-
-from .vcf import explode as explode_vcf
+from . provenance import __version__

bio2zarr/__main__.py

@@ -2,17 +2,18 @@
 
 from . import cli
 
+@cli.version
 @click.group()
-def top_level():
+def bio2zarr():
     pass
 
 # Provide a single top-level interface to all of the functionality.
 # This probably isn't the recommended way of interacting, as we
 # install individual commands as console scripts. However, this
 # is handy for development and for those whose PATHs aren't set
 # up in the right way.
-top_level.add_command(cli.vcf2zarr)
-top_level.add_command(cli.plink2zarr)
+bio2zarr.add_command(cli.vcf2zarr)
+bio2zarr.add_command(cli.plink2zarr)
 
 if __name__ == "__main__":
-    top_level()
+    bio2zarr()

bio2zarr/cli.py

@@ -4,6 +4,7 @@
 
 from . import vcf
 from . import plink
+from . import provenance
 
 # Common arguments/options
 verbose = click.option("-v", "--verbose", count=True, help="Increase verbosity")
@@ -12,6 +13,7 @@
     "-p", "--worker-processes", type=int, default=1, help="Number of worker processes"
 )
 
+version = click.version_option(version=provenance.__version__)
 
 # Note: logging hasn't been implemented in the code at all, this is just
 # a first pass to try out some ways of doing things to see what works.
@@ -111,6 +113,7 @@ def validate(vcfs, out_path):
     vcf.validate(vcfs[0], out_path, show_progress=True)
 
 
+@version
 @click.group()
 def vcf2zarr():
     pass
@@ -145,6 +148,7 @@ def convert_plink(in_path, out_path, worker_processes, chunk_width, chunk_length
     )
 
 
+@version
 @click.group()
 def plink2zarr():
     pass

bio2zarr/provenance.py

@@ -0,0 +1,7 @@
+__version__ = "undefined"
+try:
+    from . import _version
+
+    __version__ = _version.version
+except ImportError:  # pragma: nocover
+    pass

bio2zarr/vcf.py

@@ -23,6 +23,7 @@
 import zarr
 
 from . import core
+from . import provenance
 
 logger = logging.getLogger(__name__)
 
@@ -1287,6 +1288,7 @@ def convert(
 
         sgvcf.root.attrs["vcf_zarr_version"] = "0.2"
         sgvcf.root.attrs["vcf_header"] = pcvcf.vcf_header
+        sgvcf.root.attrs["source"] = f"bio2zarr-{provenance.__version__}"
 
         progress_config = core.ProgressConfig(
             total=pcvcf.total_uncompressed_bytes,

setup.py

@@ -5,5 +5,5 @@
     # The package name along with all the other metadata is specified in setup.cfg
     # However, GitHub's dependency graph can't see the package unless we put this here.
     name="bio2zarr",
-    use_scm_version=True,
+    use_scm_version={"write_to": "bio2zarr/_version.py"},
 )

tests/test_cli.py

@@ -1,8 +1,11 @@
 from unittest import mock
 
+import pytest
 import click.testing as ct
 
 from bio2zarr import cli
+from bio2zarr import __main__ as main
+from bio2zarr import provenance
 
 
 class TestWithMocks:
@@ -117,3 +120,11 @@ def test_convert_plink(self):
                 chunk_length=None,
                 show_progress=True,
             )
+
+
+@pytest.mark.parametrize("cmd", [main.bio2zarr, cli.vcf2zarr, cli.plink2zarr])
+def test_version(cmd):
+    runner = ct.CliRunner(mix_stderr=False)
+    result = runner.invoke(cmd, ["--version"], catch_exceptions=False)
+    s = f"version {provenance.__version__}\n"
+    assert result.stdout.endswith(s)

tests/test_vcf_examples.py

@@ -6,6 +6,7 @@
 import cyvcf2
 
 from bio2zarr import vcf
+from bio2zarr import provenance
 
 
 class TestSmallExample:
@@ -38,6 +39,9 @@ def test_header(self, ds):
         vcf = cyvcf2.VCF(self.data_path)
         assert ds.attrs["vcf_header"] == vcf.raw_header
 
+    def test_source(self, ds):
+        assert ds.attrs["source"] == f"bio2zarr-{provenance.__version__}"
+
     def test_contigs(self, ds):
         nt.assert_array_equal(ds["contig_id"], ["19", "20", "X"])
         assert "contig_length" not in ds