Skip to content

Commit

Permalink
raid cog
Browse files Browse the repository at this point in the history
  • Loading branch information
@Baviaan
Baviaan committed Nov 30, 2019
1 parent d54af3f commit 686b17d
Show file tree
Hide file tree
Showing 3 changed files with 735 additions and 739 deletions.
178 changes: 6 additions & 172 deletions source/main.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -6,17 +6,15 @@
from discord.ext import commands
import json
import logging
import pickle

from apply_handling import new_app
from channel_handling import get_channel
from dwarves import show_dwarves
from initialise import initialise
from raid_handling import raid_command, raid_update, Tier, Time
from reaction_handling import role_update
from role_handling import show_roles

logging.basicConfig(level=logging.INFO)
logging.basicConfig(level=logging.WARNING)

# If testing it will skip 10s delay.
launch_on_boot = False
Expand All @@ -25,10 +23,6 @@
version = "v3.0.0"
print("Running " + version)

# Get local timezone using mad hacks.
local_tz = str(datetime.datetime.now(datetime.timezone.utc).astimezone().tzinfo)
print("Default timezone: " + local_tz)

# Load config file.
with open('config.json', 'r') as f:
config = json.load(f)
Expand All @@ -43,31 +37,12 @@
# Specify names for class roles.
# These will be automatically created on the server if they do not exist.
role_names = config['CLASSES']
raid_leader_name = config['LEADER']

# change to immutable tuple
role_names = tuple(role_names)

# Get server timezone
server_tz = config['SERVER_TZ']

raids = []
# Load the saved raid posts from file.
try:
with open('raids.pkl', 'rb') as f:
raids = pickle.load(f)
except (OSError, IOError) as e:
pass
print("We have the following raid data in memory.")
for raid in raids:
print(raid)


def save(raids):
with open('raids.pkl', 'wb') as f:
pickle.dump(raids, f)
print("Saved raids to file at: " + str(datetime.datetime.now()))


if launch_on_boot:
# On boot the system launches the bot faster than it gains internet access.
Expand All @@ -77,10 +52,10 @@ def save(raids):
print("Continuing...")

launch_time = None

prefix = "!"
bot = commands.Bot(command_prefix=prefix, case_insensitive=True)


def td_format(td_object):
seconds = int(td_object.total_seconds())
periods = [
Expand All @@ -92,71 +67,15 @@ def td_format(td_object):
('second', 1)
]

strings=[]
strings = []
for period_name, period_seconds in periods:
if seconds > period_seconds:
period_value , seconds = divmod(seconds, period_seconds)
period_value, seconds = divmod(seconds, period_seconds)
has_s = 's' if period_value > 1 else ''
strings.append("%s %s%s" % (period_value, period_name, has_s))

return ", ".join(strings)

async def background_task():
await bot.wait_until_ready()
sleep_time = 300 # Run background task every five minutes.
save_time = 24 # Save to file every two hours.
expiry_time = datetime.timedelta(seconds=7200) # Delete raids after 2 hours.
notify_time = datetime.timedelta(seconds=sleep_time)
counter = 0
while not bot.is_closed():
await asyncio.sleep(sleep_time)
counter = counter + 1
current_time = datetime.datetime.utcnow() # Raid time is stored in UTC.
# Copy the list to iterate over.
for raid in raids[:]:
if current_time > raid.time + expiry_time:
# Find the raid post and delete it.
channel = bot.get_channel(raid.channel_id)
try:
post = await channel.fetch_message(raid.post_id)
except discord.NotFound:
print("Raid post already deleted.")
except discord.Forbidden:
print("We are missing required permissions to delete raid post.")
except AttributeError:
print("Raid channel has been deleted.")
else:
await post.delete()
print("Deleted old raid post.")
finally:
raids.remove(raid)
print("Deleted old raid.")
elif current_time < raid.time - notify_time and current_time >= raid.time - notify_time * 2:
channel = bot.get_channel(raid.channel_id)
try:
await channel.fetch_message(raid.post_id)
except discord.NotFound:
print("Raid post already deleted.")
raids.remove(raid)
print("Deleted old raid.")
except discord.Forbidden:
print("We are missing required permissions to see raid post.")
except AttributeError:
print("Raid channel has been deleted.")
raids.remove(raid)
print("Deleted old raid.")
else:
if raid.roster:
raid_start_msg = "Gondor calls for aid! "
for player in raid.assigned_players:
if player:
raid_start_msg = raid_start_msg + "<@{0}> ".format(player.id)
raid_start_msg = raid_start_msg + "will you answer the call? We are forming for the raid now."
await channel.send(raid_start_msg, delete_after=sleep_time * 2)
if counter >= save_time:
save(raids) # Save raids to file.
counter = 0 # Reset counter to 0.


