✨ Feature: support partial funcs in pagination (#206)

yanyongyu · web-flow · commit 26f3335b252c · 2025-04-13T12:57:01.000+08:00
diff --git a/githubkit/utils.py b/githubkit/utils.py
@@ -1,4 +1,5 @@
 from enum import Enum
+from functools import partial
 import inspect
 from typing import Any, Generic, Literal, Optional, TypeVar, final
 
@@ -61,7 +62,18 @@ def exclude_unset(data: Any) -> Any:
     return data
 
 
+def _unwrap_partial(func: T) -> T:
+    while isinstance(func, partial):
+        func = func.func
+    return func
+
+
 def is_async(obj: Any) -> bool:
+    """Check if an object is a async callable (coroutine function)."""
+
+    # unwrap partials first
+    obj = _unwrap_partial(obj)
+
     if inspect.isroutine(obj):
         return inspect.iscoroutinefunction(obj)
     if inspect.isclass(obj):
diff --git a/tests/test_rest/test_call.py b/tests/test_rest/test_call.py
@@ -1,3 +1,5 @@
+from functools import partial
+
 import pytest
 
 from githubkit import GitHub
@@ -60,10 +62,28 @@ def test_paginate(g: GitHub):
         ...
 
 
+def test_paginate_with_partial(g: GitHub):
+    paginator = g.rest.paginate(
+        partial(g.rest.issues.list_for_repo, OWNER, REPO), per_page=50
+    )
+    for _ in paginator:
+        ...
+
+
 @pytest.mark.anyio
 async def test_async_paginate(g: GitHub):
     paginator = g.rest.paginate(
         g.rest.issues.async_list_for_repo, owner=OWNER, repo=REPO, per_page=50
     )
     async for _ in paginator:
         ...
+
+
+@pytest.mark.anyio
+async def test_async_paginate_with_partial(g: GitHub):
+    paginator = g.rest.paginate(
+        partial(g.rest.issues.async_list_for_repo, OWNER, REPO),
+        per_page=50,
+    )
+    async for _ in paginator:
+        ...
