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)
