core/resolvers/auth.py

from graphql import GraphQLResolveInfo

from auth.authenticate import login_required
from auth.authorize import Authorize
from auth.identity import Identity
from auth.password import Password
from auth.validations import CreateUser
from orm import User
from orm.base import global_session
from resolvers.base import mutation, query

from settings import JWT_AUTH_HEADER

@mutation.field("registerUser")
async def register(*_, input: dict = None) -> User:
    create_user = CreateUser(**input)
    create_user.password = Password.encode(create_user.password)
    return User.create(**create_user.dict())


@query.field("signIn")
async def sign_in(_, info: GraphQLResolveInfo, id: int, password: str):
    try:
        device = info.context["request"].headers['device']
    except KeyError:
        device = "pc"
    auto_delete = False if device == "mobile" else True
    user = Identity.identity(user_id=id, password=password)
    token = await Authorize.authorize(user, device=device, auto_delete=auto_delete)
    return {"status" : True, "token" : token}


@query.field("signOut")
@login_required
async def sign_out(_, info: GraphQLResolveInfo):
    token = info.context["request"].headers[JWT_AUTH_HEADER]
    status = await Authorize.revoke(token)
    return {"status" : status}


#@query.field("getUser")
#@login_required
async def get_user(*_, id: int):
    return global_session.query(User).filter(User.id == id).first()