Fix ENABLE_PUSH

compat-fork · Aug 21, 2024 · 51c197f · 51c197f
1 parent c984257
commit 51c197f
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/hyper/http20/connection.py b/hyper/http20/connection.py
@@ -32,6 +32,12 @@
 import threading
 import itertools
 
+try:
+    from h2.settings import ENABLE_PUSH
+except ImportError:
+    from h2.settings import SettingCodes
+    ENABLE_PUSH = SettingCodes.ENABLE_PUSH
+
 log = logging.getLogger(__name__)
 
 DEFAULT_WINDOW_SIZE = 65535
@@ -403,7 +409,7 @@ def _connect_upgrade(self, sock):
         with self._conn as conn:
             conn.initiate_upgrade_connection()
             conn.update_settings(
-                {h2.settings.ENABLE_PUSH: int(self._enable_push)}
+                {ENABLE_PUSH: int(self._enable_push)}
             )
         self._send_outstanding_data()
 
@@ -424,7 +430,7 @@ def _send_preamble(self):
         with self._conn as conn:
             conn.initiate_connection()
             conn.update_settings(
-                {h2.settings.ENABLE_PUSH: int(self._enable_push)}
+                {ENABLE_PUSH: int(self._enable_push)}
             )
         self._send_outstanding_data()