@bot.event
async def on_ready():
Expand All @@ -183,42 +102,19 @@ async def on_ready():
else:
role_post_ids.append(role_post.id)
bot.load_extension('dev_cog')
bot.load_extension('raid_cog')


@bot.event
async def on_reaction_add(reaction, user):
# Check if the reaction is by the bot itself.
if user == bot.user:
return
# Check if the reaction is to the role post.
# Check if the reaction is to the role post.
if reaction.message.id in role_post_ids:
await role_update(reaction, user, role_names)


@bot.event
async def on_raw_reaction_add(payload):
update = False
guild = bot.get_guild(payload.guild_id)
user = guild.get_member(payload.user_id)
if user == bot.user:
return
for raid in raids:
if payload.message_id == raid.post_id:
update = await raid_update(bot, payload, raid, role_names, raid_leader_name, server_tz)
emoji = payload.emoji
channel = guild.get_channel(payload.channel_id)
message = await channel.fetch_message(payload.message_id)
await message.remove_reaction(emoji, user)
break
if update:
save(raids)


@bot.event
async def on_reaction_remove(reaction, user):
pass


@bot.event
async def on_command_error(ctx, error):
print("Command given: " + ctx.message.content)
Expand Down Expand Up @@ -278,63 +174,6 @@ async def apply(ctx):
await new_app(bot, ctx.message, channel_names['APPLY'])


@bot.command(aliases=['instance', 'r'])
async def raid(ctx, name, tier: Tier, *, time: Time(server_tz)):
"""Schedules a raid"""
raid = await raid_command(ctx, name, tier, "All", time, role_names, server_tz)
raids.append(raid)
save(raids)


raid_brief = "Schedules a raid"
raid_description = "Schedules a raid. Day/timezone will default to today/{0} if not specified. " \
"You can use 'server' as timezone. Usage:".format(local_tz)
raid_example = "Examples:\n!raid Anvil 2 Friday 4pm server\n!raid throne t3 21:00"
raid.update(help=raid_example, brief=raid_brief, description=raid_description)


@bot.command()
async def anvil(ctx, *, time: Time(server_tz)):
"""Shortcut to schedule Anvil raid"""
try:
tier = await Tier().converter(ctx.channel.name)
except commands.BadArgument:
await ctx.send("Channel name does not specify tier.")
else:
if '1' in tier or '2' in tier:
roster = False
else:
roster = True
raid = await raid_command(ctx, "Anvil", tier, "All", time, role_names, server_tz, roster=roster)
raids.append(raid)
save(raids)


anvil_brief = "Shortcut to schedule an Anvil raid"
anvil_description = "Schedules a raid with name 'Anvil', tier from channel name and bosses 'All'. " \
"Day/timezone will default to today/{0} if not specified. You can use 'server' as timezone. " \
"Usage:".format(local_tz)
anvil_example = "Examples:\n!anvil Friday 4pm server\n!anvil 21:00 BST"
anvil.update(help=anvil_example, brief=anvil_brief, description=anvil_description)


@bot.command()
async def thrang(ctx, *, time: Time(server_tz)):
"""Shortcut to schedule Thrang run"""
tier = 'T2'
raid = await raid_command(ctx, "Boss from the Vaults", tier, "Thrang", time, role_names, server_tz)
raids.append(raid)
save(raids)


thrang_brief = "Shortcut to schedule a Thrang run"
thrang_description = "Schedules a raid with name 'Boss from the Vaults', tier 2 and boss 'Thrang'. " \
"Day/timezone will default to today/{0} if not specified. You can use 'server' as timezone. " \
"Usage: ".format(local_tz)
thrang_example = "Examples:\n!thrang Friday 4pm server\n!thrang 21:00 BST"
thrang.update(help=thrang_example, brief=thrang_brief, description=thrang_description)


@bot.command()
@commands.is_owner()
async def delete(ctx, msg_id: int):
Expand All @@ -344,7 +183,6 @@ async def delete(ctx, msg_id: int):
await asyncio.sleep(0.25)
await msg.delete()


delete.update(hidden=True)


Expand All @@ -354,9 +192,5 @@ async def delete_error(ctx, error):
ctx.send("You do not have permission to use this command.")


bot.loop.create_task(background_task())
bot.run(token)

# Save raids if bot unexpectedly closes.
save(raids)
print("Shutting down.")
Loading

0 comments on commit 686b17d

Please sign in to comment.