1. add check_recursion to morebuiltins.funcs, to check if a function is recursive.

Author: ClericPy

CHANGELOG.md

@@ -1,3 +1,6 @@
+### 1.3.1 (2025-06-19)
+1. add `check_recursion` to `morebuiltins.funcs`, to check if a function is recursive.
+
 ### 1.3.0 (2025-03-08)
 1. **Compatibility Warning**. rename `morebuiltins.functools` to `morebuiltins.funcs` to avoid conflict with `functools` in python standard library.
 2. add `key_type` arg to `sqlite.KV`, to support `int` key.

morebuiltins/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "1.3.0"
+__version__ = "1.3.1"
 __all__ = [
     "morebuiltins.utils",
     "morebuiltins.date",

morebuiltins/cmd/log_server.py

@@ -1,7 +1,6 @@
 import asyncio
 import json
 import logging
-import logging.handlers
 import os
 import signal
 import sys

morebuiltins/funcs.py

@@ -1,3 +1,4 @@
+import ast
 import asyncio
 import importlib
 import importlib.util
@@ -44,6 +45,7 @@
     "RotatingFileWriter",
     "get_function",
     "to_thread",
+    "check_recursion",
 ]
 
 
@@ -1204,6 +1206,43 @@ async def __aexit__(self, exc_type, exc_val, exc_tb):
 async_logger = AsyncQueueListener
 
 
+def check_recursion(function: Callable, return_error=False):
+    """Check if a function is recursive by inspecting its AST.
+    Returns True if the function calls itself, otherwise False.
+
+    Demo::
+        >>> def recursive_func():
+        ...     return recursive_func()
+        >>> check_recursion(recursive_func)
+        True
+        >>> def non_recursive_func():
+        ...     return 1 + 1
+        >>> check_recursion(non_recursive_func)
+        False
+        >>> # print is a std-lib function
+        >>> check_recursion(print, return_error=False)
+        >>> type(check_recursion(print, return_error=True))
+        <class 'TypeError'>
+    """
+    try:
+        source = inspect.getsource(function)
+        tree = ast.parse(source)
+        func_name = function.__name__
+        for node in ast.walk(tree):
+            if (
+                isinstance(node, ast.Call)
+                and isinstance(node.func, ast.Name)
+                and node.func.id == func_name
+            ):
+                return True
+        return False
+    except Exception as e:
+        if return_error:
+            return e
+        else:
+            return None
+
+
 def test_bg_task():
     async def _test_bg_task():
         async def coro():