Compare commits
No commits in common. "f9593b5f44e6dbcf0f471746f33702483df62e28" and "c443d108dc8d6b0d795ce1296275aba16ff2a3cd" have entirely different histories.
f9593b5f44
...
c443d108dc
@ -4,8 +4,6 @@ from typing import Callable
|
|||||||
from cpl_query.extension import List
|
from cpl_query.extension import List
|
||||||
from discord.ext.commands import Context
|
from discord.ext.commands import Context
|
||||||
|
|
||||||
from bot_data.model.user import User
|
|
||||||
|
|
||||||
|
|
||||||
class ClientUtilsServiceABC(ABC):
|
class ClientUtilsServiceABC(ABC):
|
||||||
|
|
||||||
@ -35,6 +33,3 @@ class ClientUtilsServiceABC(ABC):
|
|||||||
|
|
||||||
@abstractmethod
|
@abstractmethod
|
||||||
def get_auto_complete_list(self, _l: List, current: str, select: Callable = None) -> List: pass
|
def get_auto_complete_list(self, _l: List, current: str, select: Callable = None) -> List: pass
|
||||||
|
|
||||||
@abstractmethod
|
|
||||||
def get_ontime_for_user(self, user: User) -> float: pass
|
|
||||||
|
@ -17,8 +17,6 @@ from bot_core.configuration.feature_flags_enum import FeatureFlagsEnum
|
|||||||
from bot_core.configuration.feature_flags_settings import FeatureFlagsSettings
|
from bot_core.configuration.feature_flags_settings import FeatureFlagsSettings
|
||||||
from bot_data.abc.client_repository_abc import ClientRepositoryABC
|
from bot_data.abc.client_repository_abc import ClientRepositoryABC
|
||||||
from bot_data.abc.server_repository_abc import ServerRepositoryABC
|
from bot_data.abc.server_repository_abc import ServerRepositoryABC
|
||||||
from bot_data.abc.user_joined_voice_channel_abc import UserJoinedVoiceChannelRepositoryABC
|
|
||||||
from bot_data.model.user import User
|
|
||||||
|
|
||||||
|
|
||||||
class ClientUtilsService(ClientUtilsServiceABC):
|
class ClientUtilsService(ClientUtilsServiceABC):
|
||||||
@ -30,7 +28,6 @@ class ClientUtilsService(ClientUtilsServiceABC):
|
|||||||
bot: DiscordBotServiceABC,
|
bot: DiscordBotServiceABC,
|
||||||
servers: ServerRepositoryABC,
|
servers: ServerRepositoryABC,
|
||||||
clients: ClientRepositoryABC,
|
clients: ClientRepositoryABC,
|
||||||
user_joined_vc: UserJoinedVoiceChannelRepositoryABC,
|
|
||||||
message_service: MessageServiceABC,
|
message_service: MessageServiceABC,
|
||||||
db: DatabaseContextABC,
|
db: DatabaseContextABC,
|
||||||
t: TranslatePipe,
|
t: TranslatePipe,
|
||||||
@ -42,7 +39,6 @@ class ClientUtilsService(ClientUtilsServiceABC):
|
|||||||
self._bot = bot
|
self._bot = bot
|
||||||
self._servers = servers
|
self._servers = servers
|
||||||
self._clients = clients
|
self._clients = clients
|
||||||
self._user_joined_voice_channel = user_joined_vc
|
|
||||||
self._message_service = message_service
|
self._message_service = message_service
|
||||||
self._db = db
|
self._db = db
|
||||||
self._t = t
|
self._t = t
|
||||||
@ -84,11 +80,7 @@ class ClientUtilsService(ClientUtilsServiceABC):
|
|||||||
async def check_if_bot_is_ready_yet_and_respond(self, ctx: Context) -> bool:
|
async def check_if_bot_is_ready_yet_and_respond(self, ctx: Context) -> bool:
|
||||||
result = await self.check_if_bot_is_ready_yet()
|
result = await self.check_if_bot_is_ready_yet()
|
||||||
if not result:
|
if not result:
|
||||||
await self._message_service.send_ctx_msg(
|
await self._message_service.send_ctx_msg(ctx, self._t.transform('common.errors.bot_not_ready_yet'), without_tracking=True)
|
||||||
ctx,
|
|
||||||
self._t.transform('common.errors.bot_not_ready_yet'),
|
|
||||||
without_tracking=True
|
|
||||||
)
|
|
||||||
|
|
||||||
return result
|
return result
|
||||||
|
|
||||||
@ -114,8 +106,3 @@ class ClientUtilsService(ClientUtilsServiceABC):
|
|||||||
_l = _l.where(lambda x: x.name in sl)
|
_l = _l.where(lambda x: x.name in sl)
|
||||||
|
|
||||||
return _l.take(25)
|
return _l.take(25)
|
||||||
|
|
||||||
def get_ontime_for_user(self, user: User) -> float:
|
|
||||||
return self._user_joined_voice_channel.get_user_joined_voice_channels_by_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))
|
|
||||||
|
@ -94,7 +94,7 @@ class UserJoinedVoiceChannelRepositoryService(UserJoinedVoiceChannelRepositoryAB
|
|||||||
)
|
)
|
||||||
|
|
||||||
def find_active_user_joined_voice_channels_by_user_id(self, user_id: int) -> List[Optional[UserJoinedVoiceChannel]]:
|
def find_active_user_joined_voice_channels_by_user_id(self, user_id: int) -> List[Optional[UserJoinedVoiceChannel]]:
|
||||||
self._logger.trace(__name__, f'Send SQL command: {UserJoinedVoiceChannel.get_select_active_by_user_id_string(user_id)}')
|
self._logger.trace(__name__, f'Send SQL command: {UserJoinedVoiceChannel.get_select_by_user_id_string(user_id)}')
|
||||||
result = List(UserJoinedVoiceChannel)
|
result = List(UserJoinedVoiceChannel)
|
||||||
db_results = self._context.select(UserJoinedVoiceChannel.get_select_active_by_user_id_string(user_id))
|
db_results = self._context.select(UserJoinedVoiceChannel.get_select_active_by_user_id_string(user_id))
|
||||||
|
|
||||||
|
@ -9,6 +9,7 @@ from cpl_translation import TranslatePipe
|
|||||||
from discord import app_commands
|
from discord import app_commands
|
||||||
from discord.ext import commands
|
from discord.ext import commands
|
||||||
from discord.ext.commands import Context
|
from discord.ext.commands import Context
|
||||||
|
from mysql.connector.errors import DatabaseError
|
||||||
|
|
||||||
from bot_core.abc.client_utils_service_abc import ClientUtilsServiceABC
|
from bot_core.abc.client_utils_service_abc import ClientUtilsServiceABC
|
||||||
from bot_core.abc.message_service_abc import MessageServiceABC
|
from bot_core.abc.message_service_abc import MessageServiceABC
|
||||||
@ -97,6 +98,10 @@ class UserGroup(DiscordCommandABC):
|
|||||||
color=int('ef9d0d', 16)
|
color=int('ef9d0d', 16)
|
||||||
)
|
)
|
||||||
|
|
||||||
|
ontime = self._user_joined_voice_channel.get_user_joined_voice_channels_by_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))
|
||||||
|
|
||||||
embed.add_field(name=self._t.transform('modules.base.user.atr.id'), value=member.id)
|
embed.add_field(name=self._t.transform('modules.base.user.atr.id'), value=member.id)
|
||||||
embed.add_field(name=self._t.transform('modules.base.user.atr.name'), value=member.name)
|
embed.add_field(name=self._t.transform('modules.base.user.atr.name'), value=member.name)
|
||||||
embed.add_field(name=self._t.transform('modules.base.user.atr.discord_join'),
|
embed.add_field(name=self._t.transform('modules.base.user.atr.discord_join'),
|
||||||
@ -104,8 +109,7 @@ class UserGroup(DiscordCommandABC):
|
|||||||
embed.add_field(name=self._t.transform('modules.base.user.atr.last_join'),
|
embed.add_field(name=self._t.transform('modules.base.user.atr.last_join'),
|
||||||
value=self._date.transform(member.joined_at), inline=False)
|
value=self._date.transform(member.joined_at), inline=False)
|
||||||
embed.add_field(name=self._t.transform('modules.base.user.atr.xp'), value=str(user.xp))
|
embed.add_field(name=self._t.transform('modules.base.user.atr.xp'), value=str(user.xp))
|
||||||
embed.add_field(name=self._t.transform('modules.base.user.atr.ontime'),
|
embed.add_field(name=self._t.transform('modules.base.user.atr.ontime'), value=str(ontime))
|
||||||
value=str(self._client_utils.get_ontime_for_user(user)))
|
|
||||||
|
|
||||||
roles = ''
|
roles = ''
|
||||||
for role in member.roles:
|
for role in member.roles:
|
||||||
@ -158,13 +162,14 @@ class UserGroup(DiscordCommandABC):
|
|||||||
value = str(user.xp)
|
value = str(user.xp)
|
||||||
|
|
||||||
elif atr == 'ontime':
|
elif atr == 'ontime':
|
||||||
value = str(self._client_utils.get_ontime_for_user(user))
|
value = str(round(
|
||||||
|
self._user_joined_voice_channel.get_user_joined_voice_channels_by_user_id(user.user_id)
|
||||||
|
.sum(lambda join: (join.leaved_on - join.joined_on).total_seconds() / 3600),
|
||||||
|
2
|
||||||
|
))
|
||||||
|
|
||||||
else:
|
else:
|
||||||
await self._message_service.send_interaction_msg(
|
await self._message_service.send_interaction_msg(ctx.interaction, self._t.transform('modules.base.user.error.atr_not_found').format(atr))
|
||||||
ctx.interaction,
|
|
||||||
self._t.transform('modules.base.user.error.atr_not_found').format(atr)
|
|
||||||
)
|
|
||||||
return
|
return
|
||||||
|
|
||||||
await self._message_service.send_interaction_msg(
|
await self._message_service.send_interaction_msg(
|
||||||
@ -191,19 +196,14 @@ class UserGroup(DiscordCommandABC):
|
|||||||
|
|
||||||
if atr == 'xp':
|
if atr == 'xp':
|
||||||
if not value.isnumeric():
|
if not value.isnumeric():
|
||||||
await self._message_service.send_interaction_msg(
|
await self._message_service.send_interaction_msg(ctx.interaction, self._t.transform('modules.base.user.set.error.value_type_not_numeric'))
|
||||||
ctx.interaction, self._t.transform('modules.base.user.set.error.value_type_not_numeric')
|
|
||||||
)
|
|
||||||
return
|
return
|
||||||
|
|
||||||
try:
|
try:
|
||||||
user.xp = int(value)
|
user.xp = int(value)
|
||||||
except TypeError as te:
|
except TypeError as te:
|
||||||
self._logger.error(__name__, f'String value couldn\'t be converted to int', te)
|
self._logger.error(__name__, f'String value couldn\'t be converted to int', te)
|
||||||
await self._message_service.send_interaction_msg(
|
await self._message_service.send_interaction_msg(ctx.interaction, self._t.transform('modules.base.user.set.error.type_error'))
|
||||||
ctx.interaction,
|
|
||||||
self._t.transform('modules.base.user.set.error.type_error')
|
|
||||||
)
|
|
||||||
return
|
return
|
||||||
else:
|
else:
|
||||||
self._users.update_user(user)
|
self._users.update_user(user)
|
||||||
@ -211,16 +211,10 @@ class UserGroup(DiscordCommandABC):
|
|||||||
await self._level.check_level(member)
|
await self._level.check_level(member)
|
||||||
|
|
||||||
else:
|
else:
|
||||||
await self._message_service.send_interaction_msg(
|
await self._message_service.send_interaction_msg(ctx.interaction, self._t.transform('modules.base.user.error.atr_not_found').format(atr))
|
||||||
ctx.interaction,
|
|
||||||
self._t.transform('modules.base.user.error.atr_not_found').format(atr)
|
|
||||||
)
|
|
||||||
return
|
return
|
||||||
|
|
||||||
await self._message_service.send_interaction_msg(
|
await self._message_service.send_interaction_msg(ctx.interaction, self._t.transform(f'modules.base.user.set.{atr.lower()}').format(member.mention, value))
|
||||||
ctx.interaction,
|
|
||||||
self._t.transform(f'modules.base.user.set.{atr.lower()}').format(member.mention, value)
|
|
||||||
)
|
|
||||||
|
|
||||||
@set.autocomplete('atr')
|
@set.autocomplete('atr')
|
||||||
async def set_autocomplete(self, interaction: discord.Interaction, current: str) -> List[app_commands.Choice[str]]:
|
async def set_autocomplete(self, interaction: discord.Interaction, current: str) -> List[app_commands.Choice[str]]:
|
||||||
@ -251,10 +245,7 @@ class UserGroup(DiscordCommandABC):
|
|||||||
self._db.save_changes()
|
self._db.save_changes()
|
||||||
|
|
||||||
else:
|
else:
|
||||||
await self._message_service.send_interaction_msg(
|
await self._message_service.send_interaction_msg(ctx.interaction, self._t.transform('modules.base.user.error.atr_not_found').format(atr))
|
||||||
ctx.interaction,
|
|
||||||
self._t.transform('modules.base.user.error.atr_not_found').format(atr)
|
|
||||||
)
|
|
||||||
return
|
return
|
||||||
|
|
||||||
await self._message_service.send_interaction_msg(
|
await self._message_service.send_interaction_msg(
|
||||||
@ -263,7 +254,5 @@ class UserGroup(DiscordCommandABC):
|
|||||||
)
|
)
|
||||||
|
|
||||||
@remove.autocomplete('atr')
|
@remove.autocomplete('atr')
|
||||||
async def remove_autocomplete(self, interaction: discord.Interaction, current: str) -> List[
|
async def remove_autocomplete(self, interaction: discord.Interaction, current: str) -> List[app_commands.Choice[str]]:
|
||||||
app_commands.Choice[str]
|
|
||||||
]:
|
|
||||||
return [app_commands.Choice(name=value, value=key) for key, value in self._atr_list]
|
return [app_commands.Choice(name=value, value=key) for key, value in self._atr_list]
|
||||||
|
@ -7,7 +7,6 @@ from cpl_core.database.context import DatabaseContextABC
|
|||||||
from cpl_core.logging import LoggerABC
|
from cpl_core.logging import LoggerABC
|
||||||
from cpl_discord.events import OnVoiceStateUpdateABC
|
from cpl_discord.events import OnVoiceStateUpdateABC
|
||||||
|
|
||||||
from bot_core.abc.client_utils_service_abc import ClientUtilsServiceABC
|
|
||||||
from bot_core.helper.event_checks import EventChecks
|
from bot_core.helper.event_checks import EventChecks
|
||||||
from bot_data.abc.known_user_repository_abc import KnownUserRepositoryABC
|
from bot_data.abc.known_user_repository_abc import KnownUserRepositoryABC
|
||||||
from bot_data.abc.server_repository_abc import ServerRepositoryABC
|
from bot_data.abc.server_repository_abc import ServerRepositoryABC
|
||||||
@ -33,7 +32,6 @@ class BaseOnVoiceStateUpdateEvent(OnVoiceStateUpdateABC):
|
|||||||
users: UserRepositoryABC,
|
users: UserRepositoryABC,
|
||||||
user_joins: UserJoinedServerRepositoryABC,
|
user_joins: UserJoinedServerRepositoryABC,
|
||||||
user_joins_vc: UserJoinedVoiceChannelRepositoryABC,
|
user_joins_vc: UserJoinedVoiceChannelRepositoryABC,
|
||||||
client_utils: ClientUtilsServiceABC,
|
|
||||||
db: DatabaseContextABC,
|
db: DatabaseContextABC,
|
||||||
):
|
):
|
||||||
OnVoiceStateUpdateABC.__init__(self)
|
OnVoiceStateUpdateABC.__init__(self)
|
||||||
@ -45,7 +43,6 @@ class BaseOnVoiceStateUpdateEvent(OnVoiceStateUpdateABC):
|
|||||||
self._users = users
|
self._users = users
|
||||||
self._user_joins = user_joins
|
self._user_joins = user_joins
|
||||||
self._user_joins_vc = user_joins_vc
|
self._user_joins_vc = user_joins_vc
|
||||||
self._client_utils = client_utils
|
|
||||||
self._db = db
|
self._db = db
|
||||||
|
|
||||||
self._logger.info(__name__, f'Module {type(self)} loaded')
|
self._logger.info(__name__, f'Module {type(self)} loaded')
|
||||||
@ -75,7 +72,7 @@ class BaseOnVoiceStateUpdateEvent(OnVoiceStateUpdateABC):
|
|||||||
join.leaved_on = datetime.now()
|
join.leaved_on = datetime.now()
|
||||||
|
|
||||||
# ontime as hours
|
# ontime as hours
|
||||||
ontime = self._client_utils.get_ontime_for_user(user)
|
ontime = round((join.leaved_on - join.joined_on).total_seconds() / 3600, 2)
|
||||||
old_xp = user.xp
|
old_xp = user.xp
|
||||||
user.xp += round(ontime * settings.xp_per_ontime_hour)
|
user.xp += round(ontime * settings.xp_per_ontime_hour)
|
||||||
|
|
||||||
@ -83,18 +80,12 @@ class BaseOnVoiceStateUpdateEvent(OnVoiceStateUpdateABC):
|
|||||||
self._users.update_user(user)
|
self._users.update_user(user)
|
||||||
self._db.save_changes()
|
self._db.save_changes()
|
||||||
|
|
||||||
self._logger.debug(__name__,
|
self._logger.debug(__name__, f'User {user} leaved_on {join.leaved_on}. Ontime: {ontime}h | xp: from {old_xp} to {user.xp}')
|
||||||
f'User {user} leaved_on {join.leaved_on}. Ontime: {ontime}h | xp: from {old_xp} to {user.xp}')
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
self._logger.error(__name__, f'Ontime validation failed', e)
|
self._logger.error(__name__, f'Ontime validation failed', e)
|
||||||
|
|
||||||
@EventChecks.check_is_ready()
|
@EventChecks.check_is_ready()
|
||||||
async def on_voice_state_update(
|
async def on_voice_state_update(self, member: discord.Member, before: discord.VoiceState, after: discord.VoiceState):
|
||||||
self,
|
|
||||||
member: discord.Member,
|
|
||||||
before: discord.VoiceState,
|
|
||||||
after: discord.VoiceState
|
|
||||||
):
|
|
||||||
self._logger.debug(__name__, f'Module {type(self)} started')
|
self._logger.debug(__name__, f'Module {type(self)} started')
|
||||||
self._logger.trace(__name__, f'Detected on_voice_state_update {member.id} from {before} to {after}')
|
self._logger.trace(__name__, f'Detected on_voice_state_update {member.id} from {before} to {after}')
|
||||||
settings: BaseServerSettings = self._base_helper.get_config(member.guild.id)
|
settings: BaseServerSettings = self._base_helper.get_config(member.guild.id)
|
||||||
|
@ -42,15 +42,8 @@ class LevelService:
|
|||||||
self._t = t
|
self._t = t
|
||||||
|
|
||||||
def get_level(self, user: User) -> Level:
|
def get_level(self, user: User) -> Level:
|
||||||
levels_by_server = self._levels.get_levels_by_server_id(user.server.server_id)
|
levels = self._levels.get_levels_by_server_id(user.server.server_id).order_by(lambda l: l.min_xp)
|
||||||
levels = levels_by_server \
|
return levels.where(lambda l: user.xp >= l.min_xp).last()
|
||||||
.order_by(lambda l: l.min_xp) \
|
|
||||||
.where(lambda l: user.xp >= l.min_xp)
|
|
||||||
|
|
||||||
if levels.count() == 0:
|
|
||||||
return levels_by_server.order_by(lambda l: l.min_xp).last()
|
|
||||||
|
|
||||||
return levels.last()
|
|
||||||
|
|
||||||
async def set_level(self, user: User):
|
async def set_level(self, user: User):
|
||||||
level_names = self._levels.get_levels_by_server_id(user.server.server_id).select(lambda l: l.name)
|
level_names = self._levels.get_levels_by_server_id(user.server.server_id).select(lambda l: l.name)
|
||||||
|
Loading…
Reference in New Issue
Block a user