UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-02-16 18:22:29 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2024-11-14 18:37:29 -08:00
parent 659f3dac44
commit 2f893fd373
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
backend/open_webui/apps/webui/models/groups.py
View File

@ -102,7 +102,7 @@ class GroupTable:
) )
try: try:
result = Groups(**group.model_dump()) result = Group(**group.model_dump())
db.add(result) db.add(result)
db.commit() db.commit()
db.refresh(result) db.refresh(result)
@ -118,13 +118,13 @@ class GroupTable:
with get_db() as db: with get_db() as db:
return [ return [
GroupModel.model_validate(group) GroupModel.model_validate(group)
for group in db.query(Groups).order_by(Groups.updated_at.desc()).all() for group in db.query(Group).order_by(Group.updated_at.desc()).all()
] ]
def get_group_by_id(self, id: str) -> Optional[GroupModel]: def get_group_by_id(self, id: str) -> Optional[GroupModel]:
try: try:
with get_db() as db: with get_db() as db:
group = db.query(Groups).filter_by(id=id).first() group = db.query(Group).filter_by(id=id).first()
return GroupModel.model_validate(group) if group else None return GroupModel.model_validate(group) if group else None
except Exception: except Exception:
return None return None
@ -134,7 +134,7 @@ class GroupTable:
) -> Optional[GroupModel]: ) -> Optional[GroupModel]:
try: try:
with get_db() as db: with get_db() as db:
db.query(Groups).filter_by(id=id).update( db.query(Group).filter_by(id=id).update(
{ {
**form_data.model_dump(exclude_none=True), **form_data.model_dump(exclude_none=True),
"updated_at": int(time.time()), "updated_at": int(time.time()),
@ -149,7 +149,7 @@ class GroupTable:
def delete_group_by_id(self, id: str) -> bool: def delete_group_by_id(self, id: str) -> bool:
try: try:
with get_db() as db: with get_db() as db:
db.query(Groups).filter_by(id=id).delete() db.query(Group).filter_by(id=id).delete()
db.commit() db.commit()
return True return True
except Exception: except Exception:
@ -158,7 +158,7 @@ class GroupTable:
def delete_all_groups(self) -> bool: def delete_all_groups(self) -> bool:
with get_db() as db: with get_db() as db:
try: try:
db.query(Groups).delete() db.query(Group).delete()
db.commit() db.commit()
return True return True