support custom names for pre and build

ofek · ofek · commit 2e0dcadd4d68 · 2017-08-24T22:11:11.000-04:00
diff --git a/semver.py b/semver.py
@@ -328,27 +328,27 @@ def bump_patch(version):
                           verinfo['patch'] + 1)
 
 
-def bump_prerelease(version):
+def bump_prerelease(version, token='rc'):
     """Raise the prerelease part of the version
 
     :param: version string
     :return: the raised version string
     :rtype: str
     """
     verinfo = parse(version)
-    verinfo['prerelease'] = _increment_string(verinfo['prerelease'] or 'rc.0')
+    verinfo['prerelease'] = _increment_string(verinfo['prerelease'] or token + '.0')
     return format_version(verinfo['major'], verinfo['minor'], verinfo['patch'],
                           verinfo['prerelease'])
 
 
-def bump_build(version):
+def bump_build(version, token='build'):
     """Raise the build part of the version
 
     :param: version string
     :return: the raised version string
     :rtype: str
     """
     verinfo = parse(version)
-    verinfo['build'] = _increment_string(verinfo['build'] or 'build.0')
+    verinfo['build'] = _increment_string(verinfo['build'] or token + '.0')
     return format_version(verinfo['major'], verinfo['minor'], verinfo['patch'],
                           verinfo['prerelease'], verinfo['build'])
diff --git a/tests.py b/tests.py
@@ -260,6 +260,7 @@ def test_prerelease_order():
 def test_should_bump_prerelease():
     assert bump_prerelease('3.4.5-rc.9') == '3.4.5-rc.10'
     assert bump_prerelease('3.4.5') == '3.4.5-rc.1'
+    assert bump_prerelease('3.4.5', 'dev') == '3.4.5-dev.1'
 
 
 def test_should_ignore_build_on_prerelease_bump():
@@ -271,6 +272,7 @@ def test_should_bump_build():
     assert bump_build('3.4.5-rc.1+0009.dev') == '3.4.5-rc.1+0010.dev'
     assert bump_build('3.4.5-rc.1') == '3.4.5-rc.1+build.1'
     assert bump_build('3.4.5') == '3.4.5+build.1'
+    assert bump_build('3.4.5', 'nightly') == '3.4.5+nightly.1'
 
 
 def test_should_compare_version_info_objects():
