Fixed syntax misc issues
This commit is contained in:
parent
c07423f8b0
commit
9cc09014a4
3 changed files with 33 additions and 14 deletions
|
|
@ -1,3 +1,5 @@
|
||||||
|
from sqlmodel import inspect
|
||||||
|
|
||||||
from app.database import (
|
from app.database import (
|
||||||
add_to_database,
|
add_to_database,
|
||||||
delete_from_database_where,
|
delete_from_database_where,
|
||||||
|
|
@ -17,10 +19,10 @@ def get_by_id(project_id: int):
|
||||||
return get_from_database_where_id(Project, project_id)
|
return get_from_database_where_id(Project, project_id)
|
||||||
|
|
||||||
|
|
||||||
def get_with_parent(project_id: int):
|
def get_with_parent(project_id: int | None):
|
||||||
return get_from_database_where(
|
return get_from_database_where(
|
||||||
Project,
|
Project,
|
||||||
Project.parent_id,
|
Project.id, # type: ignore (Type correct on runtime)
|
||||||
project_id,
|
project_id,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
@ -35,11 +37,16 @@ def update(
|
||||||
update_attribute: str,
|
update_attribute: str,
|
||||||
new_value: int | str | None,
|
new_value: int | str | None,
|
||||||
):
|
):
|
||||||
|
if update_attribute not in inspect(Project).attrs:
|
||||||
|
raise ValueError(f"Invalid attribute: {update_attribute}")
|
||||||
|
|
||||||
|
attribute = getattr(Project, update_attribute)
|
||||||
|
|
||||||
return update_database_where(
|
return update_database_where(
|
||||||
Project,
|
Project,
|
||||||
Project.id,
|
Project.id, # type: ignore (Type correct on runtime)
|
||||||
project_id,
|
project_id,
|
||||||
update_attribute,
|
attribute,
|
||||||
new_value,
|
new_value,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
@ -47,6 +54,6 @@ def update(
|
||||||
def delete(project_id: int):
|
def delete(project_id: int):
|
||||||
return delete_from_database_where(
|
return delete_from_database_where(
|
||||||
Project,
|
Project,
|
||||||
Project.id,
|
Project.id, # type: ignore (Type correct on runtime)
|
||||||
project_id,
|
project_id,
|
||||||
)
|
)
|
||||||
|
|
|
||||||
|
|
@ -1,3 +1,5 @@
|
||||||
|
from sqlmodel import inspect
|
||||||
|
|
||||||
from app.database import (
|
from app.database import (
|
||||||
add_to_database,
|
add_to_database,
|
||||||
delete_from_database_where,
|
delete_from_database_where,
|
||||||
|
|
@ -26,11 +28,16 @@ def update(
|
||||||
update_attribute: str,
|
update_attribute: str,
|
||||||
new_value: int | str | None,
|
new_value: int | str | None,
|
||||||
):
|
):
|
||||||
|
if update_attribute not in inspect(Service).attrs:
|
||||||
|
raise ValueError(f"Invalid attribute: {update_attribute}")
|
||||||
|
|
||||||
|
attribute = getattr(Service, update_attribute)
|
||||||
|
|
||||||
return update_database_where(
|
return update_database_where(
|
||||||
Service,
|
Service,
|
||||||
Service.id,
|
Service.id, # type: ignore (Type correct on runtime)
|
||||||
service_id,
|
service_id,
|
||||||
update_attribute,
|
attribute,
|
||||||
new_value,
|
new_value,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
@ -38,6 +45,6 @@ def update(
|
||||||
def delete(service_id: int):
|
def delete(service_id: int):
|
||||||
return delete_from_database_where(
|
return delete_from_database_where(
|
||||||
Service,
|
Service,
|
||||||
Service.id,
|
Service.id, # type: ignore (Type correct on runtime)
|
||||||
service_id,
|
service_id,
|
||||||
)
|
)
|
||||||
|
|
|
||||||
|
|
@ -27,9 +27,14 @@ def get_password_hash(password: str):
|
||||||
|
|
||||||
|
|
||||||
def get_user(username: str):
|
def get_user(username: str):
|
||||||
user_dict = get_from_database_where(User, User.username, username, True)
|
user = get_from_database_where(
|
||||||
if user_dict is not None:
|
User,
|
||||||
return UserInDB(**user_dict)
|
User.username, # type: ignore (Type correct on runtime)
|
||||||
|
username,
|
||||||
|
True,
|
||||||
|
)
|
||||||
|
if user is not None:
|
||||||
|
return UserInDB.model_validate(user[0])
|
||||||
|
|
||||||
|
|
||||||
def authenticate_user(username: str, password: str):
|
def authenticate_user(username: str, password: str):
|
||||||
|
|
@ -43,7 +48,7 @@ def authenticate_user(username: str, password: str):
|
||||||
|
|
||||||
|
|
||||||
def create_access_token(
|
def create_access_token(
|
||||||
data: dict[str, str],
|
data: dict[str, str | datetime],
|
||||||
expires_delta: timedelta | None = None,
|
expires_delta: timedelta | None = None,
|
||||||
):
|
):
|
||||||
to_encode = data.copy()
|
to_encode = data.copy()
|
||||||
|
|
@ -52,7 +57,7 @@ def create_access_token(
|
||||||
else:
|
else:
|
||||||
expire = datetime.now(UTC) + timedelta(minutes=15)
|
expire = datetime.now(UTC) + timedelta(minutes=15)
|
||||||
to_encode.update({"exp": expire})
|
to_encode.update({"exp": expire})
|
||||||
encoded_jwt = jwt.encode(
|
encoded_jwt = jwt.encode( # pyright: ignore[reportUnknownMemberType] (Third party skill issue)
|
||||||
to_encode,
|
to_encode,
|
||||||
Settings.secret_key,
|
Settings.secret_key,
|
||||||
algorithm=Settings.algorithm,
|
algorithm=Settings.algorithm,
|
||||||
|
|
@ -67,7 +72,7 @@ async def get_current_user(token: Annotated[str, Depends(oauth2_scheme)]):
|
||||||
headers={"WWW-Authenticate": "Bearer"},
|
headers={"WWW-Authenticate": "Bearer"},
|
||||||
)
|
)
|
||||||
try:
|
try:
|
||||||
payload = jwt.decode(
|
payload = jwt.decode( # pyright: ignore[reportUnknownMemberType] (Third party skill issue)
|
||||||
token,
|
token,
|
||||||
Settings.secret_key,
|
Settings.secret_key,
|
||||||
algorithms=[Settings.algorithm],
|
algorithms=[Settings.algorithm],
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue