Publish workflow

Author: xsyncio

tmdbfusion/features/pagination.py

@@ -101,16 +101,26 @@ def _should_skip(self) -> bool:
             return True
         return False
 
+    def _ensure_buffer(self) -> bool:
+        """Ensure buffer has items.
+
+        Returns
+        -------
+        bool
+            True if buffer has items, False otherwise.
+
+        """
+        if not self._buffer:
+            self._fill_buffer()
+        return bool(self._buffer)
+
     def __next__(self) -> T:
         """Get next item."""
         if self._check_take_limit():
             raise StopIteration
 
         while True:
-            if not self._buffer:
-                self._fill_buffer()
-
-            if not self._buffer:
+            if not self._ensure_buffer():
                 raise StopIteration
 
             item = self._buffer.popleft()
@@ -237,6 +247,19 @@ def _check_take_limit(self) -> bool:
         """
         return bool(self._take_limit is not None and self._items_yielded >= self._take_limit)
 
+    async def _ensure_buffer(self) -> bool:
+        """Ensure buffer has items.
+
+        Returns
+        -------
+        bool
+            True if buffer has items, False otherwise.
+
+        """
+        if not self._buffer:
+            await self._fill_buffer()
+        return bool(self._buffer)
+
     def _update_totals(self, response: typing.Any) -> None:  # noqa: ANN401
         """Update total pages and results from response."""
         if self._total_pages is None:
@@ -284,10 +307,7 @@ async def __anext__(self) -> T:
             raise StopAsyncIteration
 
         while True:
-            if not self._buffer:
-                await self._fill_buffer()
-
-            if not self._buffer:
+            if not await self._ensure_buffer():
                 raise StopAsyncIteration
 
             item = self._buffer.popleft()