rewrite: Initial rewrite skeleton.
Remove modules that will no longer be required. Move pending modules to pending-rewrite folders.
This commit is contained in:
53
bot/modules/pending-rewrite/sysadmin/status.py
Normal file
53
bot/modules/pending-rewrite/sysadmin/status.py
Normal file
@@ -0,0 +1,53 @@
|
||||
import time
|
||||
import asyncio
|
||||
import discord
|
||||
from meta import client
|
||||
|
||||
from .module import module
|
||||
|
||||
|
||||
_last_update = 0
|
||||
|
||||
|
||||
async def update_status():
|
||||
# TODO: Make globally configurable and saveable
|
||||
global _last_update
|
||||
|
||||
if time.time() - _last_update < 60:
|
||||
return
|
||||
|
||||
_last_update = time.time()
|
||||
|
||||
student_count, room_count = client.data.current_sessions.select_one_where(
|
||||
select_columns=("COUNT(*) AS studying_count", "COUNT(DISTINCT(channelid)) AS channel_count"),
|
||||
)
|
||||
status = "{} students in {} study rooms!".format(student_count, room_count)
|
||||
|
||||
await client.change_presence(
|
||||
activity=discord.Activity(
|
||||
type=discord.ActivityType.watching,
|
||||
name=status
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
@client.add_after_event("voice_state_update")
|
||||
async def trigger_status_update(client, member, before, after):
|
||||
if before.channel != after.channel:
|
||||
await update_status()
|
||||
|
||||
|
||||
async def _status_loop():
|
||||
while not client.is_ready():
|
||||
await asyncio.sleep(5)
|
||||
while True:
|
||||
try:
|
||||
await update_status()
|
||||
except discord.HTTPException:
|
||||
pass
|
||||
await asyncio.sleep(300)
|
||||
|
||||
|
||||
@module.launch_task
|
||||
async def launch_status_update(client):
|
||||
asyncio.create_task(_status_loop())
|
||||
Reference in New Issue
Block a user