rpm: allow to filter out elements from the version dict

1 files changed, 18 insertions, 0 deletions

diff --git a/gbp/rpm/__init__.py b/gbp/rpm/__init__.py
index 5a9a6688..b0b8a2ef 100644
--- a/gbp/rpm/__init__.py
+++ b/gbp/rpm/__init__.py
@@ -927,6 +927,7 @@ def split_version_str(version):
 
     return ret
 
+
 def compose_version_str(evr):
     """
     Compose a full version string from individual "version components",
@@ -959,4 +960,21 @@ def compose_version_str(evr):
     return None
 
 
+def filter_version(evr, key):
+    """
+    Remove entry from the version dict
+
+    @param evr: dict of version components
+    @type evr: C{dict} of C{str}
+    @param key: key to remove
+    @type evr: C{str}
+    @return: new verion dict
+    @rtype: C{dict} of C{str}
+
+    >>> filter_version({'epoch': 'foo', 'upstreamversion': 'bar', 'vendor': 'baz'}, 'vendor').keys()
+    ['epoch', 'upstreamversion']
+    """
+    return {k: evr[k] for k in evr if k != key}
+
+
 # vim:et:ts=4:sw=4:et:sts=4:ai:set list listchars=tab\:»·,trail\:·: