forked from retoor/devplacepy
Compare commits
1
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
6970f8a270 |
File diff suppressed because one or more lines are too long
@@ -93,36 +93,6 @@ def get_recent_comments_by_target_uids(target_type, target_uids, limit=3, user=N
|
||||
raw = _drop_blocked(raw, user)
|
||||
if not raw:
|
||||
return {}
|
||||
|
||||
# Fetch missing parent comments so the threaded hierarchy is preserved even
|
||||
# when a parent is outside the row-number limit.
|
||||
if "comments" in db.tables:
|
||||
comments_table = db["comments"]
|
||||
current_uids = {c["uid"] for c in raw}
|
||||
seen = set(current_uids)
|
||||
to_fetch = []
|
||||
for c in raw:
|
||||
puid = c.get("parent_uid")
|
||||
if puid and puid not in seen:
|
||||
seen.add(puid)
|
||||
to_fetch.append(puid)
|
||||
if to_fetch:
|
||||
ancestors = []
|
||||
queue = list(to_fetch)
|
||||
fetched = set()
|
||||
while queue:
|
||||
uid = queue.pop(0)
|
||||
if uid in fetched or uid in current_uids:
|
||||
continue
|
||||
row = comments_table.find_one(uid=uid, deleted_at=None)
|
||||
if row:
|
||||
ancestors.append(row)
|
||||
fetched.add(uid)
|
||||
grandparent = row.get("parent_uid")
|
||||
if grandparent and grandparent not in fetched and grandparent not in current_uids:
|
||||
queue.append(grandparent)
|
||||
raw.extend(ancestors)
|
||||
|
||||
items = _build_comment_items(raw, user)
|
||||
by_target = defaultdict(list)
|
||||
for c in raw:
|
||||
|
||||
@@ -18,6 +18,7 @@ NOTIFICATION_TYPES = [
|
||||
{"key": "reminder", "label": "Reminders", "description": "A reminder or scheduled task you asked Devii to run fires"},
|
||||
{"key": "harvest_stolen", "label": "Farm raids", "description": "Someone steals a ready build from your Code Farm"},
|
||||
{"key": "award", "label": "Awards", "description": "Someone gives you an award on your profile"},
|
||||
{"key": "admin_alert", "label": "Admin alerts", "description": "System alerts requiring admin attention"},
|
||||
]
|
||||
|
||||
|
||||
|
||||
@@ -23,6 +23,7 @@ from devplacepy.utils import (
|
||||
require_user,
|
||||
)
|
||||
from devplacepy.services.audit import record as audit
|
||||
from devplacepy.services.devii.quota import notify_admins_quota_blocked
|
||||
|
||||
logger = logging.getLogger("devii.router")
|
||||
|
||||
@@ -279,6 +280,7 @@ async def devii_ws(websocket: WebSocket):
|
||||
summary=f"AI request by {username or owner_id} blocked - 24h quota reached",
|
||||
metadata={"limit_usd": limit, "owner_kind": owner_kind},
|
||||
)
|
||||
notify_admins_quota_blocked(owner_kind, owner_id, username)
|
||||
await websocket.send_json(
|
||||
{
|
||||
"type": "error",
|
||||
|
||||
@@ -0,0 +1,26 @@
|
||||
# retoor <retoor@molodetz.nl>
|
||||
|
||||
import logging
|
||||
|
||||
from devplacepy.database import get_table
|
||||
from devplacepy.utils.notifications import create_notification
|
||||
|
||||
logger = logging.getLogger("devii.quota")
|
||||
|
||||
ADMIN_ALERT_TYPE = "admin_alert"
|
||||
|
||||
|
||||
def notify_admins_quota_blocked(owner_kind: str, owner_id: str, username: str) -> None:
|
||||
if owner_kind != "user":
|
||||
return
|
||||
admins = get_table("users").find(role="Admin")
|
||||
for admin in admins:
|
||||
if admin["uid"] == owner_id:
|
||||
continue
|
||||
create_notification(
|
||||
admin["uid"],
|
||||
ADMIN_ALERT_TYPE,
|
||||
f"User {username} blocked - 24h AI quota reached",
|
||||
owner_id,
|
||||
"/admin/users",
|
||||
)
|
||||
@@ -10,6 +10,7 @@ from typing import Any
|
||||
|
||||
from devplacepy.database import get_table
|
||||
from devplacepy.services.audit import record as audit
|
||||
from devplacepy.services.devii.quota import notify_admins_quota_blocked
|
||||
from devplacepy.services.manager import service_manager
|
||||
from devplacepy.utils import is_admin, is_primary_admin
|
||||
|
||||
@@ -400,6 +401,7 @@ class TelegramBridge:
|
||||
summary=f"Telegram AI request by {user.get('username', owner_id)} blocked - 24h quota reached",
|
||||
metadata={"limit_usd": limit},
|
||||
)
|
||||
notify_admins_quota_blocked("user", owner_id, user.get("username", ""))
|
||||
await self._service.send(
|
||||
chat_id, "Your daily AI quota is reached (100%). Please try again later."
|
||||
)
|
||||
|
||||
@@ -959,72 +959,3 @@ def test_feed_scroll_not_restored_on_fresh_visit(alice):
|
||||
page.goto(f"{BASE_URL}/feed", wait_until="domcontentloaded")
|
||||
page.wait_for_timeout(600)
|
||||
assert page.evaluate("window.scrollY") < 60
|
||||
|
||||
|
||||
def _seed_post_with_orphaned_parent():
|
||||
owner = str(uuid4())
|
||||
get_table("users").insert(
|
||||
{
|
||||
"uid": owner,
|
||||
"username": f"opseed_{owner[:8]}",
|
||||
"email": f"{owner[:8]}@op.seed",
|
||||
"password_hash": "x",
|
||||
"role": "Member",
|
||||
"is_active": True,
|
||||
"created_at": datetime.now(timezone.utc).isoformat(),
|
||||
}
|
||||
)
|
||||
post_uid = str(uuid4())
|
||||
marker = f"oppost-{post_uid[:8]}"
|
||||
get_table("posts").insert(
|
||||
{
|
||||
"deleted_at": None,
|
||||
"deleted_by": None,
|
||||
"uid": post_uid,
|
||||
"user_uid": owner,
|
||||
"slug": make_combined_slug(marker, post_uid),
|
||||
"title": None,
|
||||
"content": marker,
|
||||
"topic": "devlog",
|
||||
"project_uid": None,
|
||||
"image": None,
|
||||
"stars": 0,
|
||||
"created_at": datetime.now(timezone.utc).isoformat(),
|
||||
}
|
||||
)
|
||||
orphan_parent = str(uuid4())
|
||||
base = datetime.now(timezone.utc)
|
||||
rows = [
|
||||
(f"{marker}-orphan-parent", orphan_parent, None, -3),
|
||||
(f"{marker}-older", str(uuid4()), None, -2),
|
||||
(f"{marker}-flat", str(uuid4()), None, -1),
|
||||
(f"{marker}-reply", str(uuid4()), orphan_parent, 0),
|
||||
]
|
||||
comments = get_table("comments")
|
||||
for content, cuid, par, off in rows:
|
||||
comments.insert(
|
||||
{
|
||||
"deleted_at": None,
|
||||
"deleted_by": None,
|
||||
"uid": cuid,
|
||||
"target_type": "post",
|
||||
"target_uid": post_uid,
|
||||
"post_uid": post_uid,
|
||||
"user_uid": owner,
|
||||
"content": content,
|
||||
"parent_uid": par,
|
||||
"created_at": (base + timedelta(seconds=off)).isoformat(),
|
||||
}
|
||||
)
|
||||
return marker
|
||||
|
||||
|
||||
def test_feed_preserves_comment_hierarchy_with_orphaned_parent(page, app_server):
|
||||
marker = _seed_post_with_orphaned_parent()
|
||||
page.goto(f"{BASE_URL}/feed", wait_until="domcontentloaded")
|
||||
card = page.locator(".post-card").filter(has_text=marker).first
|
||||
card.wait_for(state="visible")
|
||||
expect(card.locator(".post-card-comments")).to_contain_text(f"{marker}-orphan-parent")
|
||||
expect(card.locator(".post-card-comments")).to_contain_text(f"{marker}-reply")
|
||||
nested = card.locator(".post-card-comments .comment-replies .comment-text")
|
||||
expect(nested).to_contain_text(f"{marker}-reply")
|
||||
|
||||
@@ -468,70 +468,3 @@ def test_gists_list_preserves_comment_hierarchy(page, app_server):
|
||||
expect(card.locator(".post-card-comments")).not_to_contain_text(f"{marker}-excluded")
|
||||
nested = card.locator(".post-card-comments .comment-replies .comment-text")
|
||||
expect(nested).to_contain_text(f"{marker}-reply")
|
||||
|
||||
|
||||
def _seed_gist_with_orphaned_parent():
|
||||
owner = str(uuid4())
|
||||
get_table("users").insert(
|
||||
{
|
||||
"uid": owner,
|
||||
"username": f"gopseed_{owner[:8]}",
|
||||
"email": f"{owner[:8]}@gop.seed",
|
||||
"password_hash": "x",
|
||||
"role": "Member",
|
||||
"is_active": True,
|
||||
"created_at": datetime.now(timezone.utc).isoformat(),
|
||||
}
|
||||
)
|
||||
uid = str(uuid4())
|
||||
marker = f"gistop-{uid[:8]}"
|
||||
get_table("gists").insert(
|
||||
{
|
||||
"deleted_at": None,
|
||||
"deleted_by": None,
|
||||
"uid": uid,
|
||||
"user_uid": owner,
|
||||
"slug": make_combined_slug(marker, uid),
|
||||
"title": marker,
|
||||
"description": "Gist with orphaned parent hierarchy.",
|
||||
"source_code": "print('x')",
|
||||
"language": "python",
|
||||
"stars": 0,
|
||||
"created_at": datetime.now(timezone.utc).isoformat(),
|
||||
}
|
||||
)
|
||||
orphan_parent = str(uuid4())
|
||||
base = datetime.now(timezone.utc)
|
||||
rows = [
|
||||
(f"{marker}-orphan-parent", orphan_parent, None, -3),
|
||||
(f"{marker}-older", str(uuid4()), None, -2),
|
||||
(f"{marker}-flat", str(uuid4()), None, -1),
|
||||
(f"{marker}-reply", str(uuid4()), orphan_parent, 0),
|
||||
]
|
||||
comments = get_table("comments")
|
||||
for content, cuid, par, off in rows:
|
||||
comments.insert(
|
||||
{
|
||||
"deleted_at": None,
|
||||
"deleted_by": None,
|
||||
"uid": cuid,
|
||||
"target_type": "gist",
|
||||
"target_uid": uid,
|
||||
"user_uid": owner,
|
||||
"content": content,
|
||||
"parent_uid": par,
|
||||
"created_at": (base + timedelta(seconds=off)).isoformat(),
|
||||
}
|
||||
)
|
||||
return marker
|
||||
|
||||
|
||||
def test_gists_list_preserves_comment_hierarchy_with_orphaned_parent(page, app_server):
|
||||
marker = _seed_gist_with_orphaned_parent()
|
||||
page.goto(f"{BASE_URL}/gists", wait_until="domcontentloaded")
|
||||
card = page.locator(".gist-card").filter(has_text=marker).first
|
||||
card.wait_for(state="visible")
|
||||
expect(card.locator(".post-card-comments")).to_contain_text(f"{marker}-orphan-parent")
|
||||
expect(card.locator(".post-card-comments")).to_contain_text(f"{marker}-reply")
|
||||
nested = card.locator(".post-card-comments .comment-replies .comment-text")
|
||||
expect(nested).to_contain_text(f"{marker}-reply")
|
||||
|
||||
@@ -387,75 +387,3 @@ def test_news_list_preserves_comment_hierarchy(page, app_server):
|
||||
expect(card.locator(".post-card-comments")).not_to_contain_text(f"{marker}-excluded")
|
||||
nested = card.locator(".post-card-comments .comment-replies .comment-text")
|
||||
expect(nested).to_contain_text(f"{marker}-reply")
|
||||
|
||||
|
||||
def _seed_news_with_orphaned_parent():
|
||||
owner = str(uuid4())
|
||||
get_table("users").insert(
|
||||
{
|
||||
"uid": owner,
|
||||
"username": f"nopseed_{owner[:8]}",
|
||||
"email": f"{owner[:8]}@nop.seed",
|
||||
"password_hash": "x",
|
||||
"role": "Member",
|
||||
"is_active": True,
|
||||
"created_at": datetime.now(timezone.utc).isoformat(),
|
||||
}
|
||||
)
|
||||
uid = str(uuid4())
|
||||
marker = f"newsop-{uid[:8]}"
|
||||
get_table("news").insert(
|
||||
{
|
||||
"deleted_at": None,
|
||||
"deleted_by": None,
|
||||
"uid": uid,
|
||||
"slug": make_combined_slug(marker, uid),
|
||||
"title": marker,
|
||||
"description": "News with orphaned parent hierarchy.",
|
||||
"content": "Body.",
|
||||
"url": "https://example.com/nop",
|
||||
"source_name": "NOPSource",
|
||||
"external_id": f"nop_{uid[:8]}",
|
||||
"status": "published",
|
||||
"show_on_landing": 0,
|
||||
"grade": 10,
|
||||
"synced_at": datetime.now(timezone.utc).isoformat(),
|
||||
}
|
||||
)
|
||||
orphan_parent = str(uuid4())
|
||||
base = datetime.now(timezone.utc)
|
||||
rows = [
|
||||
(f"{marker}-orphan-parent", orphan_parent, None, -3),
|
||||
(f"{marker}-older", str(uuid4()), None, -2),
|
||||
(f"{marker}-flat", str(uuid4()), None, -1),
|
||||
(f"{marker}-reply", str(uuid4()), orphan_parent, 0),
|
||||
]
|
||||
comments = get_table("comments")
|
||||
for content, cuid, par, off in rows:
|
||||
comments.insert(
|
||||
{
|
||||
"deleted_at": None,
|
||||
"deleted_by": None,
|
||||
"uid": cuid,
|
||||
"target_type": "news",
|
||||
"target_uid": uid,
|
||||
"user_uid": owner,
|
||||
"content": content,
|
||||
"parent_uid": par,
|
||||
"created_at": (base + timedelta(seconds=off)).isoformat(),
|
||||
}
|
||||
)
|
||||
return marker
|
||||
|
||||
|
||||
def test_news_list_preserves_comment_hierarchy_with_orphaned_parent(page, app_server):
|
||||
from playwright.sync_api import expect
|
||||
|
||||
marker = _seed_news_with_orphaned_parent()
|
||||
page.goto(f"{BASE_URL}/news", wait_until="domcontentloaded")
|
||||
card = page.locator(".news-card").filter(has_text=marker).first
|
||||
card.wait_for(state="visible")
|
||||
expect(card.locator(".post-card-comments")).to_contain_text(f"{marker}-orphan-parent")
|
||||
expect(card.locator(".post-card-comments")).to_contain_text(f"{marker}-reply")
|
||||
nested = card.locator(".post-card-comments .comment-replies .comment-text")
|
||||
expect(nested).to_contain_text(f"{marker}-reply")
|
||||
|
||||
@@ -1015,71 +1015,3 @@ def test_containers_menu_hidden_for_non_owner_admin_on_member_private_project(pa
|
||||
page.goto(f"{BASE_URL}/projects/{slug}", wait_until="domcontentloaded")
|
||||
page.locator(".project-actions-more").click()
|
||||
expect(page.locator(".context-menu-item:has-text('Containers')")).to_have_count(0)
|
||||
|
||||
|
||||
def _seed_project_with_orphaned_parent():
|
||||
owner = str(uuid4())
|
||||
get_table("users").insert(
|
||||
{
|
||||
"uid": owner,
|
||||
"username": f"popseed_{owner[:8]}",
|
||||
"email": f"{owner[:8]}@pop.seed",
|
||||
"password_hash": "x",
|
||||
"role": "Member",
|
||||
"is_active": True,
|
||||
"created_at": datetime.now(timezone.utc).isoformat(),
|
||||
}
|
||||
)
|
||||
uid = str(uuid4())
|
||||
marker = f"projop-{uid[:8]}"
|
||||
get_table("projects").insert(
|
||||
{
|
||||
"deleted_at": None,
|
||||
"deleted_by": None,
|
||||
"uid": uid,
|
||||
"user_uid": owner,
|
||||
"slug": make_combined_slug(marker, uid),
|
||||
"title": marker,
|
||||
"description": "Project with orphaned parent hierarchy.",
|
||||
"project_type": "software",
|
||||
"platforms": "Linux",
|
||||
"status": "Released",
|
||||
"stars": 0,
|
||||
"created_at": datetime.now(timezone.utc).isoformat(),
|
||||
}
|
||||
)
|
||||
orphan_parent = str(uuid4())
|
||||
base = datetime.now(timezone.utc)
|
||||
rows = [
|
||||
(f"{marker}-orphan-parent", orphan_parent, None, -3),
|
||||
(f"{marker}-older", str(uuid4()), None, -2),
|
||||
(f"{marker}-flat", str(uuid4()), None, -1),
|
||||
(f"{marker}-reply", str(uuid4()), orphan_parent, 0),
|
||||
]
|
||||
comments = get_table("comments")
|
||||
for content, cuid, par, off in rows:
|
||||
comments.insert(
|
||||
{
|
||||
"deleted_at": None,
|
||||
"deleted_by": None,
|
||||
"uid": cuid,
|
||||
"target_type": "project",
|
||||
"target_uid": uid,
|
||||
"user_uid": owner,
|
||||
"content": content,
|
||||
"parent_uid": par,
|
||||
"created_at": (base + timedelta(seconds=off)).isoformat(),
|
||||
}
|
||||
)
|
||||
return marker
|
||||
|
||||
|
||||
def test_projects_list_preserves_comment_hierarchy_with_orphaned_parent(page, app_server):
|
||||
marker = _seed_project_with_orphaned_parent()
|
||||
page.goto(f"{BASE_URL}/projects", wait_until="domcontentloaded")
|
||||
card = page.locator(".project-card").filter(has_text=marker).first
|
||||
card.wait_for(state="visible")
|
||||
expect(card.locator(".post-card-comments")).to_contain_text(f"{marker}-orphan-parent")
|
||||
expect(card.locator(".post-card-comments")).to_contain_text(f"{marker}-reply")
|
||||
nested = card.locator(".post-card-comments .comment-replies .comment-text")
|
||||
expect(nested).to_contain_text(f"{marker}-reply")
|
||||
|
||||
Reference in New Issue
Block a user