fix error when extrakeys lookup non-dicts

kynan · Feb 18, 2019 · 91f2f6d · 91f2f6d
1 parent abc61fb
commit 91f2f6d
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/nbstripout.py b/nbstripout.py
@@ -155,7 +155,12 @@ def pop_recursive(d, key, default=None):
     nested = key.split('.')
     current = d
     for k in nested[:-1]:
-        current = current.get(k, {})
+        if hasattr(current, 'get'):
+            current = current.get(k, {})
+        else:
+            return default
+    if not hasattr(current, 'pop'):
+        return default
     return current.pop(nested[-1], default)