mirror of
https://github.com/nexy7574/LCC-bot.git
synced 2024-09-20 02:26:32 +01:00
Starboard removes starless messages
This commit is contained in:
parent
ffef7e7fb0
commit
047cf6ea40
1 changed files with 20 additions and 0 deletions
|
@ -7,6 +7,7 @@ import httpx
|
||||||
from typing import Tuple
|
from typing import Tuple
|
||||||
|
|
||||||
import discord
|
import discord
|
||||||
|
import orm
|
||||||
from discord.ext import commands
|
from discord.ext import commands
|
||||||
from utils.db import StarBoardMessage
|
from utils.db import StarBoardMessage
|
||||||
|
|
||||||
|
@ -135,6 +136,25 @@ class StarBoardCog(commands.Cog):
|
||||||
star_count = 0
|
star_count = 0
|
||||||
else:
|
else:
|
||||||
star_count = star_count[0].count
|
star_count = star_count[0].count
|
||||||
|
|
||||||
|
if star_count == 0:
|
||||||
|
try:
|
||||||
|
database: StarBoardMessage = await StarBoardMessage.objects.get(id=payload.message_id)
|
||||||
|
except orm.NoMatch:
|
||||||
|
return
|
||||||
|
else:
|
||||||
|
channel = discord.utils.get(message.guild.text_channels, name="starboard")
|
||||||
|
if channel:
|
||||||
|
try:
|
||||||
|
message = await channel.fetch_message(database.id)
|
||||||
|
await message.delete(delay=0.1, reason="Starboard message lost all stars.")
|
||||||
|
except discord.HTTPException:
|
||||||
|
pass
|
||||||
|
finally:
|
||||||
|
await database.delete()
|
||||||
|
else:
|
||||||
|
await database.delete()
|
||||||
|
|
||||||
database: Tuple[StarBoardMessage, bool] = await StarBoardMessage.objects.get_or_create(
|
database: Tuple[StarBoardMessage, bool] = await StarBoardMessage.objects.get_or_create(
|
||||||
{"channel": payload.channel_id}, id=payload.message_id
|
{"channel": payload.channel_id}, id=payload.message_id
|
||||||
)
|
)
|
||||||
|
|
Loading…
Reference in a new issue