⏱️ WaitManager-Web

Intelligent rate limiting for Flask, FastAPI, and Django — inspired by Fast Rub's WaitManager.

✨ Features

• 🔌 Plug & Play — Works with Flask, FastAPI, and Django
• 🧠 Smart Rate Limiting — Low/Medium/High traffic tiers with automatic wait calculation
• 📊 Per-Channel Tracking — Separate limits for API, login, upload, download, search, signup
• 🎯 Per-IP Tracking — Track each client separately
• 🔧 Fully Customizable — Custom channels, custom sleep callback, auto-tracking
• 🪶 Zero Dependencies — Core module uses only Python stdlib
• 📦 Optional Dependencies — Install only what you need

📦 Installation

pip install waitmanager-web               # Core only
pip install waitmanager-web[flask]        # + Flask
pip install waitmanager-web[fastapi]      # + FastAPI
pip install waitmanager-web[django]       # + Django
pip install waitmanager-web[all]          # Everything

🚀 Quick Start

Flask

from flask import Flask
from waitmanager.flask import FlaskWaitManager

app = Flask(__name__)
wm = FlaskWaitManager(low_traffic=100, high_wait=2.0, auto_track=True)
wm.init_app(app)

@app.route("/api/data")
@wm.limit(channel="api")
def get_data():
    return {"status": "ok"}

@app.route("/login", methods=["POST"])
@wm.limit(channel="login", max_requests=5, window=300)
def login():
    return {"status": "ok"}

FastAPI

from fastapi import FastAPI, Depends
from waitmanager import WaitManager
from waitmanager.fastapi import RateLimiter

app = FastAPI()
wm = WaitManager(low_traffic=100, high_wait=2.0)
limiter = RateLimiter(wm)

@app.get("/api/data")
async def get_data(_=Depends(limiter.limit("api"))):
    return {"status": "ok"}

@app.post("/login")
async def login(_=Depends(limiter.limit("login", max_requests=5, window=300))):
    return {"status": "ok"}

Django

# settings.py
MIDDLEWARE = [
    "waitmanager.django.WaitManagerMiddleware",
    ...
]

WAITMANAGER_CONFIG = {
    "low_traffic": 100,
    "high_wait": 2.0,
    "auto_track": True,
}

# views.py
from waitmanager.django import limit

@limit(channel="api")
def api_view(request):
    return JsonResponse({"status": "ok"})

@limit(channel="login", max_requests=5, window=300, block=True)
def login_view(request):
    return JsonResponse({"status": "ok"})

⚙️ Configuration

🎯 Custom Sleep Callback

def my_logic(channel, messages_count, last_time, **kwargs):
    if channel == "login" and messages_count > 3:
        return 30.0  # 30 second penalty
    return None  # Use default logic

wm = WaitManager(sleep_callback=my_logic)

🧪 Custom Channels

wm = WaitManager(channels=["api", "login", "payment", "export"])

📄 License

MIT © OandONE

🙏 Acknowledgments

Inspired by Fast Rub's WaitManager — the smartest rate limiter for Rubika bots.

"Good software is built on pain. WaitManager was built on Rubika's API." — OandONE