dev #422
@ -38,7 +38,7 @@ class QueryABC(ObjectType):
|
||||
def __init__(self, name: str):
|
||||
ObjectType.__init__(self, name)
|
||||
|
||||
def _get_wrapper(self, get_collection: Callable, filter_type: type = None):
|
||||
def add_collection(self, name: str, get_collection: Callable, filter_type: type = None):
|
||||
def wrapper(*args, **kwargs):
|
||||
if filter_type is not None and "filter" in kwargs:
|
||||
kwargs["filter"] = FilterABC.get_collection_filter(filter_type, kwargs["filter"])
|
||||
@ -67,14 +67,8 @@ class QueryABC(ObjectType):
|
||||
|
||||
return self._resolve_collection(collection, *args, **kwargs)
|
||||
|
||||
return wrapper
|
||||
|
||||
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())
|
||||
self.set_field(f"{name}s", wrapper)
|
||||
self.set_field(f"{name}Count", lambda *args: wrapper(*args).count())
|
||||
|
||||
def _can_user_see_element(self, user: AuthUser, element: T) -> bool:
|
||||
@ServiceProviderABC.inject
|
||||
|
@ -61,12 +61,4 @@ input ServerFilter {
|
||||
id: ID
|
||||
discordId: String
|
||||
name: String
|
||||
}
|
||||
|
||||
type ServerStatistic {
|
||||
userJoinedVoiceChannelCount: Int
|
||||
userJoinedVoiceChannelOntime: Int
|
||||
userJoinedGameServerCount: Int
|
||||
userJoinedGameServerOntime: Int
|
||||
messageCount: Int
|
||||
}
|
17
bot/src/bot_graphql/graphql/serverStatistic.gql
Normal file
17
bot/src/bot_graphql/graphql/serverStatistic.gql
Normal file
@ -0,0 +1,17 @@
|
||||
type ServerStatistic {
|
||||
achievementsAchieved: Int
|
||||
messageCount: Int
|
||||
|
||||
userCount: Int
|
||||
activeUserCount: Int
|
||||
|
||||
userJoinedVoiceChannelCount: Int
|
||||
userJoinedVoiceChannelOntime: Int
|
||||
|
||||
userJoinedGameServerCount: Int
|
||||
userJoinedGameServerOntime: Int
|
||||
|
||||
userWarningCount: Int
|
||||
|
||||
activityScore: Int
|
||||
}
|
7
bot/src/bot_graphql/model/server_statistics.py
Normal file
7
bot/src/bot_graphql/model/server_statistics.py
Normal file
@ -0,0 +1,7 @@
|
||||
from bot_data.model.server import Server
|
||||
|
||||
|
||||
class ServerStatistics:
|
||||
def __init__(self, server: Server, kwargs: dict):
|
||||
self.server = server
|
||||
self.kwargs = kwargs
|
@ -26,6 +26,7 @@ from bot_graphql.filter.client_filter import ClientFilter
|
||||
from bot_graphql.filter.level_filter import LevelFilter
|
||||
from bot_graphql.filter.short_role_name_filter import ShortRoleNameFilter
|
||||
from bot_graphql.filter.user_filter import UserFilter
|
||||
from bot_graphql.model.server_statistics import ServerStatistics
|
||||
|
||||
|
||||
class ServerQuery(DataQueryWithHistoryABC):
|
||||
@ -107,10 +108,7 @@ class ServerQuery(DataQueryWithHistoryABC):
|
||||
"hasFeatureFlag",
|
||||
lambda server, *_, **kwargs: self._resolve_has_feature_flag(server, *_, **kwargs),
|
||||
)
|
||||
self.set_field(
|
||||
"statistic",
|
||||
lambda server, *_, **kwargs: server,
|
||||
)
|
||||
self.set_field("statistic", lambda server, *_, **kwargs: ServerStatistics(server, kwargs))
|
||||
|
||||
@staticmethod
|
||||
def resolve_id(server: Server, *_):
|
||||
|
@ -2,9 +2,12 @@ import datetime
|
||||
|
||||
from cpl_core.configuration import ConfigurationABC
|
||||
|
||||
from bot_data.abc.achievement_repository_abc import AchievementRepositoryABC
|
||||
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.abc.user_repository_abc import UserRepositoryABC
|
||||
from bot_data.abc.user_warnings_repository_abc import UserWarningsRepositoryABC
|
||||
from bot_data.model.server_config import ServerConfig
|
||||
from bot_graphql.abc.query_abc import QueryABC
|
||||
|
||||
@ -13,63 +16,140 @@ class ServerStatisticQuery(QueryABC):
|
||||
def __init__(
|
||||
self,
|
||||
config: ConfigurationABC,
|
||||
users: UserRepositoryABC,
|
||||
user_joined_voice_channels: UserJoinedVoiceChannelRepositoryABC,
|
||||
user_joined_game_servers: UserJoinedGameServerRepositoryABC,
|
||||
user_messages: UserMessageCountPerHourRepositoryABC,
|
||||
user_warnings: UserWarningsRepositoryABC,
|
||||
achievements: AchievementRepositoryABC,
|
||||
):
|
||||
QueryABC.__init__(self, "ServerStatistic")
|
||||
|
||||
self._config = config
|
||||
|
||||
self._users = users
|
||||
self._user_joined_voice_channels = user_joined_voice_channels
|
||||
self._user_joined_game_servers = user_joined_game_servers
|
||||
self._user_messages = user_messages
|
||||
self._user_warnings = user_warnings
|
||||
self._achievements = achievements
|
||||
|
||||
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(
|
||||
self.set_field(
|
||||
"achievementsAchieved",
|
||||
self._resolve_achievements,
|
||||
)
|
||||
|
||||
self.set_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()
|
||||
self.set_field("userCount", lambda server, *_: self._users.get_users_by_server_id(server.server.id).count())
|
||||
self.set_field("activeUserCount", self._resolve_active_count)
|
||||
|
||||
self.set_field("userJoinedVoiceChannelCount", self._resolve_voice_channel_count)
|
||||
self.set_field("userJoinedVoiceChannelOntime", self._resolve_voice_channel_ontime)
|
||||
|
||||
self.set_field("userJoinedGameServerCount", self._resolve_game_server_count)
|
||||
self.set_field("userJoinedGameServerOntime", self._resolve_game_server_ontime)
|
||||
|
||||
self.set_field("userWarningCount", self._resolve_user_warning_count)
|
||||
|
||||
self.set_field("activityScore", self._resolve_activity_score)
|
||||
|
||||
def _resolve_activity_score(self, server, *_):
|
||||
days = (datetime.date.today() - self._get_date(**server.kwargs)).days
|
||||
return int(
|
||||
(
|
||||
(
|
||||
self._resolve_achievements(server, *_)
|
||||
+ self._resolve_message_count(server, *_)
|
||||
+ self._resolve_voice_channel_count(server, *_)
|
||||
+ self._resolve_voice_channel_ontime(server, *_)
|
||||
+ self._resolve_game_server_count(server, *_)
|
||||
+ self._resolve_game_server_ontime(server, *_)
|
||||
- self._resolve_user_warning_count(server, *_)
|
||||
)
|
||||
/ self._resolve_active_count(server, *_)
|
||||
)
|
||||
/ days
|
||||
* 1000
|
||||
)
|
||||
|
||||
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}")
|
||||
def _resolve_message_count(self, server, *_):
|
||||
settings: ServerConfig = self._config.get_configuration(f"ServerConfig_{server.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))
|
||||
.where(
|
||||
lambda x: x.user.server.id == server.server.id
|
||||
and x.created_at.date() >= self._get_date(**server.kwargs)
|
||||
)
|
||||
.sum(lambda x: x.xp_count / settings.xp_per_message)
|
||||
)
|
||||
|
||||
def _resolve_active_count(self, server, *_):
|
||||
return self._users.get_users_by_server_id(server.server.id).where(lambda x: not x.left_server).count()
|
||||
|
||||
def _resolve_voice_channel_count(self, server, *_):
|
||||
return (
|
||||
self._user_joined_voice_channels.get_user_joined_voice_channels()
|
||||
.where(
|
||||
lambda x: x.user.server.id == server.server.id
|
||||
and x.created_at.date() >= self._get_date(**server.kwargs)
|
||||
)
|
||||
.count()
|
||||
)
|
||||
|
||||
def _resolve_voice_channel_ontime(self, server, *_):
|
||||
return (
|
||||
self._user_joined_voice_channels.get_user_joined_voice_channels()
|
||||
.where(
|
||||
lambda x: x.user.server.id == server.server.id
|
||||
and x.created_at.date() >= self._get_date(**server.kwargs)
|
||||
)
|
||||
.sum(lambda x: x.time)
|
||||
)
|
||||
|
||||
def _resolve_game_server_count(self, server, *_):
|
||||
return (
|
||||
self._user_joined_game_servers.get_user_joined_game_servers()
|
||||
.where(
|
||||
lambda x: x.user.server.id == server.server.id
|
||||
and x.created_at.date() >= self._get_date(**server.kwargs)
|
||||
)
|
||||
.count()
|
||||
)
|
||||
|
||||
def _resolve_game_server_ontime(self, server, *_):
|
||||
return (
|
||||
self._user_joined_game_servers.get_user_joined_game_servers()
|
||||
.where(
|
||||
lambda x: x.user.server.id == server.server.id
|
||||
and x.created_at.date() >= self._get_date(**server.kwargs)
|
||||
)
|
||||
.sum(lambda x: x.time)
|
||||
)
|
||||
|
||||
def _resolve_user_warning_count(self, server, *_):
|
||||
return (
|
||||
self._user_warnings.get_user_warnings()
|
||||
.where(
|
||||
lambda x: x.user.server.id == server.server.id
|
||||
and x.created_at.date() >= self._get_date(**server.kwargs)
|
||||
)
|
||||
.count()
|
||||
)
|
||||
|
||||
def _resolve_achievements(self, server, *_):
|
||||
return (
|
||||
self._achievements.get_achievements_by_server_id(server.server.id)
|
||||
.select_many(lambda x: self._achievements.get_user_got_achievements_by_achievement_id(x.id))
|
||||
.where(
|
||||
lambda x: x.user.server.id == server.server.id
|
||||
and x.created_at.date() >= self._get_date(**server.kwargs)
|
||||
)
|
||||
.count()
|
||||
)
|
||||
|
||||
def _get_date(self, **kwargs) -> datetime.date:
|
||||
if "date" not in kwargs:
|
||||
return datetime.date.today() - datetime.timedelta(days=7)
|
||||
|
Loading…
Reference in New Issue
Block a user