You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
332 lines
15 KiB
332 lines
15 KiB
import asyncio
|
|
import datetime
|
|
import datetime as dt
|
|
import logging
|
|
|
|
import aiohttp
|
|
import dateutil.parser
|
|
import discord
|
|
import pytz
|
|
from d4dj_utils.master.event_master import EventMaster, EventState
|
|
from discord.ext import commands
|
|
from pytz import UnknownTimeZoneError
|
|
|
|
from miyu_bot.bot.bot import D4DJBot
|
|
from miyu_bot.commands.common.argument_parsing import parse_arguments, ArgumentError
|
|
from miyu_bot.commands.common.asset_paths import get_event_logo_path
|
|
from miyu_bot.commands.common.emoji import attribute_emoji_ids_by_attribute_id, unit_emoji_ids_by_unit_id, \
|
|
parameter_bonus_emoji_ids_by_parameter_id, \
|
|
event_point_emoji_id
|
|
from miyu_bot.commands.common.formatting import format_info
|
|
from miyu_bot.commands.common.fuzzy_matching import romanize
|
|
from miyu_bot.bot.master_asset_manager import hash_master
|
|
from miyu_bot.commands.common.reaction_message import run_paged_message, run_dynamically_paged_message
|
|
from miyu_bot.commands.common.timezone import get_timezone
|
|
|
|
|
|
class Event(commands.Cog):
|
|
bot: D4DJBot
|
|
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
self.logger = logging.getLogger(__name__)
|
|
|
|
@commands.command(name='event',
|
|
aliases=['ev'],
|
|
description='Finds the event with the given name.',
|
|
help='!event pkcooking')
|
|
async def event(self, ctx: commands.Context, *, arg: commands.clean_content = ''):
|
|
self.logger.info(f'Searching for event "{arg}".')
|
|
|
|
event: EventMaster
|
|
|
|
try:
|
|
event, timezone = self.parse_event_argument(ctx, arg)
|
|
except ArgumentError as e:
|
|
await ctx.send(str(e))
|
|
return
|
|
|
|
if not event:
|
|
msg = f'Failed to find event "{arg}".'
|
|
await ctx.send(msg)
|
|
self.logger.info(msg)
|
|
return
|
|
self.logger.info(f'Found event "{event}" ({romanize(event.name)}).')
|
|
|
|
current_id = event.id
|
|
|
|
def generator(n):
|
|
nonlocal current_id
|
|
new_event = self.bot.asset_filters.events.get(current_id + n, ctx)
|
|
if new_event:
|
|
current_id = new_event.id
|
|
return self.get_event_embed(new_event, timezone)
|
|
|
|
asyncio.ensure_future(run_dynamically_paged_message(ctx, generator))
|
|
|
|
def parse_event_argument(self, ctx, arg):
|
|
arguments = parse_arguments(arg)
|
|
timezone = get_timezone(arguments)
|
|
text = arguments.text()
|
|
arguments.require_all_arguments_used()
|
|
|
|
if text:
|
|
# Allows relative id searches like `!event +1` for next event or `!event -2` for the event before last event
|
|
if text[0] in ['-', '+']:
|
|
try:
|
|
latest = self.bot.asset_filters.events.get_latest_event(ctx)
|
|
event = self.bot.asset_filters.events.get(str(latest.id + int(text)), ctx)
|
|
except ValueError:
|
|
event = self.bot.asset_filters.events.get(text, ctx)
|
|
else:
|
|
event = self.bot.asset_filters.events.get(text, ctx)
|
|
else:
|
|
event = self.bot.asset_filters.events.get_latest_event(ctx)
|
|
return event, timezone
|
|
|
|
def get_event_embed(self, event, timezone):
|
|
embed = discord.Embed(title=event.name)
|
|
|
|
embed.set_thumbnail(url=self.bot.asset_url + get_event_logo_path(event))
|
|
|
|
duration_hour_part = round((event.duration.seconds / 3600), 2)
|
|
duration_hour_part = duration_hour_part if not duration_hour_part.is_integer() else int(duration_hour_part)
|
|
duration_hours = round((event.duration.days * 24 + event.duration.seconds / 3600), 2)
|
|
duration_hours = duration_hours if not duration_hours.is_integer() else int(duration_hours)
|
|
|
|
embed.add_field(name='Dates',
|
|
value=format_info({
|
|
'Duration': f'{event.duration.days} days, {duration_hour_part} hours '
|
|
f'({duration_hours} hours)',
|
|
'Start': event.start_datetime.astimezone(timezone),
|
|
'Close': event.reception_close_datetime.astimezone(timezone),
|
|
'Rank Fix': event.rank_fix_start_datetime.astimezone(timezone),
|
|
'Results': event.result_announcement_datetime.astimezone(timezone),
|
|
'End': event.end_datetime.astimezone(timezone),
|
|
'Story Unlock': event.story_unlock_datetime.astimezone(timezone),
|
|
'Status': event.state().name,
|
|
}),
|
|
inline=False)
|
|
embed.add_field(name='Event Type',
|
|
value=event.event_type.name,
|
|
inline=True)
|
|
embed.add_field(name='Bonus Characters',
|
|
value='\n'.join(
|
|
f'{self.bot.get_emoji(unit_emoji_ids_by_unit_id[char.unit_id])} {char.full_name_english}'
|
|
for char in event.bonus.characters
|
|
),
|
|
inline=True)
|
|
embed.add_field(name='Bonus Attribute',
|
|
value=f'{self.bot.get_emoji(attribute_emoji_ids_by_attribute_id[event.bonus.attribute_id])} '
|
|
f'{event.bonus.attribute.en_name.capitalize()}' if event.bonus.attribute else 'None',
|
|
inline=True)
|
|
embed.add_field(name='Point Bonus',
|
|
value=format_info({
|
|
'Attribute': f'{self.bot.get_emoji(event_point_emoji_id)} +{event.bonus.attribute_match_point_bonus_value}%' if event.bonus.attribute_match_point_bonus_value else 'None',
|
|
'Character': f'{self.bot.get_emoji(event_point_emoji_id)} +{event.bonus.character_match_point_bonus_value}%' if event.bonus.character_match_point_bonus_value else 'None',
|
|
'Both': f'{self.bot.get_emoji(event_point_emoji_id)} +{event.bonus.all_match_point_bonus_value}%' if event.bonus.all_match_point_bonus_value else 'None',
|
|
}),
|
|
inline=True)
|
|
embed.add_field(name='Parameter Bonus',
|
|
value=format_info({
|
|
'Attribute': f'{self.bot.get_emoji(parameter_bonus_emoji_ids_by_parameter_id[event.bonus.attribute_match_parameter_bonus_id])} +{event.bonus.attribute_match_parameter_bonus_value}%' if event.bonus.attribute_match_parameter_bonus_value else 'None',
|
|
'Character': f'{self.bot.get_emoji(parameter_bonus_emoji_ids_by_parameter_id[event.bonus.character_match_parameter_bonus_id])} +{event.bonus.attribute_match_parameter_bonus_value}%' if event.bonus.attribute_match_parameter_bonus_value else 'None',
|
|
'Both': f'{self.bot.get_emoji(parameter_bonus_emoji_ids_by_parameter_id[event.bonus.all_match_parameter_bonus_id])} +{event.bonus.all_match_parameter_bonus_value}%' if event.bonus.all_match_parameter_bonus_value else 'None',
|
|
}),
|
|
inline=True)
|
|
embed.set_footer(text=f'Event Id: {event.id}')
|
|
|
|
return embed
|
|
|
|
@commands.command(name='time',
|
|
aliases=[],
|
|
description='Displays the current time',
|
|
help='!time')
|
|
async def time(self, ctx: commands.Context, *, arg=''):
|
|
embed = discord.Embed(title='Time')
|
|
|
|
def format_time(t: dt.datetime):
|
|
return str(t.replace(microsecond=0))
|
|
|
|
embed.add_field(name='Asia/Tokyo', value=format_time(dt.datetime.now(pytz.timezone('Asia/Tokyo'))),
|
|
inline=False)
|
|
|
|
if arg:
|
|
try:
|
|
embed.add_field(name=arg, value=format_time(dt.datetime.now(pytz.timezone(arg))), inline=False)
|
|
except UnknownTimeZoneError:
|
|
await ctx.send(content=f'Invalid timezone "{arg}".')
|
|
return
|
|
else:
|
|
embed.add_field(name='UTC', value=format_time(dt.datetime.now(dt.timezone.utc)), inline=False)
|
|
|
|
await ctx.send(embed=embed)
|
|
|
|
@commands.command(name='timeleft',
|
|
aliases=['tl', 'time_left'],
|
|
description='Displays the time left in the current event',
|
|
help='!timeleft')
|
|
async def time_left(self, ctx: commands.Context, *, arg: commands.clean_content = ''):
|
|
try:
|
|
event, timezone = self.parse_event_argument(ctx, arg)
|
|
except ArgumentError as e:
|
|
await ctx.send(str(e))
|
|
return
|
|
|
|
state = event.state()
|
|
|
|
embed = discord.Embed(title=event.name)
|
|
|
|
embed.set_thumbnail(url=self.bot.asset_url + get_event_logo_path(event))
|
|
|
|
progress = None
|
|
|
|
if state == EventState.Upcoming:
|
|
time_delta_heading = 'Time Until Start'
|
|
delta = event.start_datetime - dt.datetime.now(dt.timezone.utc)
|
|
date_heading = 'Start Date'
|
|
date_value = event.start_datetime
|
|
elif state == EventState.Open:
|
|
time_delta_heading = 'Time Until Close'
|
|
delta = event.reception_close_datetime - dt.datetime.now(dt.timezone.utc)
|
|
progress = 1 - (delta / (event.reception_close_datetime - event.start_datetime))
|
|
date_heading = 'Close Date'
|
|
date_value = event.reception_close_datetime
|
|
elif state in (EventState.Closing, EventState.Ranks_Fixed):
|
|
time_delta_heading = 'Time Until Results'
|
|
delta = event.result_announcement_datetime - dt.datetime.now(dt.timezone.utc)
|
|
date_heading = 'Results Date'
|
|
date_value = event.result_announcement_datetime
|
|
elif state == EventState.Results:
|
|
time_delta_heading = 'Time Until End'
|
|
delta = event.end_datetime - dt.datetime.now(dt.timezone.utc)
|
|
date_heading = 'End Date'
|
|
date_value = event.end_datetime
|
|
else:
|
|
time_delta_heading = 'Time Since End'
|
|
delta = dt.datetime.now(dt.timezone.utc) - event.end_datetime
|
|
date_heading = 'End Date'
|
|
date_value = event.end_datetime
|
|
|
|
date_value = date_value.astimezone(timezone)
|
|
|
|
embed.add_field(name=time_delta_heading,
|
|
value=self.format_timedelta(delta),
|
|
inline=True)
|
|
embed.add_field(name='Progress',
|
|
value=f'{round(progress * 100, 2)}%' if progress is not None else 'N/A',
|
|
inline=True)
|
|
embed.add_field(name=date_heading,
|
|
value=str(date_value),
|
|
inline=True)
|
|
|
|
await ctx.send(embed=embed)
|
|
|
|
@staticmethod
|
|
def format_timedelta(delta: datetime.timedelta):
|
|
days = delta.days
|
|
hours, rem = divmod(delta.seconds, 3600)
|
|
minutes, seconds = divmod(rem, 60)
|
|
return f'{days}d {hours}h {minutes}m'
|
|
|
|
@commands.command(name='t20',
|
|
aliases=['top20', 'top_20'],
|
|
description='Displays the top 20 in the main leaderboard',
|
|
help='!t20')
|
|
async def t20(self, ctx: commands.Context):
|
|
async with aiohttp.ClientSession() as session:
|
|
async with session.get('http://www.projectdivar.com/eventdata/t20') as resp:
|
|
leaderboard = await resp.json(encoding='utf-8')
|
|
event = self.bot.asset_filters.events.get_latest_event(ctx)
|
|
embed = discord.Embed(title=f'{event.name} t20')
|
|
embed.set_thumbnail(url=self.bot.asset_url + get_event_logo_path(event))
|
|
max_points_digits = len(str(leaderboard[0]['points']))
|
|
nl = "\n"
|
|
update_date = dateutil.parser.isoparse(leaderboard[0]["date"]).replace(microsecond=0)
|
|
update_date = update_date.astimezone(pytz.timezone('Asia/Tokyo'))
|
|
header = f'Updated {update_date}\n\nRank {"Points":<{max_points_digits}} Name'
|
|
listing = [
|
|
f'{player["rank"]:<4} {player["points"]:<{max_points_digits}} {player["name"].replace(nl, "")}'
|
|
for player in leaderboard]
|
|
paged = run_paged_message(ctx, embed, listing, header=header, page_size=10, numbered=False)
|
|
asyncio.ensure_future(paged)
|
|
|
|
valid_tiers = {50, 100, 500, 1000, 2000, 5000, 10000, 20000, 30000, 50000}
|
|
|
|
@commands.command(name='cutoff',
|
|
aliases=['co', 't50', 't100', 't500', 't1000', 't2000', 't5000',
|
|
't10000', 't20000', 't30000', 't50000',
|
|
't1k', 't2k', 't5k', 't10k', 't20k', 't30k', 't50k'],
|
|
description=f'Displays the cutoffs at different tiers. Valid tiers: {str(valid_tiers)}',
|
|
help='!cutoff 50')
|
|
async def cutoff(self, ctx: commands.Context, tier: str = ''):
|
|
def process_tier_arg(tier_arg):
|
|
if tier_arg[0] == 't':
|
|
tier_arg = tier_arg[1:]
|
|
if tier_arg[-1] == 'k':
|
|
return str(round(1000 * float(tier_arg[:-1])))
|
|
return tier_arg
|
|
|
|
if ctx.invoked_with in ['cutoff', 'co']:
|
|
tier = process_tier_arg(tier)
|
|
if not tier.isnumeric():
|
|
await ctx.send(f'Invalid tier: {tier}.')
|
|
return
|
|
else:
|
|
tier = process_tier_arg(ctx.invoked_with)
|
|
|
|
embed = await self.get_tier_embed(tier, self.bot.asset_filters.events.get_latest_event(ctx))
|
|
|
|
if embed:
|
|
await ctx.send(embed=embed)
|
|
else:
|
|
await ctx.send(f'No data available for tier {tier}.')
|
|
|
|
async def get_tier_embed(self, tier: str, event: EventMaster):
|
|
async with aiohttp.ClientSession() as session:
|
|
async with session.get('http://www.projectdivar.com/eventdata/t20?chart=true') as resp:
|
|
leaderboard = await resp.json(encoding='utf-8')
|
|
|
|
data = leaderboard['statistics'].get(tier)
|
|
if not data:
|
|
return None
|
|
|
|
if event.state() == EventState.Open:
|
|
delta = event.reception_close_datetime - dt.datetime.now(dt.timezone.utc)
|
|
time_left = self.format_timedelta(delta)
|
|
progress = f'{round(100 * (1 - (delta / (event.reception_close_datetime - event.start_datetime))), 2)}%'
|
|
else:
|
|
time_left = 'N/A'
|
|
progress = 'N/A'
|
|
|
|
embed = discord.Embed(title=f'{event.name} [t{tier}]', timestamp=dt.datetime.now(dt.timezone.utc))
|
|
embed.set_thumbnail(url=self.bot.asset_url + get_event_logo_path(event))
|
|
embed.add_field(name='Points',
|
|
value=data['points'],
|
|
inline=True)
|
|
embed.add_field(name='Last Update',
|
|
value=data['lastUpdate'],
|
|
inline=True)
|
|
embed.add_field(name='Rate',
|
|
value=f'{data["rate"]} pts/hr',
|
|
inline=True)
|
|
embed.add_field(name='Current Estimate',
|
|
value=data['estimate'],
|
|
inline=True)
|
|
embed.add_field(name='Final Prediction',
|
|
value=data['prediction'],
|
|
inline=True)
|
|
embed.add_field(name='\u200b',
|
|
value='\u200b',
|
|
inline=True)
|
|
embed.add_field(name='Time Left',
|
|
value=time_left,
|
|
inline=True)
|
|
embed.add_field(name='Progress',
|
|
value=progress,
|
|
inline=True)
|
|
return embed
|
|
|
|
|
|
def setup(bot):
|
|
bot.add_cog(Event(bot))
|
|
|