Avoid hasattr in BrotliDecoder.decompress()

urllib3 · Apr 19, 2020 · 8fadde1 · 8fadde1
1 parent 1f0825b
commit 8fadde1
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/urllib3/response.py b/src/urllib3/response.py
@@ -107,11 +107,10 @@ class BrotliDecoder(object):
         # are for 'brotlipy' and bottom branches for 'Brotli'
         def __init__(self):
             self._obj = brotli.Decompressor()
-
-        def decompress(self, data):
             if hasattr(self._obj, "decompress"):
-                return self._obj.decompress(data)
-            return self._obj.process(data)
+                self.decompress = self._obj.decompress
+            else:
+                self.decompress = self._obj.process
 
         def flush(self):
             if hasattr(self._obj, "flush"):