From: Lanqing Huang <lqhuang@outlook.com>
Date: Wed, 13 Apr 2022 08:06:01 +0000 (+0800)
Subject: Fix `AttributeError` by accessing `meta` field when return_type is dict
X-Git-Url: https://vcs.fsf.org/?a=commitdiff_plain;h=aedd145e2f0bca47d88f0f20e8926d84dda5ab0e;p=tweepy.git

Fix `AttributeError` by accessing `meta` field when return_type is dict
---

diff --git a/tweepy/pagination.py b/tweepy/pagination.py
index f61c192..1aea5a7 100644
--- a/tweepy/pagination.py
+++ b/tweepy/pagination.py
@@ -2,8 +2,13 @@
 # Copyright 2009-2022 Joshua Roesslein
 # See LICENSE for details.
 
+from typing import Union
 from math import inf
 
+import requests
+
+from tweepy.client import Response
+
 
 class Paginator:
     """:class:`Paginator` can be used to paginate for any :class:`Client`
@@ -77,7 +82,7 @@ class PaginationIterator:
     def __iter__(self):
         return self
 
-    def __next__(self):
+    def __next__(self) -> Union[Response, dict, requests.Response]:
         if self.reverse:
             pagination_token = self.previous_token
         else:
@@ -97,8 +102,17 @@ class PaginationIterator:
 
         response = self.method(*self.args, **self.kwargs)
 
-        self.previous_token = response.meta.get("previous_token")
-        self.next_token = response.meta.get("next_token")
+        if isinstance(response, Response):
+            meta = response.meta
+        elif isinstance(response, dict):
+            meta = response.get("meta", {})
+        elif isinstance(response, requests.Response):
+            meta = response.json().get("meta", {})
+        else:
+            raise NotImplementedError("Unknown `response` type to parse `meta` field")
+
+        self.previous_token = meta.get("previous_token")
+        self.next_token = meta.get("next_token")
         self.count += 1
 
         return response