async def _add_db_sighting_report(self, ctx, message):
channel = ctx.channel
guild = channel.guild
author = ctx.author
wild_dict = self.bot.guild_dict[guild.id]['wildreport_dict'][message.id]
created = round(message.created_at.timestamp())
__, __ = GuildTable.get_or_create(snowflake=guild.id)
__, __ = TrainerTable.get_or_create(snowflake=author.id, guild=guild.id)
report = TrainerReportRelation.create(created=created, trainer=author.id,
location=wild_dict['location_id'], message=message.id)
try:
SightingTable.create(trainer_report=report, pokemon=wild_dict['pokemon_id'])
except Exception as e:
self.bot.logger.info(f"Failed to create sighting table entry with error: {e}")
async def _add_db_research_report(self, ctx, message):
channel = ctx.channel
guild = channel.guild
author = ctx.author
quest_dict = self.bot.guild_dict[guild.id]['questreport_dict'][
message.id]
created = round(message.created_at.timestamp())
__, __ = GuildTable.get_or_create(snowflake=guild.id)
__, __ = TrainerTable.get_or_create(snowflake=author.id,
guild=guild.id)
report = TrainerReportRelation.create(
created=created,
trainer=author.id,
location=quest_dict['location_id'],
message=message.id)
try:
ResearchTable.create(trainer_report=report,
quest=quest_dict['quest_id'],
reward=quest_dict['reward'])
except Exception as e:
self.bot.logger.info(
f"Failed to create research table entry with error: {e}")
async def _lure(self, ctx, type, *, location):
"""**Usage**: `!lure <type> <location>`
Location should be the name of a Pokestop.
Valid lure types are: normal, glacial, mossy, magnetic"""
content = f"{type} {location}"
message = ctx.message
guild = message.guild
channel = message.channel
author = message.author
location_matching_cog = self.bot.cogs.get('LocationMatching')
subscriptions_cog = self.bot.cogs.get('Subscriptions')
if len(content.split()) <= 1:
self.bot.help_logger.info(
f"User: {ctx.author.name}, channel: {ctx.channel}, error: Insufficient info."
)
return await channel.send(embed=discord.Embed(
colour=discord.Colour.red(),
description=
'Give more details when reporting! Usage: **!lure <type> <location>**'
))
report_time = message.created_at + datetime.timedelta(
hours=self.bot.guild_dict[
guild.id]['configure_dict']['settings']['offset'])
report_time_int = round(report_time.timestamp())
timestamp = report_time.strftime('%Y-%m-%d %H:%M:%S')
luretype = content.split()[0].strip(',')
pokestop = ' '.join(content.split()[1:])
query = LureTypeTable.select()
if id is not None:
query = query.where(LureTypeTable.name == luretype)
query = query.execute()
result = [d for d in query]
if len(result) != 1:
self.bot.help_logger.info(
f"User: {ctx.author.name}, channel: {ctx.channel}, error: Lure type: {luretype} not found."
)
return await channel.send(embed=discord.Embed(
colour=discord.Colour.red(),
description=
'Unable to find the lure type provided, please try again.'))
luretype = result[0]
utils_cog = self.bot.cogs.get('Utilities')
lure_regions = utils_cog.get_channel_regions(channel, 'lure')
stops = location_matching_cog.get_stops(guild.id, lure_regions)
if stops:
stop = await location_matching_cog.match_prompt(
channel, author.id, pokestop, stops)
if not stop:
self.bot.help_logger.info(
f"User: {ctx.author.name}, channel: {ctx.channel}, error: Pokestop not found with name: {pokestop}."
)
return await channel.send(embed=discord.Embed(
colour=discord.Colour.red(),
description="Unable to find that Pokestop. "
"Please check the name and try again!"))
report = TrainerReportRelation.create(guild=ctx.guild.id,
created=report_time_int,
trainer=author.id,
location=stop.id)
lure = LureTable.create(trainer_report=report)
LureTypeRelation.create(lure=lure, type=luretype)
lure_embed = discord.Embed(
title=
f'Click here for my directions to the {luretype.name.capitalize()} lure!',
description=
f"Ask {author.display_name} if my directions aren't perfect!",
url=stop.maps_url,
colour=discord.Colour.purple())
lure_embed.set_footer(text='Reported by {author} - {timestamp}'.format(
author=author.display_name, timestamp=timestamp),
icon_url=author.avatar_url_as(
format=None, static_format='jpg', size=32))
lurereportmsg = await channel.send(
f'**{luretype.name.capitalize()}** lure reported by '
f'{author.display_name} at {stop.name}',
embed=lure_embed)
listmgmt_cog = self.bot.cogs.get('ListManagement')
await listmgmt_cog.update_listing_channels(guild,
'lure',
edit=False,
regions=lure_regions)
details = {
'regions': lure_regions,
'type': 'lure',
'lure_type': luretype.name,
'location': stop
}
send_channel = subscriptions_cog.get_region_list_channel(
guild, stop.region, 'lure')
if send_channel is None:
send_channel = message.channel
await subscriptions_cog.send_notifications_async(
'lure', details, send_channel, [message.author.id])
self.bot.event_loop.create_task(
self.lure_expiry_check(lurereportmsg, report.id))
clean_list = self.bot.guild_dict[
ctx.guild.id]['configure_dict'].setdefault('channel_auto_clean',
[])
if ctx.channel.id in clean_list:
await ctx.message.delete()
async def _invasion(self, ctx, *, info=None):
"""**Usage**: `!rocket <pokestop name> [,rocket leader, pokemon names]`
Report a Team Rocket Leader invasion at a pokestop!.
Rocket Leader and their Pokemon Lineup are optional and can be updated later."""
message = ctx.message
channel = message.channel
author = message.author
guild = message.guild
info = re.split(r',+\s+', info)
stopname = info[0]
report_time = message.created_at + \
datetime.timedelta(hours=self.bot.guild_dict[guild.id]['configure_dict']['settings']['offset'])
epoch = datetime.datetime(1970, 1, 1)
report_time_int = (report_time - epoch).total_seconds()
utilities_cog = self.bot.cogs.get('Utilities')
subscriptions_cog = self.bot.cogs.get('Subscriptions')
location_matching_cog = self.bot.cogs.get('LocationMatching')
utils_cog = self.bot.cogs.get('Utilities')
regions = utils_cog.get_channel_regions(channel, 'hideout')
stops = location_matching_cog.get_stops(guild.id, regions)
stop = await location_matching_cog.match_prompt(
channel, author.id, stopname, stops)
if not stop:
self.bot.help_logger.info(
f"User: {ctx.author.name}, channel: {ctx.channel}, error: No Pokestop found {stopname}."
)
return await channel.send(embed=discord.Embed(
colour=discord.Colour.red(),
description=f"No pokestop found with name '**{stopname}**' "
f"either. Try reporting again using the exact pokestop name!"),
delete_after=15)
existing = await self._check_existing(stop)
if existing:
if existing.leader and existing.second_pokemon and existing.third_pokemon:
return await channel.send(embed=discord.Embed(
colour=discord.Colour.red(),
description=
f"Hideout for **{stop.name}** has been completely reported already."
),
delete_after=15)
if not existing:
existing = HideoutInstance(None, report_time_int, stop.id,
stop.name, None, None, None, None,
stop.latitude, stop.longitude,
message.id, message.author.id)
regions = [stop.region]
leader = None
pokemon_names = [None]
pokemon_ids = [None]
# Check through all the info sent in, pull out any leader names and pokemon names we find
if len(info) > 1:
for i in info[1:]:
# if we find a leader name then we can also set the first pokemon
# this assumption may change in the future
if i.lower() in self.leader_strings:
leader = i
p_id = self.leader_map[leader.lower()]
pkmn = Pokemon.get_pokemon(self.bot, p_id)
pokemon_names[0] = pkmn.name
pokemon_ids[0] = pkmn.id
else:
pkmn = Pokemon.get_pokemon(self.bot, i)
if pkmn is not None:
# if the name found is the one already known to be in the first slot then ignore
if pkmn.id not in self.leader_map.values():
pokemon_names.append(pkmn.name)
pokemon_ids.append(pkmn.id)
# pad the id and names lists
if len(pokemon_names) < 3:
pokemon_names = (pokemon_names + 3 * [None])[:3]
if len(pokemon_ids) < 3:
pokemon_ids = (pokemon_ids + 3 * [None])[:3]
# if there was a pre-existing report, make sure not to override leader set before
if existing.leader:
if leader and leader != existing.leader:
await channel.send(
f"**{existing.leader.capitalize()}** has already been reported for **{stop.name}**.",
delete_after=30)
leader = existing.leader
pkmn = Pokemon.get_pokemon(self.bot,
self.leader_map[leader.lower()])
pokemon_names[0] = pkmn.name
pokemon_ids[0] = pkmn.id
# don't override 2nd slot if already set
if existing.second_pokemon:
if pokemon_ids[1]:
if pokemon_ids[1] != existing.second_pokemon:
if pokemon_ids[2]:
await channel.send(
f"The second lineup slot has already been reported for **{stop.name}**.",
delete_after=30)
else:
# but if the third slot was previously not set, we can fill that in now
if not existing.third_pokemon:
existing.third_pokemon = pokemon_ids[1]
pkmn = Pokemon.get_pokemon(self.bot, existing.second_pokemon)
pokemon_names[1] = pkmn.name
pokemon_ids[1] = pkmn.id
# don't override 3rd slot if already set. this may be unreachable
if existing.third_pokemon:
if pokemon_ids[2]:
if pokemon_ids[2] != existing.second_pokemon:
await channel.send(
f"The third lineup slot has already been reported for **{stop.name}**",
delete_after=30)
pkmn = Pokemon.get_pokemon(self.bot, existing.third_pokemon)
pokemon_names[2] = pkmn.name
pokemon_ids[2] = pkmn.id
if leader:
leader = leader.lower()
if existing.id:
report = TrainerReportRelation.get_by_id(existing.id)
HideoutTable.update(rocket_leader=leader, first_pokemon=pokemon_ids[0],
second_pokemon=pokemon_ids[1], third_pokemon=pokemon_ids[2])\
.where(HideoutTable.trainer_report == report.id).execute()
updated = True
else:
report = TrainerReportRelation.create(guild=ctx.guild.id,
created=report_time_int,
trainer=author.id,
location=stop.id,
cancelled=False)
HideoutTable.create(trainer_report=report,
rocket_leader=leader,
first_pokemon=pokemon_ids[0],
second_pokemon=pokemon_ids[1],
third_pokemon=pokemon_ids[2])
updated = False
hideout = self.get_single_hideout(report.id)
inv_embed = await self._build_hideout_embed(ctx, hideout)
listmgmt_cog = self.bot.cogs.get('ListManagement')
if updated:
hideout = self.get_single_hideout(report.id)
message = await channel.fetch_message(hideout.message)
try:
await message.delete()
except:
pass
invasionreportmsg = await channel.send(
f'**Team Rocket Hideout** report at *{hideout.location_name}* updated!',
embed=inv_embed)
await utilities_cog.reaction_delay(invasionreportmsg,
['\u270f', '🚫'])
TrainerReportRelation.update(message=invasionreportmsg.id) \
.where(TrainerReportRelation.id == report.id).execute()
await listmgmt_cog.update_listing_channels(guild,
'hideout',
edit=True,
regions=regions)
else:
invasionreportmsg = await channel.send(
f'**Team Rocket Hideout** reported at *{stop.name}*',
embed=inv_embed)
await utilities_cog.reaction_delay(invasionreportmsg,
['\u270f', '🚫'])
details = {'regions': regions, 'type': 'hideout', 'location': stop}
TrainerReportRelation.update(message=invasionreportmsg.id).where(
TrainerReportRelation.id == report.id).execute()
send_channel = subscriptions_cog.get_region_list_channel(
guild, stop.region, 'invasion')
if send_channel is None:
send_channel = message.channel
await subscriptions_cog.send_notifications_async(
'hideout', details, send_channel, [message.author.id])
await asyncio.sleep(1)
await listmgmt_cog.update_listing_channels(guild,
'hideout',
edit=False,
regions=regions)
clean_list = self.bot.guild_dict[
ctx.guild.id]['configure_dict'].setdefault('channel_auto_clean',
[])
if ctx.channel.id in clean_list:
await ctx.message.delete()