feat(bump): add flag --local-version that supports bumping only the local version instead of the public

marcossantiago · marcossantiago · commit 3b8b82922cce · 2020-11-12T19:14:56.000+01:00
diff --git a/commitizen/bump.py b/commitizen/bump.py
@@ -108,7 +108,10 @@ def semver_generator(current_version: str, increment: str = None) -> str:
 
 
 def generate_version(
-    current_version: str, increment: str, prerelease: Optional[str] = None
+    current_version: str,
+    increment: str,
+    prerelease: Optional[str] = None,
+    is_local_version: bool = False,
 ) -> Version:
     """Based on the given increment a proper semver will be generated.
 
@@ -122,10 +125,18 @@ def generate_version(
         MAJOR 1.0.0 -> 2.0.0
     """
     pre_version = prerelease_generator(current_version, prerelease=prerelease)
-    semver = semver_generator(current_version, increment=increment)
-    # TODO: post version
-    # TODO: dev version
-    return Version(f"{semver}{pre_version}")
+
+    if is_local_version:
+        version = Version(current_version)
+        local_semver = semver_generator(str(version.local), increment=increment)
+
+        return Version(f"{version.public}+{local_semver}{pre_version}")
+    else:
+        semver = semver_generator(current_version, increment=increment)
+
+        # TODO: post version
+        # TODO: dev version
+        return Version(f"{semver}{pre_version}")
 
 
 def update_version_in_files(
diff --git a/commitizen/cli.py b/commitizen/cli.py
@@ -81,6 +81,11 @@
                         "action": "store_true",
                         "help": "bump version in the files from the config",
                     },
+                    {
+                        "name": "--local-version",
+                        "action": "store_true",
+                        "help": "bump only the local version portion",
+                    },
                     {
                         "name": ["--changelog", "-ch"],
                         "action": "store_true",
diff --git a/commitizen/commands/bump.py b/commitizen/commands/bump.py
@@ -92,6 +92,7 @@ def __call__(self):  # noqa: C901
         increment: Optional[str] = self.arguments["increment"]
         prerelease: str = self.arguments["prerelease"]
         is_files_only: Optional[bool] = self.arguments["files_only"]
+        is_local_version: Optional[bool] = self.arguments["local_version"]
 
         current_tag_version: str = bump.create_tag(
             current_version, tag_format=tag_format
@@ -117,7 +118,10 @@ def __call__(self):  # noqa: C901
             increment = None
 
         new_version = bump.generate_version(
-            current_version, increment, prerelease=prerelease
+            current_version,
+            increment,
+            prerelease=prerelease,
+            is_local_version=is_local_version,
         )
         new_tag_version = bump.create_tag(new_version, tag_format=tag_format)
         message = bump.create_commit_message(
@@ -140,7 +144,7 @@ def __call__(self):  # noqa: C901
 
         bump.update_version_in_files(
             current_version,
-            new_version.public,
+            str(new_version),
             version_files,
             check_consistency=self.check_consistency,
         )
@@ -159,7 +163,7 @@ def __call__(self):  # noqa: C901
             changelog_cmd()
             c = cmd.run(f"git add {changelog_cmd.file_name}")
 
-        self.config.set_key("version", new_version.public)
+        self.config.set_key("version", str(new_version))
         c = git.commit(message, args=self._get_commit_args())
         if c.return_code != 0:
             raise BumpCommitFailedError(f'git.commit error: "{c.err.strip()}"')
