From 405a5379eff04e0be354038e28ed2c26163f51f9 Mon Sep 17 00:00:00 2001 From: Devansh Singh Date: Mon, 2 Sep 2024 11:55:09 +0530 Subject: [PATCH] Convert username to lowercase, update request models for presets Removed the suite attribute from PresetArgs, created a new model for /update with optional preset fields Signed-off-by: Devansh Singh --- src/teuthology_api/routes/presets.py | 20 +++++++++++++------- src/teuthology_api/schemas/preset.py | 14 ++++++++++++-- 2 files changed, 25 insertions(+), 9 deletions(-) diff --git a/src/teuthology_api/routes/presets.py b/src/teuthology_api/routes/presets.py index 9cb8578..755d260 100644 --- a/src/teuthology_api/routes/presets.py +++ b/src/teuthology_api/routes/presets.py @@ -5,7 +5,7 @@ from sqlmodel import Session from teuthology_api.models import get_db, Presets -from teuthology_api.schemas.preset import PresetArgs +from teuthology_api.schemas.preset import PresetArgs, PresetUpdateArgs from teuthology_api.services.helpers import get_token, get_username from teuthology_api.services.presets import PresetsDatabaseException, PresetsService @@ -16,6 +16,8 @@ @router.get("/", status_code=status.HTTP_200_OK) def read_preset(username: str, name: str, db: Session = Depends(get_db)): + # GitHub usernames are case-insensitive + username = username.lower() db_preset = PresetsService(db).get_by_username_and_name(username, name) if db_preset is None: raise HTTPException( @@ -29,6 +31,7 @@ def read_preset(username: str, name: str, db: Session = Depends(get_db)): def read_all_presets( username: str, suite: Optional[str] = None, db: Session = Depends(get_db) ): + username = username.lower() if suite: db_presets = PresetsService(db).get_by_username_and_suite(username, suite) else: @@ -56,7 +59,7 @@ def add_preset( headers={"WWW-Authenticate": "Bearer"}, ) - username = get_username(request) + username = get_username(request).lower() db_presets = PresetsService(db).get_by_username(username) if len(db_presets) == 10: if not replace: @@ -77,14 +80,16 @@ def add_preset( detail=f"Preset with name {preset.name} exists", ) - db_preset = Presets(**preset.model_dump(), username=username) + db_preset = Presets( + **preset.model_dump(), username=username, suite=preset.cmd["--suite"] + ) return PresetsService(db).create(db_preset) @router.put("/edit/{preset_id}", status_code=status.HTTP_200_OK) def update_preset( preset_id: int, - updated_preset: PresetArgs, + preset: PresetUpdateArgs, db: Session = Depends(get_db), access_token: str = Depends(get_token), ): @@ -95,9 +100,10 @@ def update_preset( headers={"WWW-Authenticate": "Bearer"}, ) try: - return PresetsService(db).update( - preset_id, updated_preset.model_dump(exclude_unset=True) - ) + updated_preset = preset.model_dump(exclude_unset=True) + if preset.cmd: + updated_preset["suite"] = preset.cmd["--suite"] + return PresetsService(db).update(preset_id, updated_preset) except PresetsDatabaseException as exc: raise HTTPException(status_code=exc.code, detail=str(exc)) diff --git a/src/teuthology_api/schemas/preset.py b/src/teuthology_api/schemas/preset.py index 23fe19b..828aae3 100644 --- a/src/teuthology_api/schemas/preset.py +++ b/src/teuthology_api/schemas/preset.py @@ -1,3 +1,5 @@ +from typing import Optional + from pydantic import BaseModel from teuthology_api.schemas.suite import SuiteArgs @@ -5,8 +7,16 @@ class PresetArgs(BaseModel): name: str - suite: str cmd: SuiteArgs def model_post_init(self, __context): - self.cmd = self.cmd.model_dump() + self.cmd = self.cmd.model_dump(by_alias=True) + + +class PresetUpdateArgs(BaseModel): + name: Optional[str] = None + cmd: Optional[SuiteArgs] = None + + def model_post_init(self, __context): + if self.cmd: + self.cmd = self.cmd.model_dump(by_alias=True)