github-starred/open-webui
2
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-04-29 17:06:40 -05:00
Code Issues 786 Packages Projects Releases 126 Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2026-02-11 16:45:47 -06:00
parent 60ada21c15
commit 96c07f44a8
1 changed files with 20 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
backend/open_webui/models/groups.py
View File

@@ -216,13 +216,13 @@ class GroupTable:
).filter(GroupMember.user_id == filter["member_id"])
groups = query.order_by(Group.updated_at.desc()).all()
group_ids = [group.id for group in groups]
member_counts = self.get_group_member_counts_by_ids(group_ids, db=db)
return [
GroupResponse.model_validate(
{
**GroupModel.model_validate(group).model_dump(),
"member_count": self.get_group_member_count_by_id(
group.id, db=db
),
"member_count": member_counts.get(group.id, 0),
}
)
for group in groups
@@ -257,12 +257,14 @@ class GroupTable:
total = query.count()
query = query.order_by(Group.updated_at.desc())
groups = query.offset(skip).limit(limit).all()
group_ids = [group.id for group in groups]
member_counts = self.get_group_member_counts_by_ids(group_ids, db=db)
return {
"items": [
GroupResponse.model_validate(
**GroupModel.model_validate(group).model_dump(),
member_count=self.get_group_member_count_by_id(group.id, db=db),
member_count=member_counts.get(group.id, 0),
)
for group in groups
],
@@ -379,6 +381,20 @@ class GroupTable:
)
return count if count else 0
def get_group_member_counts_by_ids(
self, ids: list[str], db: Optional[Session] = None
) -> dict[str, int]:
if not ids:
return {}
with get_db_context(db) as db:
rows = (
db.query(GroupMember.group_id, func.count(GroupMember.user_id))
.filter(GroupMember.group_id.in_(ids))
.group_by(GroupMember.group_id)
.all()
)
return {group_id: count for group_id, count in rows}
def update_group_by_id(
self,
id: str,