Merge pull request #1253 from dralley/fix-migration

Fix migration for cases in which no vendor field is provided
pulp · Jan 10, 2019 · 5772781 · 5772781
2 parents 223df1f + 3f54007
commit 5772781
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/plugins/pulp_rpm/plugins/migrations/0045_populate_vendor.py b/plugins/pulp_rpm/plugins/migrations/0045_populate_vendor.py
@@ -37,10 +37,12 @@ def migrate_rpm(collection, unit):
     primary_xml = _HEADER.format(gzip.zlib.decompress(primary_xml))
     root_element = ET.fromstring(primary_xml)
     delta = {}
-    delta['vendor'] = root_element.find(
+    vendor_element = root_element.find(
         './common:package/common:format/rpm:vendor', _NAMESPACES
-    ).text
-    collection.update_one({'_id': unit['_id']}, {'$set': delta})
+    )
+    if vendor_element:
+        delta['vendor'] = vendor_element.text
+        collection.update_one({'_id': unit['_id']}, {'$set': delta})
 
 
 def migrate(*args, **kwargs):