Added logic to append received messages
This commit is contained in:
@@ -1,17 +1,33 @@
|
||||
import discord
|
||||
from cpl_core.database.context import DatabaseContextABC
|
||||
from cpl_core.logging.logger_abc import LoggerABC
|
||||
|
||||
from gismo_core.abc.bot_service_abc import BotServiceABC
|
||||
from gismo_data.abc.client_repository_abc import ClientRepositoryABC
|
||||
from gismo_data.abc.server_repository_abc import ServerRepositoryABC
|
||||
from modules_core.abc.events.on_member_join_abc import OnMemberJoinABC
|
||||
from modules_core.abc.events.on_message_abc import OnMessageABC
|
||||
from modules_core.abc.events.on_voice_state_update_abc import OnVoiceStateUpdateABC
|
||||
from modules_core.abc.events.on_voice_state_update_abc import \
|
||||
OnVoiceStateUpdateABC
|
||||
from modules_core.abc.module_abc import ModuleABC
|
||||
|
||||
|
||||
class Base(ModuleABC, OnMemberJoinABC, OnMessageABC, OnVoiceStateUpdateABC):
|
||||
|
||||
def __init__(self, logger: LoggerABC):
|
||||
def __init__(
|
||||
self,
|
||||
logger: LoggerABC,
|
||||
clients: ClientRepositoryABC,
|
||||
servers: ServerRepositoryABC,
|
||||
bot: BotServiceABC,
|
||||
db: DatabaseContextABC
|
||||
):
|
||||
|
||||
self._logger = logger
|
||||
self._clients = clients
|
||||
self._servers = servers
|
||||
self._bot = bot
|
||||
self._db = db
|
||||
|
||||
ModuleABC.__init__(self)
|
||||
self._priorities[OnMemberJoinABC] = 1
|
||||
@@ -19,10 +35,18 @@ class Base(ModuleABC, OnMemberJoinABC, OnMessageABC, OnVoiceStateUpdateABC):
|
||||
self._priorities[OnVoiceStateUpdateABC] = 10
|
||||
self._logger.trace(__name__, f'Module {type(self)} loaded')
|
||||
|
||||
def _apppend_received_message_count(self, g_id: int):
|
||||
try:
|
||||
self._clients.apppend_received_message_count(self._bot.user.id, g_id, 1)
|
||||
self._db.save_changes()
|
||||
except Exception as e:
|
||||
self._logger.error(__name__, f'Cannot edit client {self._bot.user.id}@{g_id}')
|
||||
|
||||
async def on_member_join(self, member: discord.Member):
|
||||
self._logger.debug(__name__, f'Module {type(self)} started')
|
||||
|
||||
async def on_message(self, message: discord.Message):
|
||||
self._apppend_received_message_count(message.guild.id)
|
||||
self._logger.debug(__name__, f'Module {type(self)} started')
|
||||
|
||||
async def on_voice_state_update(self, member: discord.Member, before: discord.VoiceState, after: discord.VoiceState):
|
||||
|
Reference in New Issue
Block a user