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

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

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

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

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

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

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