added better retry system

Author: YonkoSam

.coverage

@@ -0,0 +1,218 @@
+#!/usr/bin/env python3
+"""
+Example usage of the Wasender API SDK
+
+This example demonstrates:
+1. Sending a text message
+2. Getting all contacts
+
+Set the following environment variables before running:
+- WASENDER_API_KEY: Your Wasender API key
+- WASENDER_ACCESS_TOKEN: Your personal access token (optional)
+
+Usage:
+    python example.py
+"""
+
+import os
+import asyncio
+from wasenderapi import create_sync_wasender, create_async_wasender
+from wasenderapi.models import RetryConfig
+from wasenderapi.errors import WasenderAPIError
+
+
+def get_api_credentials():
+    """Get API credentials from environment variables."""
+    api_key = os.getenv('WASENDER_API_KEY')
+    access_token = os.getenv('WASENDER_ACCESS_TOKEN')
+    
+    if not api_key:
+        raise ValueError(
+            "WASENDER_API_KEY environment variable is required. "
+            "Please set it to your Wasender API key."
+        )
+    
+    return api_key, access_token
+
+
+def sync_example():
+    """Example using the synchronous client."""
+    print("=== Synchronous Client Example ===")
+    
+    try:
+        # Get credentials from environment
+        api_key, access_token = get_api_credentials()
+        
+        # Create sync client with retry configuration
+        retry_config = RetryConfig(enabled=True, max_retries=3)
+        client = create_sync_wasender(
+            api_key=api_key,
+            personal_access_token=access_token,
+            retry_options=retry_config
+        )
+        
+        # Example 1: Send a text message
+        print("\n1. Sending text message...")
+        try:
+            # Replace with an actual phone number for testing
+            phone_number = "+1234567890"  # Update this with a real number
+            message_text = "Hello from Wasender API SDK! 🚀"
+            
+            result = client.send_text(
+                to=phone_number,
+                text_body=message_text
+            )
+            
+            print(f"✅ Message sent successfully!")
+            print(f"   Status: {result.response.message}")
+            
+            # Show rate limit info
+            if result.rate_limit:
+                print(f"   Rate limit: {result.rate_limit.remaining}/{result.rate_limit.limit} remaining")
+                
+        except WasenderAPIError as e:
+            print(f"❌ Failed to send message: {e.message}")
+            if e.status_code == 429:
+                print(f"   Rate limited. Retry after: {e.retry_after} seconds")
+            elif e.error_details:
+                print(f"   Error details: {e.error_details}")
+          # Example 2: Get all contacts
+        print("\n2. Getting contacts...")
+        try:
+            contacts_result = client.get_contacts()
+            
+            print(f"✅ Retrieved contacts successfully!")
+            print(f"   Total contacts: {len(contacts_result.response.data)}")
+            
+            # Show first few contacts
+            for i, contact in enumerate(contacts_result.response.data[:3]):
+                contact_name = contact.name or "Unknown"
+                contact_id = contact.id or "No ID"
+                print(f"   Contact {i+1}: {contact_name} ({contact_id})")
+            
+            if len(contacts_result.response.data) > 3:
+                print(f"   ... and {len(contacts_result.response.data) - 3} more")
+                
+        except WasenderAPIError as e:
+            print(f"❌ Failed to get contacts: {e.message}")
+            if e.status_code == 401:
+                print("   Check your API key and access token")
+            elif e.error_details:
+                print(f"   Error details: {e.error_details}")
+        
+    except ValueError as e:
+        print(f"❌ Configuration error: {e}")
+    except Exception as e:
+        print(f"❌ Unexpected error: {e}")
+
+
+async def async_example():
+    """Example using the asynchronous client."""
+    print("\n=== Asynchronous Client Example ===")
+    
+    try:
+        # Get credentials from environment
+        api_key, access_token = get_api_credentials()
+        
+        # Create async client with retry configuration
+        retry_config = RetryConfig(enabled=True, max_retries=3)
+        
+        async with create_async_wasender(
+            api_key=api_key,
+            personal_access_token=access_token,
+            retry_options=retry_config
+        ) as client:
+            
+            # Example 1: Send a text message
+            print("\n1. Sending text message (async)...")
+            try:
+                # Replace with an actual phone number for testing
+                phone_number = "+1234567890"  # Update this with a real number
+                message_text = "Hello from Wasender API SDK (async)! 🚀"
+                
+                result = await client.send_text(
+                    to=phone_number,
+                    text_body=message_text
+                )
+                
+                print(f"✅ Message sent successfully!")
+                print(f"   Status: {result.response.message}")
+                
+                # Show rate limit info
+                if result.rate_limit:
+                    print(f"   Rate limit: {result.rate_limit.remaining}/{result.rate_limit.limit} remaining")
+                    
+            except WasenderAPIError as e:
+                print(f"❌ Failed to send message: {e.message}")
+                if e.status_code == 429:
+                    print(f"   Rate limited. Retry after: {e.retry_after} seconds")
+                elif e.error_details:
+                    print(f"   Error details: {e.error_details}")
+              # Example 2: Get all contacts
+            print("\n2. Getting contacts (async)...")
+            try:
+                contacts_result = await client.get_contacts()
+                
+                print(f"✅ Retrieved contacts successfully!")
+                print(f"   Total contacts: {len(contacts_result.response.data)}")
+                
+                # Show first few contacts
+                for i, contact in enumerate(contacts_result.response.data[:3]):
+                    contact_name = contact.name or "Unknown"
+                    contact_id = contact.id or "No ID"
+                    print(f"   Contact {i+1}: {contact_name} ({contact_id})")
+                
+                if len(contacts_result.response.data) > 3:
+                    print(f"   ... and {len(contacts_result.response.data) - 3} more")
+                    
+            except WasenderAPIError as e:
+                print(f"❌ Failed to get contacts: {e.message}")
+                if e.status_code == 401:
+                    print("   Check your API key and access token")
+                elif e.error_details:
+                    print(f"   Error details: {e.error_details}")
+    
+    except ValueError as e:
+        print(f"❌ Configuration error: {e}")
+    except Exception as e:
+        print(f"❌ Unexpected error: {e}")
+
+
+def main():
+    """Main function to run both sync and async examples."""
+    print("Wasender API SDK Example")
+    print("=" * 40)
+    
+    # Check if environment variables are set
+    api_key = os.getenv('WASENDER_API_KEY')
+    if not api_key:
+        print("❌ Environment variable WASENDER_API_KEY is not set!")
+        print("\nTo run this example, please set the required environment variables:")
+        print("  WASENDER_API_KEY=your_api_key_here")
+        print("  WASENDER_ACCESS_TOKEN=your_access_token_here  # Optional")
+        print("\nExample:")
+        print("  export WASENDER_API_KEY=wsk_...")
+        print("  export WASENDER_ACCESS_TOKEN=pat_...")
+        print("  python example.py")
+        return
+    
+    print(f"✅ API Key found: {api_key[:10]}...")
+    
+    access_token = os.getenv('WASENDER_ACCESS_TOKEN')
+    if access_token:
+        print(f"✅ Access Token found: {access_token[:10]}...")
+    else:
+        print("ℹ️  Access Token not provided (optional)")
+    
+    # Run synchronous example
+    sync_example()
+    
+    # Run asynchronous example
+    asyncio.run(async_example())
+    
+    print("\n" + "=" * 40)
+    print("Examples completed!")
+
+
+if __name__ == "__main__":
+    main()

