Преглед на файлове

use ensure_mapped context manager in dump

Christian Mönch преди 2 години
родител
ревизия
8c17ad3274
променени са 1 файла, в които са добавени 2 реда и са изтрити 1 реда
  1. 2 1
      dataladmetadatamodel/mappableobject.py

+ 2 - 1
dataladmetadatamodel/mappableobject.py

@@ -17,7 +17,8 @@ from dataladmetadatamodel.mapper.reference import Reference
 def ensure_mapped(mappable_object):
     needs_purge = False
     try:
-        needs_purge = mappable_object.ensure_mapped()
+        if mappable_object is not None:
+            needs_purge = mappable_object.ensure_mapped()
         yield mappable_object
     finally:
         if needs_purge: