Support vendor tag for rpm generation

1 files changed, 4 insertions, 0 deletions

diff --git a/ybd/rpm.py b/ybd/rpm.py
index ed855b5..0278ff5 100644
--- a/ybd/rpm.py
+++ b/ybd/rpm.py
@@ -147,6 +147,10 @@ def generate_spec(dn, stage_dir, metafile, output, name, system):
         # Add all the main package Provides:
         for package in package_list:
             if is_main_package(system, dn, package['name']):
+                vendor = next((d['vendor'] for d in (package, rpm_metadata)
+                               if 'vendor' in d), None)
+                if vendor:
+                    output_f.write('Vendor: %s\n' % vendor)
                 version = next((d['version'] for d in (package, rpm_metadata)
                                 if 'version' in d), None)
                 if version: