Added base for server statistics #420
This commit is contained in:
parent
6ae7ab5c5a
commit
f7f25e9428
@ -38,7 +38,7 @@ class QueryABC(ObjectType):
|
|||||||
def __init__(self, name: str):
|
def __init__(self, name: str):
|
||||||
ObjectType.__init__(self, name)
|
ObjectType.__init__(self, name)
|
||||||
|
|
||||||
def add_collection(self, name: str, get_collection: Callable, filter_type: type = None):
|
def _get_wrapper(self, get_collection: Callable, filter_type: type = None):
|
||||||
def wrapper(*args, **kwargs):
|
def wrapper(*args, **kwargs):
|
||||||
if filter_type is not None and "filter" in kwargs:
|
if filter_type is not None and "filter" in kwargs:
|
||||||
kwargs["filter"] = FilterABC.get_collection_filter(filter_type, kwargs["filter"])
|
kwargs["filter"] = FilterABC.get_collection_filter(filter_type, kwargs["filter"])
|
||||||
@ -67,8 +67,14 @@ class QueryABC(ObjectType):
|
|||||||
|
|
||||||
return self._resolve_collection(collection, *args, **kwargs)
|
return self._resolve_collection(collection, *args, **kwargs)
|
||||||
|
|
||||||
self.set_field(f"{name}s", wrapper)
|
return wrapper
|
||||||
self.set_field(f"{name}Count", lambda *args: wrapper(*args).count())
|
|
||||||
|
def add_field(self, name: str, get_collection: Callable, filter_type: type = None):
|
||||||
|
self.set_field(name, self._get_wrapper(get_collection, filter_type))
|
||||||
|
|
||||||
|
def add_collection(self, name: str, get_collection: Callable, filter_type: type = None):
|
||||||
|
self.add_field(f"{name}s", get_collection, filter_type)
|
||||||
|
self.set_field(f"{name}Count", lambda *args: self._get_wrapper(get_collection, filter_type)(*args).count())
|
||||||
|
|
||||||
def _can_user_see_element(self, user: AuthUser, element: T) -> bool:
|
def _can_user_see_element(self, user: AuthUser, element: T) -> bool:
|
||||||
@ServiceProviderABC.inject
|
@ServiceProviderABC.inject
|
||||||
|
@ -38,6 +38,8 @@ type Server implements TableWithHistoryQuery {
|
|||||||
config: ServerConfig
|
config: ServerConfig
|
||||||
hasFeatureFlag(flag: String): FeatureFlag
|
hasFeatureFlag(flag: String): FeatureFlag
|
||||||
|
|
||||||
|
statistic(date: String): ServerStatistic
|
||||||
|
|
||||||
createdAt: String
|
createdAt: String
|
||||||
modifiedAt: String
|
modifiedAt: String
|
||||||
|
|
||||||
@ -59,4 +61,12 @@ input ServerFilter {
|
|||||||
id: ID
|
id: ID
|
||||||
discordId: String
|
discordId: String
|
||||||
name: String
|
name: String
|
||||||
|
}
|
||||||
|
|
||||||
|
type ServerStatistic {
|
||||||
|
userJoinedVoiceChannelCount: Int
|
||||||
|
userJoinedVoiceChannelOntime: Int
|
||||||
|
userJoinedGameServerCount: Int
|
||||||
|
userJoinedGameServerOntime: Int
|
||||||
|
messageCount: Int
|
||||||
}
|
}
|
@ -57,6 +57,7 @@ from bot_graphql.queries.level_query import LevelQuery
|
|||||||
from bot_graphql.queries.server_config_query import ServerConfigQuery
|
from bot_graphql.queries.server_config_query import ServerConfigQuery
|
||||||
from bot_graphql.queries.server_history_query import ServerHistoryQuery
|
from bot_graphql.queries.server_history_query import ServerHistoryQuery
|
||||||
from bot_graphql.queries.server_query import ServerQuery
|
from bot_graphql.queries.server_query import ServerQuery
|
||||||
|
from bot_graphql.queries.server_statistic_query import ServerStatisticQuery
|
||||||
from bot_graphql.queries.short_role_name_history_query import ShortRoleNameHistoryQuery
|
from bot_graphql.queries.short_role_name_history_query import ShortRoleNameHistoryQuery
|
||||||
from bot_graphql.queries.short_role_name_query import ShortRoleNameQuery
|
from bot_graphql.queries.short_role_name_query import ShortRoleNameQuery
|
||||||
from bot_graphql.queries.technician_config_history_query import (
|
from bot_graphql.queries.technician_config_history_query import (
|
||||||
@ -138,6 +139,7 @@ class GraphQLModule(ModuleABC):
|
|||||||
services.add_transient(QueryABC, ShortRoleNameQuery)
|
services.add_transient(QueryABC, ShortRoleNameQuery)
|
||||||
services.add_transient(QueryABC, UserWarningHistoryQuery)
|
services.add_transient(QueryABC, UserWarningHistoryQuery)
|
||||||
services.add_transient(QueryABC, UserWarningQuery)
|
services.add_transient(QueryABC, UserWarningQuery)
|
||||||
|
services.add_transient(QueryABC, ServerStatisticQuery)
|
||||||
|
|
||||||
services.add_transient(QueryABC, DiscordQuery)
|
services.add_transient(QueryABC, DiscordQuery)
|
||||||
services.add_transient(QueryABC, GuildQuery)
|
services.add_transient(QueryABC, GuildQuery)
|
||||||
|
@ -107,6 +107,10 @@ class ServerQuery(DataQueryWithHistoryABC):
|
|||||||
"hasFeatureFlag",
|
"hasFeatureFlag",
|
||||||
lambda server, *_, **kwargs: self._resolve_has_feature_flag(server, *_, **kwargs),
|
lambda server, *_, **kwargs: self._resolve_has_feature_flag(server, *_, **kwargs),
|
||||||
)
|
)
|
||||||
|
self.set_field(
|
||||||
|
"statistic",
|
||||||
|
lambda server, *_, **kwargs: server,
|
||||||
|
)
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def resolve_id(server: Server, *_):
|
def resolve_id(server: Server, *_):
|
||||||
|
77
bot/src/bot_graphql/queries/server_statistic_query.py
Normal file
77
bot/src/bot_graphql/queries/server_statistic_query.py
Normal file
@ -0,0 +1,77 @@
|
|||||||
|
import datetime
|
||||||
|
|
||||||
|
from cpl_core.configuration import ConfigurationABC
|
||||||
|
|
||||||
|
from bot_data.abc.user_joined_game_server_repository_abc import UserJoinedGameServerRepositoryABC
|
||||||
|
from bot_data.abc.user_joined_voice_channel_repository_abc import UserJoinedVoiceChannelRepositoryABC
|
||||||
|
from bot_data.abc.user_message_count_per_hour_repository_abc import UserMessageCountPerHourRepositoryABC
|
||||||
|
from bot_data.model.server_config import ServerConfig
|
||||||
|
from bot_graphql.abc.query_abc import QueryABC
|
||||||
|
|
||||||
|
|
||||||
|
class ServerStatisticQuery(QueryABC):
|
||||||
|
def __init__(
|
||||||
|
self,
|
||||||
|
config: ConfigurationABC,
|
||||||
|
user_joined_voice_channels: UserJoinedVoiceChannelRepositoryABC,
|
||||||
|
user_joined_game_servers: UserJoinedGameServerRepositoryABC,
|
||||||
|
user_messages: UserMessageCountPerHourRepositoryABC,
|
||||||
|
):
|
||||||
|
QueryABC.__init__(self, "ServerStatistic")
|
||||||
|
|
||||||
|
self._config = config
|
||||||
|
|
||||||
|
self._user_joined_voice_channels = user_joined_voice_channels
|
||||||
|
self._user_joined_game_servers = user_joined_game_servers
|
||||||
|
self._user_messages = user_messages
|
||||||
|
|
||||||
|
self.add_field("userJoinedVoiceChannelCount", self._resolve_voice_channel_count)
|
||||||
|
self.add_field("userJoinedVoiceChannelOntime", self._resolve_voice_channel_ontime)
|
||||||
|
self.add_field("userJoinedGameServerCount", self._resolve_game_server_count)
|
||||||
|
self.add_field("userJoinedGameServerOntime", self._resolve_game_server_ontime)
|
||||||
|
self.add_field(
|
||||||
|
"messageCount",
|
||||||
|
self._resolve_message_count,
|
||||||
|
)
|
||||||
|
|
||||||
|
def _resolve_voice_channel_count(self, server, *_, **kwargs):
|
||||||
|
return (
|
||||||
|
self._user_joined_voice_channels.get_user_joined_voice_channels()
|
||||||
|
.where(lambda x: x.user.server.id == server.id and x.created_at.date() >= self._get_date(**kwargs))
|
||||||
|
.count()
|
||||||
|
)
|
||||||
|
|
||||||
|
def _resolve_voice_channel_ontime(self, server, *_, **kwargs):
|
||||||
|
return (
|
||||||
|
self._user_joined_voice_channels.get_user_joined_voice_channels()
|
||||||
|
.where(lambda x: x.user.server.id == server.id and x.created_at.date() >= self._get_date(**kwargs))
|
||||||
|
.sum(lambda x: x.time)
|
||||||
|
)
|
||||||
|
|
||||||
|
def _resolve_game_server_count(self, server, *_, **kwargs):
|
||||||
|
return (
|
||||||
|
self._user_joined_game_servers.get_user_joined_game_servers()
|
||||||
|
.where(lambda x: x.user.server.id == server.id and x.created_at.date() >= self._get_date(**kwargs))
|
||||||
|
.count()
|
||||||
|
)
|
||||||
|
|
||||||
|
def _resolve_game_server_ontime(self, server, *_, **kwargs):
|
||||||
|
return (
|
||||||
|
self._user_joined_game_servers.get_user_joined_game_servers()
|
||||||
|
.where(lambda x: x.user.server.id == server.id and x.created_at.date() >= self._get_date(**kwargs))
|
||||||
|
.sum(lambda x: x.time)
|
||||||
|
)
|
||||||
|
|
||||||
|
def _resolve_message_count(self, server, *_, **kwargs):
|
||||||
|
settings: ServerConfig = self._config.get_configuration(f"ServerConfig_{server.discord_id}")
|
||||||
|
return (
|
||||||
|
self._user_messages.get_user_message_count_per_hours()
|
||||||
|
.where(lambda x: x.user.server.id == server.id and x.created_at.date() >= self._get_date(**kwargs))
|
||||||
|
.sum(lambda x: x.xp_count / settings.xp_per_message)
|
||||||
|
)
|
||||||
|
|
||||||
|
def _get_date(self, **kwargs) -> datetime.date:
|
||||||
|
if "date" not in kwargs:
|
||||||
|
return datetime.date.today() - datetime.timedelta(days=7)
|
||||||
|
|
||||||
|
return datetime.datetime.strptime(kwargs["date"], "%d.%m.%Y").date()
|
Loading…
Reference in New Issue
Block a user