91 lines
3.2 KiB
Python
91 lines
3.2 KiB
Python
from typing import Optional
|
|
import strawberry
|
|
from graphql import GraphQLError
|
|
from passlib.hash import bcrypt
|
|
from sqlalchemy import select
|
|
from strawberry.file_uploads import Upload
|
|
from database import models
|
|
from database.models import User
|
|
from graphql_schema.sqlalchemy_to_strawberry_type import strawberry_sqlalchemy_type, strawberry_sqlalchemy_input
|
|
from upload_utils import handle_file_upload, delete_file
|
|
|
|
|
|
@strawberry_sqlalchemy_type(User, exclude_fields=['password_hashed'])
|
|
class User:
|
|
|
|
async def load_avatar_image_url(root):
|
|
if not root.avatar_image_filename:
|
|
return None
|
|
|
|
return f"http://localhost:8000/uploads/profile/{root.id}/{root.avatar_image_filename}"
|
|
|
|
async def load_title_image_url(root):
|
|
if not root.title_image_filename:
|
|
return "http://localhost:8000/static/default-title-image.jpg"
|
|
|
|
return f"http://localhost:8000/uploads/profile/{root.id}/{root.title_image_filename}"
|
|
|
|
avatar_image_url: Optional[str] = strawberry.field(resolver=load_avatar_image_url)
|
|
title_image_url: str = strawberry.field(resolver=load_title_image_url)
|
|
|
|
|
|
@strawberry.type
|
|
class UserQueries:
|
|
@strawberry.field
|
|
async def user(root, info, username: str) -> User:
|
|
print("username")
|
|
if len(username) == 0:
|
|
raise GraphQLError("Username not set!")
|
|
|
|
return (await info.context.db.scalars(
|
|
select(models.User).filter_by(public_username=username)
|
|
)).one()
|
|
|
|
@strawberry.field
|
|
async def logged_user(root, info) -> User:
|
|
return (await info.context.db.scalars(
|
|
select(models.User).filter_by(id=info.context.user_id)
|
|
)).one()
|
|
|
|
|
|
@strawberry.type
|
|
class EditUserMutation:
|
|
@strawberry_sqlalchemy_input(
|
|
models.User,
|
|
exclude_fields=['id', 'email', 'avatar_image_filename', 'password_hashed', 'title_image_filename'],
|
|
all_optional=True
|
|
)
|
|
class EditUserInput:
|
|
old_password: Optional[str] = None
|
|
new_password: Optional[str] = None
|
|
avatar_image: Optional[Upload] = None
|
|
title_image: Optional[Upload] = None
|
|
|
|
@strawberry.mutation
|
|
async def edit_logged_user(root, info, input: EditUserInput) -> User:
|
|
user = (await info.context.db.scalars(
|
|
select(models.User).filter_by(id=info.context.user_id)
|
|
)).one()
|
|
|
|
user_image_path = f"/app/uploads/profile/{user.id}"
|
|
data = input.to_dict()
|
|
if input.avatar_image:
|
|
if user.avatar_image_filename:
|
|
delete_file(f"{user_image_path}/{user.avatar_image_filename}", silent=True)
|
|
|
|
data['avatar_image_filename'] = await handle_file_upload(input.avatar_image, user_image_path)
|
|
|
|
if input.title_image:
|
|
if user.title_image_filename:
|
|
delete_file(f"{user_image_path}/{user.title_image_filename}", silent=True)
|
|
|
|
data['title_image_filename'] = await handle_file_upload(input.title_image, user_image_path)
|
|
|
|
if input.old_password and input.new_password:
|
|
if not bcrypt.verify(input.old_password, user.password_hashed):
|
|
raise GraphQLError("Bad password")
|
|
|
|
data['password_hashed'] = bcrypt.hash(input.new_password)
|
|
|
|
return await models.User.update(info.context.db, obj=user, data=data)
|