From: Harmon <Harmon758@gmail.com>
Date: Tue, 4 Oct 2022 19:11:39 +0000 (-0500)
Subject: Add oauth_1 parameter to AsyncBaseStream._connect for AsyncStream
X-Git-Url: https://vcs.fsf.org/?a=commitdiff_plain;h=29bbb7be378fef8347858c9825f0b48e2c380c90;p=tweepy.git

Add oauth_1 parameter to AsyncBaseStream._connect for AsyncStream

This allows the Authorization header to be regenerated in AsyncBaseStream._connect prior to any reconnection requests, when used through AsyncStream._connect
---

diff --git a/tweepy/asynchronous/streaming.py b/tweepy/asynchronous/streaming.py
index 3b3f28d..530179c 100644
--- a/tweepy/asynchronous/streaming.py
+++ b/tweepy/asynchronous/streaming.py
@@ -38,7 +38,7 @@ class AsyncBaseStream:
         )
 
     async def _connect(
-        self, method, url, params=None, headers=None, body=None
+        self, method, url, params=None, headers=None, body=None, oauth_1=False
     ):
         error_count = 0
         # https://developer.twitter.com/en/docs/twitter-api/v1/tweets/filter-realtime/guides/connecting
@@ -60,6 +60,14 @@ class AsyncBaseStream:
         try:
             while error_count <= self.max_retries:
                 try:
+                    if oauth_1:
+                        oauth_client = OAuthClient(
+                            self.consumer_key, self.consumer_secret,
+                            self.access_token, self.access_token_secret
+                        )
+                        url, headers, body = oauth_client.sign(
+                            url, http_method=method, headers=headers, body=body
+                        )
                     async with self.session.request(
                         method, url, params=params, headers=headers, data=body,
                         proxy=self.proxy
@@ -248,14 +256,11 @@ class AsyncStream(AsyncBaseStream):
     async def _connect(
         self, method, endpoint, params={}, headers=None, body=None
     ):
-        oauth_client = OAuthClient(self.consumer_key, self.consumer_secret,
-                                   self.access_token, self.access_token_secret)
         url = f"https://stream.twitter.com/1.1/{endpoint}.json"
         url = str(URL(url).with_query(sorted(params.items())))
-        url, headers, body = oauth_client.sign(
-            url, http_method=method, headers=headers, body=body
+        await super()._connect(
+            method, url, headers=headers, body=body, oauth_1=True
         )
-        await super()._connect(method, url, headers=headers, body=body)
 
     def filter(self, *, follow=None, track=None, locations=None,
                filter_level=None, languages=None, stall_warnings=False):