example.py

@@ -30,6 +30,7 @@ keywords = ["whatsapp", "api", "sdk", "wasender", "messaging", "chatbot"]
 dependencies = [
     "httpx>=0.23.0,<0.28.0",
     "pydantic>=2.0,<3.0",
+    "requests>=2.25.0,<3.0.0",
 ]
 
 [project.urls]

pyproject.toml

@@ -1,10 +1,13 @@
 import pytest
-from unittest.mock import AsyncMock, patch
-from wasenderapi import create_async_wasender, __version__ as SDK_VERSION
+from unittest.mock import AsyncMock, patch, Mock, MagicMock
+import time
+import asyncio
+from wasenderapi import create_async_wasender, create_sync_wasender, __version__ as SDK_VERSION
 from wasenderapi.models import RetryConfig, WasenderSendResult
 from wasenderapi.errors import WasenderAPIError
 import json
 import httpx
+import requests
 
 @pytest.fixture
 async def async_client_with_mocked_post():
@@ -19,6 +22,19 @@ async def async_client_with_mocked_post():
     client._post_internal.return_value = mock_post_return_value
     return client
 
+@pytest.fixture
+def sync_client_with_mocked_post():
+    retry_config_disabled = RetryConfig(enabled=False)
+    client = create_sync_wasender("test_api_key", retry_options=retry_config_disabled)
+    client._post_internal = Mock(name="_post_internal")
+    
+    mock_post_return_value = {
+        "response": {"success": True, "message": "ok", "data": {"messageId": "mock_message_id"}},
+        "rate_limit": {"limit": 1000, "remaining": 999, "reset_timestamp": 1620000000}
+    }
+    client._post_internal.return_value = mock_post_return_value
+    return client
+
 @pytest.fixture
 def success_api_response_data():
     return {"success": True, "message": "Message sent successfully", "data": {"messageId": "test-message-id"}}
