def stars_stats(self, event, user=None): if user: try: given_stars = list(StarboardEntry.select( fn.COUNT('*'), ).join(Message).where( (~ (StarboardEntry.star_message_id >> None)) & (StarboardEntry.stars.contains(user.id)) & (Message.guild_id == event.guild.id) ).tuples())[0][0] recieved_stars_posts, recieved_stars_total = list(StarboardEntry.select( fn.COUNT('*'), fn.SUM(fn.array_length(StarboardEntry.stars, 1)), ).join(Message).where( (~ (StarboardEntry.star_message_id >> None)) & (Message.author_id == user.id) & (Message.guild_id == event.guild.id) ).tuples())[0] except: return event.msg.reply(':warning: failed to crunch the numbers on that user') embed = MessageEmbed() embed.color = 0xffd700 embed.title = user.username embed.set_thumbnail(url=user.avatar_url) embed.add_field(name='Total Stars Given', value=str(given_stars), inline=True) embed.add_field(name='Total Posts w/ Stars', value=str(recieved_stars_posts), inline=True) embed.add_field(name='Total Stars Recieved', value=str(recieved_stars_total), inline=True) # embed.add_field(name='Star Rank', value='#{}'.format(recieved_stars_rank), inline=True) return event.msg.reply('', embed=embed) total_starred_posts, total_stars = list(StarboardEntry.select( fn.COUNT('*'), fn.SUM(fn.array_length(StarboardEntry.stars, 1)), ).join(Message).where( (~ (StarboardEntry.star_message_id >> None)) & (Message.guild_id == event.guild.id) ).tuples())[0] top_users = list(StarboardEntry.select(fn.SUM(fn.array_length(StarboardEntry.stars, 1)), User.user_id).join( Message, ).join( User, on=(Message.author_id == User.user_id), ).where( (~ (StarboardEntry.star_message_id >> None)) & (fn.array_length(StarboardEntry.stars, 1) > 0) & (Message.guild_id == event.guild.id) ).group_by(User).order_by(fn.SUM(fn.array_length(StarboardEntry.stars, 1)).desc()).limit(5).tuples()) embed = MessageEmbed() embed.color = 0xffd700 embed.title = 'Star Stats' embed.add_field(name='Total Stars Given', value=total_stars, inline=True) embed.add_field(name='Total Starred Posts', value=total_starred_posts, inline=True) embed.add_field(name='Top Star Recievers', value='\n'.join( '{}. <@{}> ({})'.format(idx + 1, row[1], row[0]) for idx, row in enumerate(top_users) )) event.msg.reply('', embed=embed)
def force_update_stars(self, event): # First, iterate over stars and repull their reaction count stars = StarboardEntry.select().join(Message).where( (Message.guild_id == event.guild.id) & (~(StarboardEntry.star_message_id >> None))).order_by( Message.timestamp.desc()).limit(100) msg = event.msg.reply('Updating starboard...') for star in stars: info_msg = self.client.api.channels_messages_get( star.message.channel_id, star.message_id) users = [i.id for i in msg.get_reactors(STAR_EMOJI)] if set(users) != set(star.stars): self.log.warning('star %s had outdated reactors list %s vs %s', star.message_id, users, star.stars) StarboardEntry.update( stars=users, dirty=True, ).where( (StarboardEntry.message_id == star.message_id)).execute() info_msg.edit('Starboard updated!') self.queue_update(event.guild.id, event.config)
def stars_update(self, event, mid): try: entry = StarboardEntry.select(StarboardEntry, Message).join( Message ).where( (Message.guild_id == event.guild.id) & (StarboardEntry.message_id == mid) ).get() except StarboardEntry.DoesNotExist: return event.msg.reply(':warning: no starboard entry exists with that message id') msg = self.client.api.channels_messages_get( entry.message.channel_id, entry.message_id) users = [i.id for i in msg.get_reactors(STAR_EMOJI)] if set(users) != set(entry.stars): StarboardEntry.update( stars=users, dirty=True ).where( (StarboardEntry.message_id == entry.message_id) ).execute() else: StarboardEntry.update( dirty=True ).where( (StarboardEntry.message_id == mid) ).execute() self.queue_update(event.guild.id, event.config) event.msg.reply(u'Forcing an update on message {}'.format(mid))
def update_starboard(self, guild_id, config, event): sb_id2, board = config.get_board(event.channel.id) if not sb_id2: return # Grab all dirty stars that where posted in the last 32 hours stars = StarboardEntry.select().join(Message).where( (StarboardEntry.dirty == 1) & (Message.guild_id == guild_id) & (Message.timestamp > (datetime.utcnow() - timedelta(hours=board.max_age)))) for star in stars: sb_id, sb_config = config.get_board(star.message.channel_id) if not sb_id: StarboardEntry.update(dirty=False).where( StarboardEntry.message_id == star.message_id).execute() continue # If this star has no stars, delete it from the starboard if not star.stars: if not star.star_channel_id: StarboardEntry.update( dirty=False).where(StarboardEntry.message_id == star.message_id).execute() continue self.delete_star(star) continue # Grab the original message try: source_msg = self.client.api.channels_messages_get( star.message.channel_id, star.message_id) except: self.log.exception('Star message went missing %s / %s: ', star.message.channel_id, star.message_id) # TODO: really delete this self.delete_star(star, update=True) continue # If we previously posted this in the wrong starboard, delete it if star.star_channel_id and (star.star_channel_id != sb_id or len( star.stars) < sb_config.min_stars) or star.blocked: self.delete_star(star, update=True) if len(star.stars) < sb_config.min_stars or star.blocked: StarboardEntry.update(dirty=False).where( StarboardEntry.message_id == star.message_id).execute() continue self.post_star(star, source_msg, sb_id, sb_config)
def stars_show(self, event, mid): try: star = StarboardEntry.select().join(Message).where( (Message.guild_id == event.guild.id) & (~(StarboardEntry.star_message_id >> None)) & ((Message.id == mid) | (StarboardEntry.star_message_id == mid))).get() except StarboardEntry.DoesNotExist: raise CommandFail('no starboard message with that id') _, sb_config = event.config.get_board(star.message.channel_id) try: source_msg = self.client.api.channels_messages_get( star.message.channel_id, star.message_id) except: raise CommandFail('no starboard message with that id') content, embed = self.get_embed(star, source_msg, sb_config) event.msg.reply(content, embed=embed)