forked from retoor/devplacepy
Compare commits
1
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
8dde41b483 |
File diff suppressed because one or more lines are too long
@@ -128,9 +128,6 @@ RATE_LIMIT = int(os.environ.get("DEVPLACE_RATE_LIMIT", "60"))
|
|||||||
RATE_WINDOW = 60
|
RATE_WINDOW = 60
|
||||||
WEB_WORKERS = max(1, int(os.environ.get("DEVPLACE_WEB_WORKERS", "1")))
|
WEB_WORKERS = max(1, int(os.environ.get("DEVPLACE_WEB_WORKERS", "1")))
|
||||||
RATE_LIMIT_DISABLED = os.environ.get("DEVPLACE_DISABLE_RATE_LIMIT") == "1"
|
RATE_LIMIT_DISABLED = os.environ.get("DEVPLACE_DISABLE_RATE_LIMIT") == "1"
|
||||||
LOGIN_EMAIL_RATE_LIMIT = int(os.environ.get("DEVPLACE_LOGIN_EMAIL_RATE_LIMIT", "10"))
|
|
||||||
|
|
||||||
_email_rate_limit_store: dict[str, list[float]] = defaultdict(list)
|
|
||||||
|
|
||||||
HOT_SETTINGS_TTL = 2.0
|
HOT_SETTINGS_TTL = 2.0
|
||||||
_hot_settings_value: dict = {}
|
_hot_settings_value: dict = {}
|
||||||
@@ -170,20 +167,6 @@ def _worker_rate_limit(limit: int) -> int:
|
|||||||
return max(1, -(-limit // WEB_WORKERS))
|
return max(1, -(-limit // WEB_WORKERS))
|
||||||
|
|
||||||
|
|
||||||
def check_email_rate_limit(email: str) -> bool:
|
|
||||||
now = time.time()
|
|
||||||
window_start = now - RATE_WINDOW
|
|
||||||
limit = LOGIN_EMAIL_RATE_LIMIT
|
|
||||||
timestamps = [
|
|
||||||
t for t in _email_rate_limit_store.get(email, []) if t > window_start
|
|
||||||
]
|
|
||||||
if len(timestamps) >= limit:
|
|
||||||
return False
|
|
||||||
timestamps.append(now)
|
|
||||||
_email_rate_limit_store[email] = timestamps
|
|
||||||
return True
|
|
||||||
|
|
||||||
|
|
||||||
_service_lock_handle = None
|
_service_lock_handle = None
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -20,7 +20,6 @@ from devplacepy.responses import respond, action_result, wants_json, json_error
|
|||||||
from devplacepy.schemas import AuthPageOut
|
from devplacepy.schemas import AuthPageOut
|
||||||
from devplacepy.services.audit import record as audit
|
from devplacepy.services.audit import record as audit
|
||||||
from devplacepy.dependencies import json_or_form
|
from devplacepy.dependencies import json_or_form
|
||||||
from devplacepy.main import check_email_rate_limit
|
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
router = APIRouter()
|
router = APIRouter()
|
||||||
@@ -73,22 +72,6 @@ async def login(request: Request, data: Annotated[LoginForm, Depends(json_or_for
|
|||||||
metadata={"email": email},
|
metadata={"email": email},
|
||||||
summary=f"failed login attempt for {email}",
|
summary=f"failed login attempt for {email}",
|
||||||
)
|
)
|
||||||
if not check_email_rate_limit(email):
|
|
||||||
audit.record(
|
|
||||||
request,
|
|
||||||
"security.rate_limit.email_block",
|
|
||||||
user=None,
|
|
||||||
actor_kind="guest",
|
|
||||||
result="denied",
|
|
||||||
metadata={"email": email},
|
|
||||||
summary=f"email rate limit reached for {email}",
|
|
||||||
)
|
|
||||||
if wants_json(request):
|
|
||||||
return json_error(429, "Too many login attempts for this account")
|
|
||||||
return HTMLResponse(
|
|
||||||
"Too many login attempts for this account",
|
|
||||||
status_code=429,
|
|
||||||
)
|
|
||||||
if wants_json(request):
|
if wants_json(request):
|
||||||
return json_error(401, "; ".join(errors), errors=errors)
|
return json_error(401, "; ".join(errors), errors=errors)
|
||||||
seo_ctx = base_seo_context(request, title="Sign In", robots="noindex,nofollow")
|
seo_ctx = base_seo_context(request, title="Sign In", robots="noindex,nofollow")
|
||||||
|
|||||||
@@ -18,7 +18,9 @@ from devplacepy.services.openai_gateway.usage import (
|
|||||||
usage_metric_cards,
|
usage_metric_cards,
|
||||||
)
|
)
|
||||||
from devplacepy.utils import generate_uid, make_combined_slug
|
from devplacepy.utils import generate_uid, make_combined_slug
|
||||||
|
from devplacepy.utils.notifications import create_notification
|
||||||
from devplacepy.services.audit import record as audit
|
from devplacepy.services.audit import record as audit
|
||||||
|
from devplacepy.services.openai_gateway.reliability import retry_send
|
||||||
from devplacepy.services.seo_meta import schedule_seo_meta
|
from devplacepy.services.seo_meta import schedule_seo_meta
|
||||||
|
|
||||||
from . import _get_ai_key
|
from . import _get_ai_key
|
||||||
@@ -151,6 +153,32 @@ class NewsService(BaseService):
|
|||||||
),
|
),
|
||||||
group="AI formatting",
|
group="AI formatting",
|
||||||
),
|
),
|
||||||
|
ConfigField(
|
||||||
|
"news_fetch_retries",
|
||||||
|
"News API fetch retries",
|
||||||
|
type="int",
|
||||||
|
default=3,
|
||||||
|
minimum=0,
|
||||||
|
maximum=10,
|
||||||
|
help=(
|
||||||
|
"How many times to retry the upstream news API on failure. "
|
||||||
|
"Each retry waits longer (linear backoff). Set 0 for no retries."
|
||||||
|
),
|
||||||
|
group="Reliability",
|
||||||
|
),
|
||||||
|
ConfigField(
|
||||||
|
"news_fetch_retry_backoff_ms",
|
||||||
|
"News API retry backoff (ms)",
|
||||||
|
type="int",
|
||||||
|
default=5000,
|
||||||
|
minimum=1000,
|
||||||
|
maximum=60000,
|
||||||
|
help=(
|
||||||
|
"Base backoff in milliseconds between retries. The actual "
|
||||||
|
"delay is backoff * attempt number."
|
||||||
|
),
|
||||||
|
group="Reliability",
|
||||||
|
),
|
||||||
]
|
]
|
||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
@@ -165,13 +193,38 @@ class NewsService(BaseService):
|
|||||||
format_enabled = config["news_format_enabled"]
|
format_enabled = config["news_format_enabled"]
|
||||||
|
|
||||||
self.log(f"Fetching news from {api_url}")
|
self.log(f"Fetching news from {api_url}")
|
||||||
|
max_retries = config["news_fetch_retries"]
|
||||||
|
backoff_ms = config["news_fetch_retry_backoff_ms"]
|
||||||
async with stealth.stealth_async_client(timeout=30.0) as client:
|
async with stealth.stealth_async_client(timeout=30.0) as client:
|
||||||
try:
|
try:
|
||||||
resp = await client.get(api_url)
|
resp, exc, attempts = await retry_send(
|
||||||
resp.raise_for_status()
|
do_call=lambda: client.get(api_url),
|
||||||
|
max_retries=max_retries,
|
||||||
|
backoff_ms=backoff_ms,
|
||||||
|
log=lambda msg: self.log(msg),
|
||||||
|
)
|
||||||
|
if exc is not None:
|
||||||
|
raise exc
|
||||||
|
if resp is None or resp.status_code >= 400:
|
||||||
|
raise httpx.HTTPError(f"status {resp.status_code if resp else 0}")
|
||||||
data = resp.json()
|
data = resp.json()
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
self.log(f"Failed to fetch news API: {e}")
|
self.log(f"Failed to fetch news API after {attempts} attempts: {e}")
|
||||||
|
audit.record_system(
|
||||||
|
"news.service.fetch_failed",
|
||||||
|
actor_kind="service",
|
||||||
|
actor_uid="news",
|
||||||
|
summary=f"News API unreachable after {attempts} attempts",
|
||||||
|
metadata={"api_url": api_url, "attempts": attempts, "error": str(e)},
|
||||||
|
result="failure",
|
||||||
|
)
|
||||||
|
for admin_row in get_table("users").find(role="Admin"):
|
||||||
|
create_notification(
|
||||||
|
admin_row["uid"],
|
||||||
|
"system",
|
||||||
|
f"News API unreachable after {attempts} attempts",
|
||||||
|
related_uid="",
|
||||||
|
)
|
||||||
return
|
return
|
||||||
|
|
||||||
articles = data.get("articles", [])
|
articles = data.get("articles", [])
|
||||||
|
|||||||
@@ -181,27 +181,3 @@ def test_rate_limit_block_recorded(monkeypatch):
|
|||||||
event_key="security.rate_limit.block", result="denied"
|
event_key="security.rate_limit.block", result="denied"
|
||||||
)
|
)
|
||||||
assert event is not None
|
assert event is not None
|
||||||
|
|
||||||
|
|
||||||
def test_rate_limit_email_block_recorded(monkeypatch):
|
|
||||||
import devplacepy.main as m
|
|
||||||
from starlette.testclient import TestClient
|
|
||||||
|
|
||||||
monkeypatch.setattr(m, "LOGIN_EMAIL_RATE_LIMIT", 2)
|
|
||||||
m._email_rate_limit_store.clear()
|
|
||||||
client = TestClient(m.app)
|
|
||||||
email = "rate-limited@test.dev"
|
|
||||||
codes = [
|
|
||||||
client.post(
|
|
||||||
"/auth/login",
|
|
||||||
data={"email": email, "password": "wrong"},
|
|
||||||
).status_code
|
|
||||||
for _ in range(3)
|
|
||||||
]
|
|
||||||
first_two = codes[:2]
|
|
||||||
assert all(c == 401 for c in first_two), first_two
|
|
||||||
assert codes[2] == 429, codes
|
|
||||||
event = get_table("audit_log").find_one(
|
|
||||||
event_key="security.rate_limit.email_block", result="denied"
|
|
||||||
)
|
|
||||||
assert event is not None
|
|
||||||
|
|||||||
@@ -23,7 +23,6 @@ os.environ["DEVPLACE_DATA_DIR"] = str(_TEST_DATA_DIR)
|
|||||||
os.environ["SECRET_KEY"] = "test-secret-key"
|
os.environ["SECRET_KEY"] = "test-secret-key"
|
||||||
os.environ["DEVPLACE_DISABLE_SERVICES"] = "1"
|
os.environ["DEVPLACE_DISABLE_SERVICES"] = "1"
|
||||||
os.environ["DEVPLACE_RATE_LIMIT"] = "1000000"
|
os.environ["DEVPLACE_RATE_LIMIT"] = "1000000"
|
||||||
os.environ["DEVPLACE_LOGIN_EMAIL_RATE_LIMIT"] = "1000000"
|
|
||||||
# Pin a single web worker so the per-worker rate-limit divisor is 1 regardless of
|
# Pin a single web worker so the per-worker rate-limit divisor is 1 regardless of
|
||||||
# any DEVPLACE_WEB_WORKERS the host (or a server .env) exports.
|
# any DEVPLACE_WEB_WORKERS the host (or a server .env) exports.
|
||||||
os.environ["DEVPLACE_WEB_WORKERS"] = "1"
|
os.environ["DEVPLACE_WEB_WORKERS"] = "1"
|
||||||
|
|||||||
@@ -67,9 +67,14 @@ class FakeClient_news_service:
|
|||||||
grade = "9" if "HighArticle" in prompt else "3"
|
grade = "9" if "HighArticle" in prompt else "3"
|
||||||
return FakeResp_news_service(json_data={"choices": [{"message": {"content": grade}}]})
|
return FakeResp_news_service(json_data={"choices": [{"message": {"content": grade}}]})
|
||||||
class FailingApiClient(FakeClient_news_service):
|
class FailingApiClient(FakeClient_news_service):
|
||||||
|
def __init__(self, *args, **kwargs):
|
||||||
|
super().__init__(*args, **kwargs)
|
||||||
|
self.call_count = 0
|
||||||
|
|
||||||
async def get(self, url, timeout=None):
|
async def get(self, url, timeout=None):
|
||||||
|
self.call_count += 1
|
||||||
if url == API_URL:
|
if url == API_URL:
|
||||||
raise httpx.HTTPError("api down")
|
raise httpx.RequestError("api down")
|
||||||
return FakeResp_news_service(text="")
|
return FakeResp_news_service(text="")
|
||||||
GATEWAY_HEADERS = {
|
GATEWAY_HEADERS = {
|
||||||
"X-Gateway-Cost-USD": "0.00010000",
|
"X-Gateway-Cost-USD": "0.00010000",
|
||||||
@@ -98,7 +103,7 @@ class UsageClient(FakeClient_news_service):
|
|||||||
json_data={"choices": [{"message": {"content": grade}}]},
|
json_data={"choices": [{"message": {"content": grade}}]},
|
||||||
headers=GATEWAY_HEADERS,
|
headers=GATEWAY_HEADERS,
|
||||||
)
|
)
|
||||||
def _settings_stub(threshold="7"):
|
def _settings_stub(threshold="7", retries="3", backoff="5"):
|
||||||
def fake_get_setting(key, default=None):
|
def fake_get_setting(key, default=None):
|
||||||
return {
|
return {
|
||||||
"news_api_url": API_URL,
|
"news_api_url": API_URL,
|
||||||
@@ -106,6 +111,8 @@ def _settings_stub(threshold="7"):
|
|||||||
"news_ai_model": "test-model",
|
"news_ai_model": "test-model",
|
||||||
"news_grade_threshold": threshold,
|
"news_grade_threshold": threshold,
|
||||||
"news_ai_key": "",
|
"news_ai_key": "",
|
||||||
|
"news_fetch_retries": retries,
|
||||||
|
"news_fetch_retry_backoff_ms": backoff,
|
||||||
}.get(key, default)
|
}.get(key, default)
|
||||||
|
|
||||||
return fake_get_setting
|
return fake_get_setting
|
||||||
@@ -171,12 +178,47 @@ def test_grade_article_unparseable_returns_none(local_db, monkeypatch):
|
|||||||
|
|
||||||
|
|
||||||
def test_run_once_handles_api_failure(local_db, monkeypatch):
|
def test_run_once_handles_api_failure(local_db, monkeypatch):
|
||||||
|
admin_uid = generate_uid()
|
||||||
|
get_table("users").insert({
|
||||||
|
"uid": admin_uid,
|
||||||
|
"username": "testadmin",
|
||||||
|
"role": "Admin",
|
||||||
|
"email": "admin@test.test",
|
||||||
|
"password": "hash",
|
||||||
|
"api_key": generate_uid(),
|
||||||
|
"created_at": "2025-01-01T00:00:00Z",
|
||||||
|
})
|
||||||
monkeypatch.setattr(news_mod, "get_setting", _settings_stub())
|
monkeypatch.setattr(news_mod, "get_setting", _settings_stub())
|
||||||
monkeypatch.setattr(base_mod, "get_setting", _settings_stub())
|
monkeypatch.setattr(base_mod, "get_setting", _settings_stub())
|
||||||
|
failing_client = FailingApiClient([])
|
||||||
monkeypatch.setattr(
|
monkeypatch.setattr(
|
||||||
news_mod.httpx, "AsyncClient", lambda *a, **k: FailingApiClient([])
|
news_mod.stealth, "stealth_async_client",
|
||||||
|
lambda *a, **k: failing_client,
|
||||||
|
)
|
||||||
|
notifications = []
|
||||||
|
monkeypatch.setattr(
|
||||||
|
news_mod, "create_notification",
|
||||||
|
lambda user_uid, notification_type, message, related_uid, target_url=None: (
|
||||||
|
notifications.append((user_uid, message))
|
||||||
|
),
|
||||||
)
|
)
|
||||||
run_async(NewsService().run_once())
|
run_async(NewsService().run_once())
|
||||||
|
assert failing_client.call_count == 4
|
||||||
|
assert len(notifications) == 1
|
||||||
|
assert notifications[0][0] == admin_uid
|
||||||
|
assert "News API unreachable" in notifications[0][1]
|
||||||
|
|
||||||
|
|
||||||
|
def test_run_once_handles_api_failure_zero_retries(local_db, monkeypatch):
|
||||||
|
monkeypatch.setattr(news_mod, "get_setting", _settings_stub(retries="0"))
|
||||||
|
monkeypatch.setattr(base_mod, "get_setting", _settings_stub(retries="0"))
|
||||||
|
failing_client = FailingApiClient([])
|
||||||
|
monkeypatch.setattr(
|
||||||
|
news_mod.stealth, "stealth_async_client",
|
||||||
|
lambda *a, **k: failing_client,
|
||||||
|
)
|
||||||
|
run_async(NewsService().run_once())
|
||||||
|
assert failing_client.call_count == 1
|
||||||
|
|
||||||
|
|
||||||
def test_run_once_updates_existing_news_row(local_db, monkeypatch):
|
def test_run_once_updates_existing_news_row(local_db, monkeypatch):
|
||||||
|
|||||||
Reference in New Issue
Block a user