update endopoints
This commit is contained in:
+1
-8
@@ -18,9 +18,6 @@ from fastapi.middleware.cors import CORSMiddleware
|
|||||||
from events import delete_event, get_room_events
|
from events import delete_event, get_room_events
|
||||||
from rooms import regenerate_invite_code
|
from rooms import regenerate_invite_code
|
||||||
|
|
||||||
class InviteCodeUpdate(BaseModel):
|
|
||||||
invite_code: str
|
|
||||||
|
|
||||||
class Login(BaseModel):
|
class Login(BaseModel):
|
||||||
username: str
|
username: str
|
||||||
password: str
|
password: str
|
||||||
@@ -269,7 +266,6 @@ def api_join_room(
|
|||||||
@app.post("/rooms/{room_id}/invite-code")
|
@app.post("/rooms/{room_id}/invite-code")
|
||||||
def update_invite_code(
|
def update_invite_code(
|
||||||
room_id: int,
|
room_id: int,
|
||||||
data: InviteCodeUpdate,
|
|
||||||
user_id: int = Depends(get_current_user)
|
user_id: int = Depends(get_current_user)
|
||||||
):
|
):
|
||||||
if not has_role(room_id, user_id, ["owner", "admin"]):
|
if not has_role(room_id, user_id, ["owner", "admin"]):
|
||||||
@@ -278,10 +274,7 @@ def update_invite_code(
|
|||||||
"reason": "no_permission"
|
"reason": "no_permission"
|
||||||
}
|
}
|
||||||
|
|
||||||
return regenerate_invite_code(
|
return regenerate_invite_code(room_id)
|
||||||
room_id,
|
|
||||||
data.invite_code
|
|
||||||
)
|
|
||||||
|
|
||||||
# events
|
# events
|
||||||
|
|
||||||
|
|||||||
+10
-5
@@ -280,17 +280,22 @@ def change_room_name(room_id, new_name, user_id):
|
|||||||
"updated_by": user_id
|
"updated_by": user_id
|
||||||
}
|
}
|
||||||
|
|
||||||
def regenerate_invite_code(room_id, new_code):
|
import secrets
|
||||||
|
import string
|
||||||
|
|
||||||
|
def regenerate_invite_code(room_id):
|
||||||
|
new_code = ''.join(
|
||||||
|
secrets.choice(string.ascii_uppercase + string.digits)
|
||||||
|
for _ in range(8)
|
||||||
|
)
|
||||||
|
|
||||||
execute(
|
execute(
|
||||||
"""
|
"""
|
||||||
UPDATE rooms
|
UPDATE rooms
|
||||||
SET invite_code = ?
|
SET invite_code = ?
|
||||||
WHERE id = ?
|
WHERE id = ?
|
||||||
""",
|
""",
|
||||||
(
|
(new_code, room_id)
|
||||||
new_code,
|
|
||||||
room_id
|
|
||||||
)
|
|
||||||
)
|
)
|
||||||
|
|
||||||
return {
|
return {
|
||||||
|
|||||||
Reference in New Issue
Block a user