Croc/croccybot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(premium): Ad-whitelist premium guilds.

Browse Source
This commit is contained in:
Conatum
2023-10-26 13:49:03 +03:00
parent d3bd233b3e
commit 73aad8a721
4 changed files with 16 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/modules/sponsors/cog.py
View File

@@ -36,8 +36,12 @@ class SponsorCog(LionCog):
""" """
if not interaction.is_expired(): if not interaction.is_expired():
# TODO: caching # TODO: caching
if interaction.guild:
whitelist = (await self.settings.Whitelist.get(self.bot.appname)).value whitelist = (await self.settings.Whitelist.get(self.bot.appname)).value
if interaction.guild and interaction.guild.id in whitelist: if interaction.guild.id in whitelist:
return
premiumcog = self.bot.get_cog('PremiumCog')
if premiumcog and await premiumcog.is_premium_guild(interaction.guild.id):
return return
setting = await self.settings.SponsorPrompt.get(self.bot.appname) setting = await self.settings.SponsorPrompt.get(self.bot.appname)
value = setting.value value = setting.value

3
src/modules/statistics/ui/leaderboard.py
View File

@@ -537,9 +537,10 @@ class LeaderboardUI(StatsUI):
period_row, period_row,
page_row page_row
] ]
voting = self.bot.get_cog('TopggCog') voting = self.bot.get_cog('TopggCog')
if voting and not await voting.check_voted_recently(self.userid): if voting and not await voting.check_voted_recently(self.userid):
premiumcog = self.bot.get_cog('PremiumCog')
if not (premiumcog and await premiumcog.is_premium_guild(self.guild.id)):
self._layout.append((voting.vote_button(),)) self._layout.append((voting.vote_button(),))
async def reload(self): async def reload(self):

2
src/modules/statistics/ui/profile.py
View File

@@ -299,6 +299,8 @@ class ProfileUI(StatsUI):
voting = self.bot.get_cog('TopggCog') voting = self.bot.get_cog('TopggCog')
if voting and not await voting.check_voted_recently(self.userid): if voting and not await voting.check_voted_recently(self.userid):
premiumcog = self.bot.get_cog('PremiumCog')
if not (premiumcog and await premiumcog.is_premium_guild(self.guild.id)):
self._layout.append((voting.vote_button(),)) self._layout.append((voting.vote_button(),))
async def _render_stats(self): async def _render_stats(self):

2
src/modules/statistics/ui/weeklymonthly.py
View File

@@ -753,6 +753,8 @@ class WeeklyMonthlyUI(StatsUI):
voting = self.bot.get_cog('TopggCog') voting = self.bot.get_cog('TopggCog')
if voting and not await voting.check_voted_recently(self.userid): if voting and not await voting.check_voted_recently(self.userid):
premiumcog = self.bot.get_cog('PremiumCog')
if not (premiumcog and await premiumcog.is_premium_guild(self.guild.id)):
self._layout.append((voting.vote_button(),)) self._layout.append((voting.vote_button(),))
if self._showing_selector: if self._showing_selector: