Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
open-webui
Commits
68cfcced
Commit
68cfcced
authored
May 01, 2024
by
Timothy J. Baek
Browse files
feat: super-admin (first one to signup)
parent
a4333153
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
8 additions
and
1 deletion
+8
-1
backend/apps/web/models/users.py
backend/apps/web/models/users.py
+7
-0
backend/apps/web/routers/users.py
backend/apps/web/routers/users.py
+1
-1
No files found.
backend/apps/web/models/users.py
View file @
68cfcced
...
...
@@ -123,6 +123,13 @@ class UsersTable:
def
get_num_users
(
self
)
->
Optional
[
int
]:
return
User
.
select
().
count
()
def
get_first_user
(
self
)
->
UserModel
:
try
:
user
=
User
.
select
().
order_by
(
User
.
created_at
).
first
()
return
UserModel
(
**
model_to_dict
(
user
))
except
:
return
None
def
update_user_role_by_id
(
self
,
id
:
str
,
role
:
str
)
->
Optional
[
UserModel
]:
try
:
query
=
User
.
update
(
role
=
role
).
where
(
User
.
id
==
id
)
...
...
backend/apps/web/routers/users.py
View file @
68cfcced
...
...
@@ -58,7 +58,7 @@ async def update_user_permissions(
@
router
.
post
(
"/update/role"
,
response_model
=
Optional
[
UserModel
])
async
def
update_user_role
(
form_data
:
UserRoleUpdateForm
,
user
=
Depends
(
get_admin_user
)):
if
user
.
id
!=
form_data
.
id
:
if
user
.
id
!=
form_data
.
id
and
form_data
.
id
!=
Users
.
get_first_user
().
id
:
return
Users
.
update_user_role_by_id
(
form_data
.
id
,
form_data
.
role
)
raise
HTTPException
(
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment