refactor: Combine code for version version files

1 files changed, 15 insertions, 24 deletions

diff --git a/bumper b/bumper
index 9528bf5..6c949bd 100755
--- a/bumper
+++ b/bumper
@@ -39,20 +39,17 @@ class Bumper(cliapp.Application):
                 'New version {} is older than current version {}'.format(
                     version, current_version))
 
-        version_py = self.find_version_file('version.py')
-        if version_py:
-            print '... {}'.format(version_py)
-            self.write_version_py(version_py, version, '')
-
-        version_txt = self.find_version_file('version.txt')
-        if version_txt:
-            print '... {}'.format(version_txt)
-            self.write_version_txt(version_txt, version, '')
+        version_files = [
+            ('version.py', self.write_version_py),
+            ('version.txt', self.write_version_txt),
+            ('version.yaml', self.write_version_yaml),
+        ]
 
-        version_yaml = self.find_version_file('version.yaml')
-        if version_yaml:
-            print '... {}'.format(version_yaml)
-            self.write_version_yaml(version_yaml, version, '')
+        for basename, func in version_files:
+            filename = self.find_version_file(basename)
+            if filename:
+                print '.. {}'.format(filename)
+                func(filename, version, '')
 
         print '... debian/changelog'
         self.update_debian_changelog(version, '')
@@ -70,17 +67,11 @@ class Bumper(cliapp.Application):
         gitversion = version + '+git'
         print 'Updating in-development version to', gitversion
 
-        if version_py:
-            print '... {}'.format(version_py)
-            self.write_version_py(version_py, version, '+git')
-
-        if version_txt:
-            print '... {}'.format(version_txt)
-            self.write_version_txt(version_txt, version, '+git')
-
-        if version_yaml:
-            print '... {}'.format(version_yaml)
-            self.write_version_yaml(version_yaml, version, '+git')
+        for basename, func in version_files:
+            filename = self.find_version_file(basename)
+            if filename:
+                print '... {}'.format(filename)
+                func(filename, version, '+git')
 
         print '... debian/changelog'
         self.update_debian_changelog(version, 'New upstream version.')