@@ -39,6 +55,18 @@ def error_api_response_data():
         "errors": [{"field": "to", "message": "The 'to' field is invalid."}]
     }
 
+@pytest.fixture
+def sync_client_with_retry_enabled():
+    retry_config = RetryConfig(enabled=True, max_retries=3)
+    client = create_sync_wasender("test_api_key", retry_options=retry_config)
+    return client
+
+@pytest.fixture
+async def async_client_with_retry_enabled():
+    retry_config = RetryConfig(enabled=True, max_retries=3)
+    client = create_async_wasender("test_api_key", retry_options=retry_config)
+    return client
+
 @pytest.mark.asyncio
 async def test_send_text_constructs_correct_payload_and_calls_post_internal(async_client_with_mocked_post):
     client = async_client_with_mocked_post
@@ -150,6 +178,7 @@ async def test_send_document(async_client_with_mocked_post, success_api_response
             "to": test_to,
             "messageType": "document",
             "documentUrl": test_url,
+            "fileName": test_filename,
             "text": test_caption
         }
     )
@@ -244,6 +273,43 @@ async def test_send_location(async_client_with_mocked_post, success_api_response
     assert response.response.success == True
     assert response.response.message == success_api_response_data["message"]
 
+@pytest.mark.asyncio
+async def test_send_poll(async_client_with_mocked_post, success_api_response_data, rate_limit_data):
+        client = async_client_with_mocked_post
+        client._post_internal.return_value = {
+            "response": success_api_response_data,
+            "rate_limit": rate_limit_data
+        }
+        
+        test_to = "1234567890"
+        test_question = "What's your favorite color?"
+        test_options = ["Red", "Blue", "Green", "Yellow"]
+        test_multiple_answers = True
+
+        response = await client.send_poll(
+            to=test_to, 
+            question=test_question, 
+            options=test_options, 
+            is_multiple_choice=test_multiple_answers
+        )
+            
+        client._post_internal.assert_called_once_with(
+            "/send-message",
+            {
+            "to": test_to,
+            "messageType": "poll",
+            "poll": {
+                "question": test_question,
+                "options": test_options,
+                "multiSelect": test_multiple_answers
+            }
+            }
+        )
+        assert isinstance(response, WasenderSendResult)
+        assert response.response.success == True
+        assert response.response.message == success_api_response_data["message"]
+        assert response.rate_limit.limit == rate_limit_data["limit"]
+
 @pytest.mark.asyncio
 async def test_api_error_raised_from_post_internal(async_client_with_mocked_post, error_api_response_data):
     client = async_client_with_mocked_post
@@ -263,4 +329,10 @@ async def test_api_error_raised_from_post_internal(async_client_with_mocked_post
 
     assert exc_info.value.status_code == 400
     assert exc_info.value.api_message == original_api_message
-    assert exc_info.value.error_details == original_error_details 
+    assert exc_info.value.error_details == original_error_details
+
+
+
+
+
+