core/resolvers/profile.py

from orm import User, UserRole, Role, UserRating
from orm.user import AuthorSubscription
from orm.base import local_session
from resolvers.base import mutation, query, subscription
from auth.authenticate import login_required

from sqlalchemy import func, and_
from sqlalchemy.orm import selectinload
import asyncio

@query.field("getCurrentUser")
@login_required
async def get_current_user(_, info):
	user = info.context["request"].user
	return { "user": user }

@query.field("getUsersBySlugs")
async def get_users_by_slugs(_, info, slugs):
	with local_session() as session:
		users = session.query(User).\
			options(selectinload(User.ratings)).\
			filter(User.slug.in_(slugs)).all()
	return users

@query.field("getUserRoles")
async def get_user_roles(_, info, slug):

	with local_session() as session:
		user = session.query(User).where(User.slug == slug).first()

		roles = session.query(Role).\
			options(selectinload(Role.permissions)).\
			join(UserRole).\
			where(UserRole.user_id == user.id).all()

	return roles

@mutation.field("updateProfile")
@login_required
async def update_profile(_, info, profile):
	auth = info.context["request"].auth
	user_id = auth.user_id

	with local_session() as session:
		user = session.query(User).filter(User.id == user_id).first()
		user.update(profile)
		session.commit()

	return {}

@mutation.field("authorSubscribe")
@login_required
async def author_subscribe(_, info, slug):
	user = info.context["request"].user

	AuthorSubscription.create(
		subscriber = user.slug, 
		author = slug
	)

	return {}

@mutation.field("authorUnsubscribe")
@login_required
async def author_unsubscribe(_, info, slug):
	user = info.context["request"].user

	with local_session() as session:
		sub = session.query(AuthorSubscription).\
			filter(and_(AuthorSubscription.subscriber == user.slug, AuthorSubscription.author == slug)).\
			first()
		if not sub:
			return { "error" : "subscription not exist" }
		session.delete(sub)
		session.commit()

	return {}