From 7129bd112bd0ec5f2d80d1f3ecc673d24b4eb960 Mon Sep 17 00:00:00 2001
From: fovc9 <66397013+fovc9@users.noreply.github.com>
Date: Wed, 17 Jan 2024 18:54:36 -0600
Subject: [PATCH] psycopg2: Add ReplicationCursor.consume_stream (#11283)

---
 stubs/psycopg2/psycopg2/extras.pyi | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/stubs/psycopg2/psycopg2/extras.pyi b/stubs/psycopg2/psycopg2/extras.pyi
index f732bc5ad..b3f953cdb 100644
--- a/stubs/psycopg2/psycopg2/extras.pyi
+++ b/stubs/psycopg2/psycopg2/extras.pyi
@@ -196,6 +196,9 @@ class ReplicationCursor(_replicationCursor):
         status_interval: int = 10,
     ) -> None: ...
     def fileno(self): ...
+    def consume_stream(
+        self, consume: Callable[[ReplicationMessage], object], keepalive_interval: float | None = None
+    ) -> None: ...
 
 class UUID_adapter:
     def __init__(self, uuid) -> None: ...