add UserStorage

resolvers/auth.py

@@ -40,12 +40,14 @@ async def register(*_, email: str, password: str = ""):
 	if not password:
 		user = User.create(**user_dict)
 		await send_confirm_email(user)
+		UserStorage.add_user(user)
 		return { "user": user }
-	else:
-		user_dict["password"] = Password.encode(password)
-		user = User.create(**user_dict)
-		token = await Authorize.authorize(user)
-		return {"user": user, "token": token }
+
+	user_dict["password"] = Password.encode(password)
+	user = User.create(**user_dict)
+	token = await Authorize.authorize(user)
+	UserStorage.add_user(user)
+	return {"user": user, "token": token }
 
 
 @query.field("signIn")

resolvers/profile.py

@@ -22,12 +22,7 @@ async def get_user_by_slug(_, info, slug):
 @query.field("getCurrentUser")
 @login_required
 async def get_current_user(_, info):
-	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 = info.context["request"].user
 	return { "user": user }
 
 @query.field("userRoles")