service: dev: Add version bump command

Signed-off-by: John Andersen <[email protected]>

Author: pdxjohnny

dffml/service/dev.py

@@ -1,8 +1,10 @@
 import os
 import sys
+import ast
 import json
 import pydoc
 import asyncio
+import pathlib
 import getpass
 import importlib
 import contextlib
@@ -434,6 +436,127 @@ async def run(self):
                     raise RuntimeError
 
 
+class BumpMain(CMD):
+    """
+    Bump the version number of the main package within the dependency list of
+    each plugin.
+    """
+
+    async def run(self):
+        main_package = is_develop("dffml")
+        if not main_package:
+            raise NotImplementedError(
+                "Need to reinstall the main package in development mode."
+            )
+        # TODO Implement this in Python
+        proc = await asyncio.create_subprocess_exec(
+            "bash", str(pathlib.Path(main_package, "scripts", "bump_deps.sh"))
+        )
+        await proc.wait()
+        if proc.returncode != 0:
+            raise RuntimeError
+
+
+class BumpPackages(CMD):
+    """
+    Bump all the versions of all the packages and increment the version number
+    given.
+    """
+
+    arg_skip = Arg(
+        "-skip",
+        help="Do not increment versions in these packages",
+        nargs="+",
+        default=[],
+        required=False,
+    )
+    arg_only = Arg(
+        "-only",
+        help="Only increment versions in these packages",
+        nargs="+",
+        default=[],
+        required=False,
+    )
+    arg_version = Arg("version", help="Version to increment by")
+
+    @staticmethod
+    def bump_version(original, increment):
+        # Split on .
+        # map: int: Convert to an int
+        # zip: Create three instances of (original[i], increment[i])
+        # map: sum: Add each pair together
+        # map: str: Convert back to strings
+        return ".".join(
+            map(
+                str,
+                map(
+                    sum,
+                    zip(
+                        map(int, original.split(".")),
+                        map(int, increment.split(".")),
+                    ),
+                ),
+            )
+        )
+
+    async def run(self):
+        main_package = is_develop("dffml")
+        if not main_package:
+            raise NotImplementedError(
+                "Need to reinstall the main package in development mode."
+            )
+        main_package = pathlib.Path(main_package)
+        skel = main_package / "dffml" / "skel"
+        # Update all the version files
+        for version_file in main_package.rglob("**/version.py"):
+            # Ignore skel
+            if skel in version_file.parents:
+                self.logger.debug(
+                    "Skipping skel verison file %s", version_file
+                )
+                continue
+            # If we're only supposed to increment versions of some packages,
+            # check we're in the right package, skip if not.
+            setup_filepath = version_file.parent.parent / "setup.py"
+            with chdir(setup_filepath.parent):
+                name = SetupPyKWArg.get_kwargs(setup_filepath)["name"]
+                if self.only and name not in self.only:
+                    self.logger.debug(
+                        "Verison file not in only %s", version_file
+                    )
+                    continue
+                elif name in self.skip:
+                    self.logger.debug("Skipping verison file %s", version_file)
+                    continue
+            # Read the file
+            filetext = version_file.read_text()
+            # Find the version as a string
+            modified_lines = []
+            for line in filetext.split("\n"):
+                # Look for the line containing the version string
+                if line.startswith("VERSION"):
+                    # Parse the version string
+                    version = ast.literal_eval(line.split("=")[-1].strip())
+                    # Increment the version string
+                    version = self.bump_version(version, self.version)
+                    # Modify the line to use the new version string
+                    line = f'VERSION = "{version}"'
+                # Append line to list of lines to write back
+                modified_lines.append(line)
+            # Write back the file using the modified lines
+            filetext = version_file.write_text("\n".join(modified_lines))
+            self.logger.debug("Updated verison file %s", version_file)
+
+
+class Bump(CMD):
+    """
+    Bump the the main package in the versions plugins, or any or all libraries.
+    """
+
+    main = BumpMain
+    packages = BumpPackages
+
+
 class Develop(CMD):
     """
     Development utilities for hacking on DFFML itself
@@ -447,3 +570,4 @@ class Develop(CMD):
     install = Install
     release = Release
     setuppy = SetupPy
+    bump = Bump

docs/cli.rst

@@ -201,3 +201,27 @@ argument.
 
     $ dffml service dev setuppy kwarg name model/tensorflow/setup.py
     dffml-model-tensorflow
+
+bump
+++++
+
+Utilities for bumping version numbers.
+
+main
+____
+
+Update the version of DFFML used by all of the plugins.
+
+.. code-block:: console
+
+    dffml service dev bump main
+
+packages
+________
+
+Update the version number of a package or all packages. Increments the version
+of each packages by the version string given.
+
+.. code-block:: console
+
+    dffml service dev bump packages -log debug -skip dffml -- 0.0.1

tests/service/test_dev.py

@@ -13,7 +13,13 @@
 
 from dffml.version import VERSION
 from dffml.df.types import DataFlow
-from dffml.service.dev import Develop, RepoDirtyError, Export, Run
+from dffml.service.dev import (
+    Develop,
+    RepoDirtyError,
+    Export,
+    Run,
+    BumpPackages,
+)
 from dffml.util.os import chdir
 from dffml.util.skel import Skel
 from dffml.util.packaging import is_develop
@@ -229,6 +235,11 @@ async def test_okay(self):
         )
 
 
+class TestBumpPackages(AsyncTestCase):
+    async def test_bump_version(self):
+        self.assertEqual(BumpPackages.bump_version("1.2.3", "5.6.7"), "6.8.10")
+
+
 class TestExport(AsyncTestCase):
     async def test_run(self):
         stdout = io.BytesIO()