LITERALLY THE WORST CODE I HAVE EVER WRITTEN BUT IT SOMEHOW WORKS I THINK

pull/9/head
sgoudham 4 years ago
parent 4db28b100d
commit eae78ea498

@ -133,7 +133,8 @@ class Bot(commands.Bot):
# Store the information for starboard within cache # Store the information for starboard within cache
for row in results: for row in results:
self.starboard_cache[row["guild_id"]] = {"channel_id": row["channel_id"]} self.starboard_cache[row["guild_id"]] = {"channel_id": row["channel_id"],
"min_stars": row["min_stars"]}
# Catch errors # Catch errors
except asyncpg.PostgresError as e: except asyncpg.PostgresError as e:
@ -250,22 +251,34 @@ class Bot(commands.Bot):
# --------------------------------------------!Starboard Section!--------------------------------------------------- # --------------------------------------------!Starboard Section!---------------------------------------------------
def cache_store_starboard(self, guild_id, channel_id): def cache_store_starboard(self, guild_id, channel_id, min_stars):
"""Storing starboard within cache""" """Storing starboard within cache"""
self.starboard_cache[guild_id] = {"channel_id": channel_id} self.starboard_cache[guild_id] = {"channel_id": channel_id,
"min_stars": min_stars}
def get_starboard(self, guild_id): def get_starboard_channel(self, guild_id):
"""Returning the starboard of the guild""" """Returning the starboard channel of the guild"""
starboard = self.starboard_cache.get(guild_id) starboard = self.starboard_cache.get(guild_id)
return starboard.get("channel_id") if starboard else None return starboard.get("channel_id") if starboard else None
def update_starboard(self, guild_id, channel_id): def get_starboard_min_stars(self, guild_id):
"""Returning the starboard minimum stars of the guild"""
starboard = self.starboard_cache.get(guild_id)
return starboard.get("min_stars") if starboard else None
def update_starboard_channel(self, guild_id, channel_id):
"""Update the starboard channel""" """Update the starboard channel"""
self.starboard_cache[guild_id]["channel_id"] = channel_id self.starboard_cache[guild_id]["channel_id"] = channel_id
def update_starboard_min_stars(self, guild_id, min_stars):
"""Update the starboard minimum stars"""
self.starboard_cache[guild_id]["min_stars"] = min_stars
def delete_starboard(self, guild_id): def delete_starboard(self, guild_id):
"""Deleting the starboard of the guild""" """Deleting the starboard of the guild"""
@ -293,6 +306,12 @@ class Bot(commands.Bot):
self.starboard_messages_cache[root_message_id, guild_id] = {"star_message_id": star_message_id, self.starboard_messages_cache[root_message_id, guild_id] = {"star_message_id": star_message_id,
"stars": 1} "stars": 1}
def update_starboard_message_stars(self, root_message_id, guild_id, star_message_id):
"""Store the starboard messages within cache"""
self.starboard_messages_cache[root_message_id, guild_id] = {"star_message_id": star_message_id,
"stars": 1}
def update_starboard_message(self, root_message_id, guild_id, reactions): def update_starboard_message(self, root_message_id, guild_id, reactions):
"""Update the stored starboard message""" """Update the stored starboard message"""
@ -336,7 +355,7 @@ class Bot(commands.Bot):
finally: finally:
await pool.release(conn) await pool.release(conn)
# --------------------------------------------!EndStarbard Section!------------------------------------------------- # --------------------------------------------!EndStarboard Section!-------------------------------------------------
# --------------------------------------------!Modmail Section!----------------------------------------------------- # --------------------------------------------!Modmail Section!-----------------------------------------------------
@ -674,7 +693,7 @@ class Bot(commands.Bot):
# Delete all information about the starboard and any messages stored # Delete all information about the starboard and any messages stored
else: else:
print(rowcount, f"Starboard deleted successfully from Guild {guild}") print(rowcount, f"Starboard deleted successfully from Guild {guild}")
if self.get_starboard(guild.id): if self.get_starboard_channel(guild.id):
self.delete_starboard(guild.id) self.delete_starboard(guild.id)
self.delete_starboard_messages(guild.id) self.delete_starboard_messages(guild.id)
@ -840,7 +859,7 @@ class Bot(commands.Bot):
# Get the modlogs channel (channel or none) # Get the modlogs channel (channel or none)
modlogs = self.get_modlog_for_guild(channel.guild.id) modlogs = self.get_modlog_for_guild(channel.guild.id)
# Get the starboard (record or none) # Get the starboard (record or none)
starboard = self.get_starboard(channel.guild.id) starboard = self.get_starboard_channel(channel.guild.id)
# Get the modmail record - (normal and logging channels) # Get the modmail record - (normal and logging channels)
modmail_record = self.get_modmail(channel.guild.id) modmail_record = self.get_modmail(channel.guild.id)
@ -891,7 +910,7 @@ class Bot(commands.Bot):
# Delete all information about the starboard and any messages stored # Delete all information about the starboard and any messages stored
else: else:
print(rowcount, f"Starboard deleted successfully from Guild {channel.guild}") print(rowcount, f"Starboard deleted successfully from Guild {channel.guild}")
if self.get_starboard(channel.guild.id): if self.get_starboard_channel(channel.guild.id):
self.delete_starboard(channel.guild.id) self.delete_starboard(channel.guild.id)
self.delete_starboard_messages(channel.guild.id) self.delete_starboard_messages(channel.guild.id)

Loading…
Cancel
Save