Moved to ariadne

kdb-bot/src/bot_data/abc/query_abc.py

@@ -1,6 +1,7 @@
+from ariadne_graphql_modules import ObjectType
 from cpl_core.dependency_injection import ServiceProviderABC
+from cpl_discord.service import DiscordBotServiceABC
 from cpl_query.extension import List
-from graphene import ObjectType
 
 from bot_data.abc.auto_role_repository_abc import AutoRoleRepositoryABC
 from bot_data.abc.client_repository_abc import ClientRepositoryABC
@@ -18,9 +19,12 @@ from bot_data.model.server import Server
 from bot_data.model.user import User
 from bot_data.model.user_joined_server import UserJoinedServer
 from bot_data.model.user_joined_voice_channel import UserJoinedVoiceChannel
+from modules.level.service.level_service import LevelService
 
 
 class QueryABC(ObjectType):
+    __abstract__ = True
+
     _auto_roles: List[AutoRole]
     _clients: List[Client]
     _known_users: List[KnownUser]
@@ -30,6 +34,9 @@ class QueryABC(ObjectType):
     _user_joined_voice_channel: List[UserJoinedVoiceChannel]
     _users: List[User]
 
+    _bot: DiscordBotServiceABC
+    _level_service: LevelService
+
     @classmethod
     @ServiceProviderABC.inject
     def init(
@@ -42,6 +49,8 @@ class QueryABC(ObjectType):
             user_joined_servers: UserJoinedServerRepositoryABC,
             user_joined_voice_channel: UserJoinedVoiceChannelRepositoryABC,
             users: UserRepositoryABC,
+            bot: DiscordBotServiceABC,
+            level_service: LevelService,
     ):
         cls._auto_roles = auto_roles.get_auto_roles()
         cls._clients = clients.get_clients()
@@ -51,3 +60,5 @@ class QueryABC(ObjectType):
         cls._user_joined_servers = user_joined_servers.get_user_joined_servers()
         cls._user_joined_voice_channel = user_joined_voice_channel.get_user_joined_voice_channels()
         cls._users = users.get_users()
+        cls._bot = bot
+        cls._level_service = level_service

kdb-bot/src/bot_data/graphql/graphql.py

@@ -1,16 +0,0 @@
-import graphene
-from graphene import Schema
-
-from bot_data.abc.queryABC import QueryABC
-from bot_data.graphql.root_query import RootQuery
-
-
-class GraphQL:
-
-    def __init__(self):
-        QueryABC.init()
-        self._schema = graphene.Schema(query=RootQuery)
-
-    @property
-    def schema(self) -> Schema:
-        return self._schema

kdb-bot/src/bot_data/graphql/model.gql

@@ -0,0 +1,64 @@
+type Query {
+    servers: [Server]
+    known_users: [User]
+}
+
+type Server {
+    id: ID
+    discord_id: String
+    clients: [Client]
+    members: [User]
+    level: [Level]
+}
+
+type Client {
+    id: ID
+    discord_id: String
+    sent_message_count: Int
+    received_message_count: Int
+    deleted_message_count: Int
+    received_command_count: Int
+    moved_users_count: Int
+
+    server: Server
+}
+
+type User {
+    id: ID
+    discord_id: String
+    name: String
+    xp: Int
+    ontime: Int
+    level: Level
+
+    joined_servers: [UserJoinedServer]
+    joined_voice_channel: [UserJoinedVoiceChannel]
+
+    server: Server
+}
+
+type UserJoinedServer {
+    id: ID
+    user: User
+    server: Server
+    joined_on: String
+    leaved_on: String
+}
+
+type UserJoinedVoiceChannel {
+    id: ID
+    channel_id: String
+    user: User
+    joined_on: String
+    leaved_on: String
+}
+
+type Level {
+    id: ID
+    name: String
+    color: String
+    min_xp: Int
+    permissions: String
+
+    server: Server
+}

kdb-bot/src/bot_data/graphql/queries/client_query.py

@@ -1,12 +0,0 @@
-import graphene
-
-from bot_data.abc.queryABC import QueryABC
-from bot_data.graphql.types.client_query_type import ClientQueryType
-
-
-class ClientQuery(QueryABC):
-    clients = graphene.List(ClientQueryType)
-
-    @classmethod
-    def resolve_clients(cls, root, info):
-        return cls._clients

kdb-bot/src/bot_data/graphql/queries/server_query.py

@@ -1,19 +0,0 @@
-import graphene
-from cpl_core.dependency_injection import ServiceProviderABC
-
-from bot_data.abc.queryABC import QueryABC
-from bot_data.graphql.types.server_query_type import ServerQueryType
-from bot_data.service.server_repository_service import ServerRepositoryService
-
-
-class ServerQuery(QueryABC):
-    servers = graphene.List(ServerQueryType)
-
-    @staticmethod
-    @ServiceProviderABC.inject
-    def _get_servers(servers: ServerRepositoryService):
-        return servers.get_servers()
-
-    @classmethod
-    def resolve_servers(cls, root, info):
-        return cls._servers

kdb-bot/src/bot_data/graphql/query.py

@@ -0,0 +1,17 @@
+from ariadne import gql
+
+from bot_data.abc.query_abc import QueryABC
+from bot_data.graphql.types.server_query_type import ServerQueryType
+
+
+class Query(QueryABC):
+    __schema__ = gql("""
+        type Query {
+            servers: [Server]
+        }
+    """)
+    __requires__ = [ServerQueryType]
+
+    @classmethod
+    def resolve_servers(cls, *_):
+        return cls._servers

