fix: validator

Author: mic1on

src/usepy/validator/is_async_function.py

@@ -1,12 +1,22 @@
 import asyncio
-from typing import Callable
+from typing import Callable, Any
 
 
-def is_async_function(func: Callable) -> bool:
-    """
-    check if the function is an async function
-    >>> is_async_function(lambda x: x)
-    False
+def is_async_function(func: Callable[..., Any]) -> bool:
     """
+    Check if the function is an async function.
+
+    Args:
+        func (Callable[..., Any]): The function to check.
 
+    Returns:
+        bool: True if the function is async, False otherwise.
+
+    Examples:
+        >>> async def async_func(): pass
+        >>> is_async_function(async_func)
+        True
+        >>> is_async_function(lambda x: x)
+        False
+    """
     return asyncio.iscoroutinefunction(func)

src/usepy/validator/is_url.py

@@ -1,30 +1,42 @@
-def is_url(url: str) -> bool:
+from urllib.parse import urlparse
+from typing import List, Optional
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+def is_url(url: str, allowed_schemes: Optional[List[str]] = None) -> bool:
     """
-    check if the given string is a valid URL.
+    Check if the given string is a valid URL.
 
     Args:
-        url (str): the string to check.
+        url (str): The string to check.
+        allowed_schemes (Optional[List[str]]): List of allowed URL schemes.
+            Defaults to ["http", "https"] if None.
 
     Returns:
-        bool: if the string is a valid URL, return True, otherwise return False.
+        bool: If the string is a valid URL, return True, otherwise return False.
 
     Examples:
         >>> is_url("https://www.google.com")
         True
+        >>> is_url("ftp://example.com", allowed_schemes=["ftp"])
+        True
+        >>> is_url("mailto:user@example.com", allowed_schemes=["mailto"])
+        True
     """
     if not url or not isinstance(url, str):
         return False
 
-    from urllib.parse import urlparse
+    if allowed_schemes is None:
+        allowed_schemes = ["http", "https"]
 
     try:
         result = urlparse(url)
-        if all([result.scheme, result.netloc]) and result.scheme in [
-            "http",
-            "https",
-        ]:
+        if result.scheme and result.scheme in allowed_schemes:
             return True
     except ValueError:
+        logger.debug(f"Invalid URL format: {url}")
         return False
 
     return False

tests/test_validator.py

@@ -17,4 +17,5 @@ def test_is_url():
     assert is_url("http://localhost:8000")
     assert not is_url("not a url")
     assert not is_url("mailto:user@example.com")
+    assert is_url("mailto:user@example.com", allowed_schemes=["mailto"])
     assert not is_url("ftp://invalid-scheme.com")