71 lines
2.5 KiB
Python
71 lines
2.5 KiB
Python
import aiohttp
|
|
from discord.ext import commands, tasks
|
|
import os
|
|
|
|
class StatChannel(commands.Cog):
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
|
|
def _parse_id(key: str) -> int | None:
|
|
val = os.getenv(key, "")
|
|
return int(val) if val.isdigit() else None
|
|
|
|
self.channel_id = _parse_id('STRINOVA_STAT_CHANNEL_ID')
|
|
self.channel_id_mao = _parse_id('STRINOVA_STAT_CHANNEL_ID_MAO')
|
|
self.member_channel_id = _parse_id('MEMBER_COUNT_CHANNEL_ID')
|
|
|
|
self.update_stat.start()
|
|
if self.channel_id_mao is not None:
|
|
self.update_stat_mao.start()
|
|
if self.member_channel_id is not None:
|
|
self.update_members.start()
|
|
|
|
def cog_unload(self):
|
|
self.update_stat.cancel()
|
|
self.update_stat_mao.cancel()
|
|
self.update_members.cancel()
|
|
|
|
async def _fetch_and_update(self, channel_id):
|
|
try:
|
|
async with aiohttp.ClientSession() as session:
|
|
async with session.get('https://api.steampowered.com/ISteamUserStats/GetNumberOfCurrentPlayers/v1/?appid=1282270') as r:
|
|
data = await r.json()
|
|
count = data['response']['player_count']
|
|
channel = self.bot.get_channel(channel_id)
|
|
if channel:
|
|
await channel.edit(name=f'Strinova Players From Steam: {count:,}')
|
|
except Exception as e:
|
|
print(f'stat channel update failed (channel {channel_id}): {e}')
|
|
|
|
@tasks.loop(minutes=10)
|
|
async def update_stat(self):
|
|
await self._fetch_and_update(self.channel_id)
|
|
|
|
@tasks.loop(minutes=10)
|
|
async def update_stat_mao(self):
|
|
await self._fetch_and_update(self.channel_id_mao)
|
|
|
|
@tasks.loop(minutes=10)
|
|
async def update_members(self):
|
|
try:
|
|
channel = self.bot.get_channel(self.member_channel_id)
|
|
if channel:
|
|
await channel.edit(name=f'Member Count: {channel.guild.member_count:,}')
|
|
except Exception as e:
|
|
print(f'member count update failed (channel {self.member_channel_id}): {e}')
|
|
|
|
@update_stat.before_loop
|
|
async def before_update_stat(self):
|
|
await self.bot.wait_until_ready()
|
|
|
|
@update_stat_mao.before_loop
|
|
async def before_update_stat_mao(self):
|
|
await self.bot.wait_until_ready()
|
|
|
|
@update_members.before_loop
|
|
async def before_update_members(self):
|
|
await self.bot.wait_until_ready()
|
|
|
|
async def setup(bot):
|
|
await bot.add_cog(StatChannel(bot))
|