query userRoles

resolvers/profile.py

@@ -1,7 +1,9 @@
-from orm import User
+from orm import User, UserRole, Role
 from orm.base import local_session
 from resolvers.base import mutation, query, subscription
 from auth.authenticate import login_required
+
+from sqlalchemy.orm import selectinload
 import asyncio
 
 @query.field("getUserBySlug") # get a public profile
@@ -19,3 +21,17 @@ async def get_current_user(_, info):
 	with local_session() as session:
 		user = session.query(User).filter(User.id == user_id).first()
 	return { "user": user }
+
+@query.field("userRoles")
+@login_required
+async def user_roles(_, info):
+	auth = info.context["request"].auth
+	user_id = auth.user_id
+
+	with local_session() as session:
+		roles = session.query(Role).\
+			options(selectinload(Role.permissions)).\
+			join(UserRole).\
+			where(UserRole.user_id == user_id).all()
+
+	return roles