fixup! Don't leak compressed stream (piskvorky#630)

ampanasiuk · Aug 4, 2021 · 46ea033 · 46ea033
1 parent d70a7dc
commit 46ea033
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/smart_open/compression.py b/smart_open/compression.py
@@ -72,10 +72,12 @@ def adapted_closer(adapter, adapted):
 
     def close_both(*args):
         nonlocal adapted
-        orig_close()
-        if adapted:
-            adapted.close()
-            adapted = None
+        try:
+            orig_close()
+        finally:
+            if adapted:
+                adapted, fp = None, adapted
+                fp.close()
 
     adapter.close = close_both
     return adapter