kdb-bot/src/bot_data/graphql/root_query.py

@@ -1,10 +0,0 @@
-from graphene import ObjectType, relay
-
-from bot_data.graphql.queries.server_query import ServerQuery
-
-
-class RootQuery(
-    ServerQuery,
-    ObjectType
-):
-    node = relay.Node.Field()

kdb-bot/src/bot_data/graphql/types/client_query_type.py

@@ -1,18 +1,22 @@
-from graphene import Int, String
+from ariadne import gql
 
-from bot_data.abc.queryABC import QueryABC
+from bot_data.abc.query_abc import QueryABC
 from bot_data.model.client import Client
 
 
 class ClientQueryType(QueryABC):
-    id = Int()
-    discord_id = String()
-    sent_message_count = Int()
-    received_message_count = Int()
-    deleted_message_count = Int()
-    received_command_count = Int()
-    moved_users_count = Int()
+    __schema__ = gql("""
+        type Client {
+            id: ID
+            discord_id: String
+            sent_message_count: Int
+            received_message_count: Int
+            deleted_message_count: Int
+            received_command_count: Int
+            moved_users_count: Int
+        }
+    """)
 
-    @staticmethod
-    def resolve_id(client: Client, info):
+    @classmethod
+    def resolve_id(cls, client: Client, *_):
         return client.client_id

kdb-bot/src/bot_data/graphql/types/level_query_type.py

@@ -0,0 +1,21 @@
+from ariadne import gql
+
+from bot_data.abc.query_abc import QueryABC
+from bot_data.model.client import Client
+from bot_data.model.level import Level
+
+
+class LevelQueryType(QueryABC):
+    __schema__ = gql("""
+        type Level {
+            id: ID
+            name: String
+            color: String
+            min_xp: Int
+            permissions: String
+        }
+    """)
+
+    # @classmethod
+    # def resolve_id(cls, level: Level, *_):
+    #     return level.id

kdb-bot/src/bot_data/graphql/types/server_query_type.py

@@ -1,23 +1,40 @@
-from graphene import Int, DateTime, String, List
+from ariadne import gql
 
-from bot_data.abc.queryABC import QueryABC
+from bot_data.abc.query_abc import QueryABC
 from bot_data.graphql.types.client_query_type import ClientQueryType
+from bot_data.graphql.types.level_query_type import LevelQueryType
+from bot_data.graphql.types.user_query_type import UserQueryType
 from bot_data.model.server import Server
 
 
 class ServerQueryType(QueryABC):
-    id = Int()
-    discord_server_id = String()
+    __schema__ = gql("""
+        type Server {
+            id: ID
+            discord_id: String
+            clients: [Client]
+            users: [User]
+            levels: [Level]
+        }
+    """)
+    __requires__ = [ClientQueryType, UserQueryType, LevelQueryType]
 
-    created_at = DateTime()
-    modified_at = DateTime()
-
-    clients = List(ClientQueryType)
-
-    @staticmethod
-    def resolve_id(server: Server, info):
+    @classmethod
+    def resolve_id(cls, server: Server, *_):
         return server.server_id
 
     @classmethod
-    def resolve_clients(cls, server: Server, info):
+    def resolve_discord_id(cls, server: Server, *_):
+        return server.discord_server_id
+
+    @classmethod
+    def resolve_clients(cls, server: Server, *_):
         return cls._clients.where(lambda c: c.server.server_id == server.server_id)
+
+    @classmethod
+    def resolve_users(cls, server: Server, *_):
+        return cls._users.where(lambda u: u.server.server_id == server.server_id)
+
+    @classmethod
+    def resolve_levels(cls, server: Server, *_):
+        return cls._levels.where(lambda l: l.server.server_id == server.server_id)

kdb-bot/src/bot_data/graphql/types/user_query_type.py

@@ -0,0 +1,39 @@
+from ariadne import gql
+
+from bot_data.abc.query_abc import QueryABC
+from bot_data.graphql.types.level_query_type import LevelQueryType
+from bot_data.model.user import User
+
+
+class UserQueryType(QueryABC):
+    __schema__ = gql("""
+        type User {
+            id: ID
+            discord_id: String
+            name: String
+            xp: Int
+            ontime: Int
+            level: Level
+        }
+    """)
+    __requires__ = [LevelQueryType]
+
+    @classmethod
+    def resolve_id(cls, user: User, *_):
+        return user.user_id
+
+    @classmethod
+    def resolve_name(cls, user: User, *_):
+        return cls._bot.get_user(user.discord_id).name
+
+    @classmethod
+    def resolve_ontime(cls, user: User, *_):
+        return cls._user_joined_voice_channel.where(lambda ujvc: ujvc.user.user_id == user.user_id) \
+            .where(lambda x: x.leaved_on is not None and x.joined_on is not None) \
+            .sum(lambda join: round((join.leaved_on - join.joined_on).total_seconds() / 3600, 2))
+
+    @classmethod
+    def resolve_level(cls, user: User, *_):
+        levels = cls._levels.where(lambda l: l.server.server_id == user.server.server_id).order_by(lambda l: l.min_xp)
+        return levels.where(lambda l: user.xp >= l.min_xp).last_or_default()
+