wip: redis, sqlalchemy, structured, etc

resolvers/__init__.py

@@ -0,0 +1,3 @@
+from resolvers.login import get_user, login, logout, register
+
+__all__ = ["get_user", "login", "logout", "register"]

resolvers/auth.py

@@ -0,0 +1,46 @@
+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("SignUp")
+async def register(*_, create: dict = None) -> User:
+    create_user = CreateUser(**create)
+    create_user.password = Password.encode(create_user.password)
+    return User.create(**create_user.dict())
+
+
+@query.field("SignIn")
+async def login(_, info: GraphQLResolveInfo, id: int, password: str) -> 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)
+    return await Authorize.authorize(user, device=device, auto_delete=auto_delete)
+
+
+# TODO: implement some queries, ex. @query.field("isUsernameFree")
+
+@query.field("logout")
+@login_required
+async def logout(_, info: GraphQLResolveInfo, id: int) -> bool:
+    token = info.context["request"].headers[JWT_AUTH_HEADER]
+    return await Authorize.revoke(token)
+
+
+@query.field("getUser")
+@login_required
+async def get_user(*_, id: int):
+    return global_session.query(User).filter(User.id == id).first()
+
+

resolvers/base.py

@@ -0,0 +1,6 @@
+from ariadne import MutationType, QueryType
+
+query = QueryType()
+mutation = MutationType()
+
+resolvers = [query, mutation]

resolvers/editor.py

@@ -0,0 +1 @@
+# TODO: implement me

resolvers/inbox.py

@@ -0,0 +1,27 @@
+from orm import message, user
+
+from ariadne import ObjectType, convert_kwargs_to_snake_case
+
+query = ObjectType("Query")
+
+
+@query.field("messages")
+@convert_kwargs_to_snake_case
+async def resolve_messages(obj, info, user_id):
+    def filter_by_userid(message):
+        return message["sender_id"] == user_id or \
+            message["recipient_id"] == user_id
+
+    user_messages = filter(filter_by_userid, messages)
+    return {
+        "success": True,
+        "messages": user_messages
+    }
+
+
+@query.field("userId")
+@convert_kwargs_to_snake_case
+async def resolve_user_id(obj, info, username):
+    user = users.get(username)
+    if user:
+        return user["user_id"]