mirror of
https://github.com/Specoolazius/mcserver-discordbot.git
synced 2024-12-01 10:39:42 +01:00
80 lines
2.5 KiB
Python
80 lines
2.5 KiB
Python
import asyncio
|
|
import time
|
|
|
|
import discord
|
|
from discord.ext import tasks
|
|
|
|
|
|
class Presence(object):
|
|
"""<object>
|
|
|
|
A class for simply managing the bot's presence.
|
|
"""
|
|
|
|
def __init__(self, bot):
|
|
from libs import Client
|
|
bot: Client
|
|
|
|
self.bot = bot
|
|
self.mc_server = self.bot.mc_server
|
|
self.retry_in_seconds = self.bot.config.retry_in_seconds
|
|
|
|
self.__server_online_presence.start()
|
|
|
|
@tasks.loop()
|
|
async def __server_online_presence(self):
|
|
_time = time.time()
|
|
|
|
try:
|
|
self.bot.logger.debug('Getting server information')
|
|
status = await self.mc_server.async_status()
|
|
|
|
self.bot.is_server_starting = False
|
|
|
|
await self.bot.change_presence(
|
|
activity=discord.Game(
|
|
name=f'with {status.players.online} player{"s" if status.players.online != 1 else ""}',
|
|
),
|
|
status=discord.Status.online
|
|
)
|
|
|
|
await asyncio.sleep(40)
|
|
await self.bot.change_presence(
|
|
activity=discord.Activity(
|
|
type=discord.ActivityType.watching,
|
|
name=self.bot.config.server_address,
|
|
)
|
|
)
|
|
|
|
await asyncio.sleep(20)
|
|
|
|
except asyncio.TimeoutError:
|
|
if self.bot.is_server_starting and self.bot.last_start + self.bot.config.server_start_timout < time.time():
|
|
self.bot.is_server_starting = False
|
|
|
|
# ToDo: better presence
|
|
await self.bot.change_presence(
|
|
activity=discord.Activity(
|
|
type=discord.ActivityType.watching,
|
|
name=f'{self.bot.config.server_address} starting' if self.bot.is_server_starting else
|
|
f'offline: [self.server.address]',
|
|
),
|
|
status=discord.Status.idle
|
|
)
|
|
|
|
# abs -> simple fix if for any reason the sleep_time is negative
|
|
sleep_time = abs(_time - time.time() + self.retry_in_seconds)
|
|
self.bot.logger.debug(f'Server offline, retrying in {round(sleep_time, 4)} seconds')
|
|
await asyncio.sleep(sleep_time)
|
|
|
|
@__server_online_presence.before_loop
|
|
async def __before_status(self) -> None:
|
|
"""< coroutine >
|
|
|
|
Waits until the bot has fully started in case the server
|
|
is already online
|
|
"""
|
|
|
|
await self.bot.wait_until_ready()
|
|
self.bot.logger.info('presence loaded')
|