Mass commit, i should be cleaner in the future.

1 files changed, 33 insertions, 0 deletions

diff --git a/base/rpm_utils.py b/base/rpm_utils.py
new file mode 100644
index 0000000..bfdae46
--- /dev/null
+++ b/base/rpm_utils.py
@@ -0,0 +1,33 @@
+from __future__ import with_statement
+
+import re
+import rpm
+
+from contextlib import contextmanager
+from base import log
+
+class RPMSpec(object):
+    def __init__(self, spec_file):
+        self.contents = [x for x in file(spec_file)]
+    
+    def version(self):
+        ver_line = [x for x in self.contents if x.startswith('Version:')][0]
+        ver_re = re.search(r'Version: (.*)', ver_line)
+        return ver_re.groups()[0]
+    
+    def rel(self):
+        rel_line = [x for x in self.contents if x.startswith('Release:')][0]
+        rel_re = re.search(r'Release: (.*)', rel_line)
+        return rel_re.groups()[0] 
+
+@contextmanager
+def rpm_macros(**keys):
+    for key, value in keys.iteritems():
+        log.debug('setting...')
+        log.debug(key + ' ' + value)
+        rpm.addMacro(key, value)
+    yield
+    for key, value in keys.iteritems():
+        rpm.delMacro(key)
+
+__all__ = ['rpm_macros', 'RPMSpec']
\ No newline at end of file