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.
479 lines
16 KiB
479 lines
16 KiB
import discord
|
|
import os
|
|
import math
|
|
from discord.ext import commands
|
|
|
|
from .utils.dataIO import dataIO
|
|
from .utils import checks
|
|
from random import randint
|
|
|
|
|
|
class Timezone:
|
|
"""Cog for organizing fights"""
|
|
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
self.path = "data/Fox-Cogs/fight/"
|
|
self.file_path = "data/Fox-Cogs/fight/fight.json"
|
|
self.the_data = dataIO.load_json(self.file_path)
|
|
|
|
def save_data(self):
|
|
"""Saves the json"""
|
|
dataIO.save_json(self.file_path, self.the_data)
|
|
|
|
|
|
# ************************Fight command group start************************
|
|
@commands.group(pass_context=True, no_pm=True)
|
|
async def fight(self, ctx):
|
|
"""Participate in active fights!"""
|
|
server = ctx.message.server
|
|
|
|
if not self._activefight(server.id):
|
|
await self.bot.say("No tournament currently running!")
|
|
else:
|
|
await self.bot.say("Current tournament ID: " + self.the_data[server.id]["CURRENT"])
|
|
|
|
if ctx.invoked_subcommand is None:
|
|
await self.bot.send_cmd_help(ctx)
|
|
# await self.bot.say("I can do stuff!")
|
|
|
|
@fight.command(name="join", pass_context=True)
|
|
async def fight_join(self, ctx, user: discord.Member=None):
|
|
"""Join the active fight"""
|
|
server = ctx.message.server
|
|
if not user:
|
|
user = ctx.message.author
|
|
|
|
currFight = self._getcurrentfight(server.id)
|
|
currID = self._activefight(server.id)
|
|
if not currFight:
|
|
await self.bot.say("No tournament currently running!")
|
|
return
|
|
|
|
if not currFight["OPEN"]:
|
|
await self.bot.say("Tournament currently not accepting new players")
|
|
return
|
|
|
|
if self._infight(user.id, server.id, currID):
|
|
await self.bot.say("You are already in this tournament!")
|
|
return
|
|
|
|
currFight["PLAYERS"].append(user.id)
|
|
|
|
await self.bot.say("User has been added to tournament")
|
|
|
|
@fight.command(name="score", pass_context=True)
|
|
async def fight_score(self, ctx, gameID):
|
|
"""Enters score for current match, or for passed game ID"""
|
|
if gameID is None:
|
|
if ctx.message.author not in currFight["PLAYERS"]:
|
|
await self.bot.say("You are not in a current tournament")
|
|
|
|
await self.bot.say("Todo Score")
|
|
|
|
@fight.command(name="leave", pass_context=True)
|
|
async def fight_leave(self, ctx):
|
|
"""Forfeit your match and all future matches"""
|
|
await self.bot.say("Todo Leave")
|
|
|
|
# @fight.command(name="leaderboard", pass_context=True)
|
|
# async def fight_leaderboard(self, ctx, ctag, ckind="Unranked", irank=0):
|
|
# """Adds clan to grab-list"""
|
|
# await self.bot.say("Todo Leaderboard")
|
|
|
|
@fight.group(name="bracket", pass_context=True)
|
|
async def fight_bracket(self, ctx, ctag):
|
|
"""Shows your current match your next opponent,
|
|
run [p]fight bracket full to see all matches"""
|
|
await self.bot.say("Todo Bracket")
|
|
|
|
# Your code will go here
|
|
if ctx.invoked_subcommand is None:
|
|
await self.bot.send_cmd_help(ctx)
|
|
# await self.bot.say("I can do stuff!")
|
|
|
|
@fight_bracket.command(name="full")
|
|
async def fight_bracket_full(self, ctag):
|
|
"""Shows the full bracket"""
|
|
await self.bot.say("Todo Bracket Full")
|
|
|
|
# **********************Fightset command group start*********************
|
|
# def fightsetdec(func):
|
|
# async def decorated(self, ctx, *args, **kwargs):
|
|
# server = ctx.message.server
|
|
# await func(self, ctx, server, *args, **kwargs)
|
|
# return decorated
|
|
|
|
@commands.group(pass_context=True, no_pm=True, aliases=['setfight'])
|
|
@checks.mod_or_permissions(administrator=True)
|
|
async def fightset(self, ctx):
|
|
"""Admin command for starting or managing tournaments"""
|
|
server = ctx.message.server
|
|
|
|
if server.id not in self.the_data:
|
|
self.the_data[server.id] = {
|
|
"CURRENT": None,
|
|
"TOURNEYS": {}
|
|
}
|
|
self.save_data()
|
|
|
|
if ctx.invoked_subcommand is None:
|
|
await self.bot.send_cmd_help(ctx)
|
|
# await self.bot.say("I can do stuff!")
|
|
|
|
@fightset.command(name="bestof", pass_context=True)
|
|
async def fightset_bestof(self, ctx, incount, tID=None):
|
|
"""Adjust # of games played per match. Must be an odd number"""
|
|
server = ctx.message.server
|
|
if not tID and not self._activefight(server.id):
|
|
await self.bot.say("No active fight to adjust")
|
|
return
|
|
|
|
if not tID:
|
|
tID = self._activefight(server.id)
|
|
|
|
try:
|
|
num = int(incount)
|
|
except:
|
|
await self.bot.say("That is not a number")
|
|
return
|
|
|
|
if num % 2 != 1:
|
|
await self.bot.say("Must be an odd number")
|
|
return
|
|
|
|
if num < 1:
|
|
await self.bot.say("Must be greater than 0, idiot")
|
|
return
|
|
|
|
self._getfight(server.id, tID)["RULES"]["BESTOF"] = num
|
|
self.save_data()
|
|
await self.bot.say("Tourney ID "+tID+" is now Best of "+str(num))
|
|
|
|
@fightset.command(name="bestoffinal", pass_context=True)
|
|
async def fightset_bestoffinal(self, ctx, incount, tID=None):
|
|
"""Adjust # of games played in finals. Must be an odd number
|
|
(Does not apply to tournament types without finals, such as Round Robin)"""
|
|
server = ctx.message.server
|
|
if not tID and not self._activefight(server.id):
|
|
await self.bot.say("No active fight to adjust")
|
|
return
|
|
|
|
if not tID:
|
|
tID = self._activefight(server.id)
|
|
|
|
try:
|
|
num = int(incount)
|
|
except:
|
|
await self.bot.say("That is not a number")
|
|
return
|
|
|
|
if num % 2 != 1:
|
|
await self.bot.say("Must be an odd number")
|
|
return
|
|
|
|
if num < 1:
|
|
await self.bot.say("Must be greater than 0, idiot")
|
|
return
|
|
|
|
self._getfight(server.id, tID)["RULES"]["BESTOFFINAL"] = num
|
|
self.save_data()
|
|
await self.bot.say("Tourney ID "+tID+" is now Best of "+str(num))
|
|
|
|
@fightset.command(name="current", pass_context=True)
|
|
async def fightset_current(self, ctx, tID):
|
|
"""Sets the current tournament to passed ID"""
|
|
server = ctx.message.server
|
|
try:
|
|
aFight = self.the_data[server.id]["TOURNEYS"][tID]
|
|
|
|
except:
|
|
await self.bot.say("No tourney found with that ID")
|
|
|
|
self.the_data[server.id]["CURRENT"] = tID
|
|
self.save_data()
|
|
|
|
await self.bot.say("Current tournament set to "+tID)
|
|
|
|
@fightset.command(name="list", pass_context=True)
|
|
async def fightset_list(self, ctx):
|
|
"""Lists all current and past fights"""
|
|
server = ctx.message.server
|
|
|
|
await self.bot.say(self.the_data[server.id]["TOURNEYS"])
|
|
await self.bot.say("Done")
|
|
|
|
@fightset.command(name="toggleopen", pass_context=True)
|
|
async def fightset_toggleopen(self, ctx):
|
|
"""Toggles the open status of current tournament"""
|
|
server = ctx.message.server
|
|
if not self._activefight(server.id):
|
|
await self.bot.say("No active fight to adjust")
|
|
return
|
|
|
|
currFight = self._getcurrentfight(server.id)
|
|
currFight["OPEN"] = not currFight["OPEN"]
|
|
|
|
self.save_data()
|
|
|
|
await self.bot.say("Tournament Open status is now set to: " + str(currFight["OPEN"]))
|
|
|
|
@fightset.command(name="setup", pass_context=True)
|
|
async def fightset_setup(self, ctx):
|
|
"""Setup a new tournament!
|
|
Default settings are as follows
|
|
Name: Tourney # (counts from 0)
|
|
Best of: 1
|
|
Best of (final): 1
|
|
Self Report: True
|
|
Type: 0 (Round Robin)"""
|
|
server = ctx.message.server
|
|
currServ = self.the_data[server.id]
|
|
tourID = str(len(currServ["TOURNEYS"])) # Can just be len without +1, tourney 0 makes len 1, tourney 1 makes len 2, etc
|
|
currServ["CURRENT"] = tourID
|
|
currServ["TOURNEYS"][tourID] = {
|
|
"PLAYERS": [],
|
|
"NAME": "Tourney "+str(tourID),
|
|
"RULES": {"BESTOF": 1, "BESTOFFINAL": 1, "SELFREPORT": True, "TYPE": 0},
|
|
"TYPEDATA": {},
|
|
"OPEN": False,
|
|
"WINNER": None
|
|
}
|
|
|
|
self.save_data()
|
|
|
|
await self.bot.say("Tournament has been created!\n\n" + str(currServ["TOURNEYS"][tourID]))
|
|
|
|
await self.bot.say("Adjust settings as necessary, then open the tournament with [p]fightset toggleopen")
|
|
|
|
@fightset.command(name="stop", pass_context=True)
|
|
async def fightset_stop(self, ctx):
|
|
"""Stops current tournament"""
|
|
server = ctx.message.server
|
|
if not self._activefight(server.id):
|
|
await self.bot.say("No active fight to adjust")
|
|
return
|
|
|
|
author = ctx.message.author
|
|
currServ = self.the_data[server.id]
|
|
|
|
await self.bot.say("Current fight ID is "+str(currServ["CURRENT"])+"\nOkay to stop? (yes/no)")
|
|
|
|
answer = await self.bot.wait_for_message(timeout=120, author=author)
|
|
|
|
if not answer.content.upper() in ["YES", "Y"]:
|
|
await self.bot.say("Cancelled")
|
|
return
|
|
|
|
currServ["CURRENT"] = None
|
|
|
|
self.save_data()
|
|
await self.bot.say("Fight has been stopped")
|
|
|
|
# **********************Private command group start*********************
|
|
def _activefight(self, serverid):
|
|
"""Returns id for active fight, or None if no active fight"""
|
|
return self.the_data[serverid]["CURRENT"]
|
|
|
|
def _infight(self, userid, serverid, tID):
|
|
"""Checks if passed member is already in the tournament"""
|
|
|
|
return userid in self.the_data[serverid]["TOURNEYS"][tID]["PLAYERS"]
|
|
|
|
async def _openregistration(self, serverid, tID):
|
|
"""Checks if fight is accepting joins"""
|
|
await self.bot.say("_openregistration Todo")
|
|
|
|
async def _comparescores(self):
|
|
"""Checks user submitted scores for inconsistancies"""
|
|
await self.bot.say("_comparescores Todo")
|
|
|
|
async def _parsemember(self):
|
|
await self.bot.say("Parsemember Todo")
|
|
|
|
def _get_user_from_id(self, server, userid):
|
|
return discord.utils.get(server.members, id=userid)
|
|
|
|
def _get_server_from_id(self, serverid):
|
|
return discord.utils.get(self.bot.servers, id=serverid)
|
|
|
|
def _getfight(self, serverid, tID):
|
|
return self.the_data[serverid]["TOURNEYS"][tID]
|
|
|
|
def _getcurrentfight(self, serverid):
|
|
if not self._activefight(serverid):
|
|
return None
|
|
|
|
return self._getfight(serverid, self._activefight(serverid))
|
|
|
|
# **********************Single Elimination***************************
|
|
async def _elim_setup(self, tID):
|
|
await self.bot.say("Elim setup todo")
|
|
|
|
async def _elim_start(self, tID):
|
|
await self.bot.say("Elim start todo")
|
|
|
|
async def _elim_update(self, matchID, ):
|
|
await self.bot.say("Elim update todo")
|
|
|
|
|
|
# **********************Round-Robin**********************************
|
|
def _rr_setup(self, serverid, tID):
|
|
|
|
theT = self.the_data[serverid]["TOURNEYS"][tID]
|
|
theD = theT["TYPEDATA"]
|
|
|
|
get_schedule = self._rr_schedule(theT["PLAYERS"])
|
|
|
|
theD = {"SCHEDULE": get_schedule[0], "MATCHES": get_schedule[1], "ROUND": 0}
|
|
|
|
self.save_data()
|
|
|
|
async def _rr_printround(self, serverid, tID, rID):
|
|
|
|
theT = self.the_data[serverid]["TOURNEYS"][tID]
|
|
theD = theT["TYPEDATA"]
|
|
|
|
await self.bot.say("Round "+str(rID))
|
|
|
|
for match in theD["SCHEDULE"][rID]:
|
|
await self.bot.say(theD["MATCHES"][match]["TEAM1"] + " vs " + theD["MATCHES"][match]["TEAM2"] + " || Match ID: " + match)
|
|
|
|
async def _rr_start(self, serverid, tID):
|
|
|
|
self._rr_setup(serverid, tID)
|
|
|
|
await self.bot.say("**Tournament is Starting**")
|
|
|
|
await self._rr_printround(tID)
|
|
|
|
async def _rr_update(self, serverid, tID=None, mID=None, t1points=None, t2points=None):
|
|
|
|
theT = self.the_data[serverid]["TOURNEYS"][tID]
|
|
theD = theT["TYPEDATA"]
|
|
|
|
if t1points and t2points:
|
|
theD["MATCHES"][mID]["TEAM1"] = t1points
|
|
theD["MATCHES"][mID]["TEAM2"] = t2points
|
|
self.save_data()
|
|
return
|
|
|
|
await self.bot.say("Entering scores for match ID: " + mID + "\n\n")
|
|
await self.bot.say("How many points did " + theD["MATCHES"][mID]["TEAM1"] + " get?")
|
|
answer = await self.bot.wait_for_message(timeout=120, author=author)
|
|
try:
|
|
t1points = int(answer.content)
|
|
except:
|
|
await self.bot.say("That's not a number!")
|
|
return
|
|
|
|
await self.bot.say("How many points did " + theD["MATCHES"][mID]["TEAM2"] + " get?")
|
|
answer = await self.bot.wait_for_message(timeout=120, author=author)
|
|
try:
|
|
t2points = int(answer.content)
|
|
except:
|
|
await self.bot.say("That's not a number!")
|
|
return
|
|
|
|
if t1points == math.ceil(theD["RULES"]["BESTOF"]/2) or t2points == math.ceil(theD["RULES"]["BESTOF"]/2):
|
|
theD["MATCHES"][mID]["TEAM1"] = t1points
|
|
theD["MATCHES"][mID]["TEAM2"] = t2points
|
|
self.save_data()
|
|
return
|
|
else:
|
|
await self.bot.say("Invalid scores, nothing will be updated")
|
|
return
|
|
|
|
def _rr_schedule(inlist):
|
|
""" Create a schedule for the teams in the list and return it"""
|
|
s = [] # Schedule list
|
|
outID = {} # Matches
|
|
|
|
firstID = ["A", "B", "C", "D", "E", "F",
|
|
"G", "H", "I", "J", "K", "L",
|
|
"M", "N", "O", "P", "Q", "R",
|
|
"S", "T", "U", "V", "W", "X",
|
|
"Y", "Z"] # God dammit this could've been a string
|
|
|
|
if len(inlist) % 2 == 1:
|
|
inlist = inlist + ["BYE"]
|
|
|
|
for i in range(len(inlist)-1):
|
|
|
|
mid = int(len(inlist) / 2)
|
|
l1 = inlist[:mid]
|
|
l2 = inlist[mid:]
|
|
l2.reverse()
|
|
|
|
matchLetter = ""
|
|
j = i
|
|
while j+1 > 26:
|
|
|
|
matchLetter += firstID[int(j + 1) % 26 - 1]
|
|
|
|
j = (j + 1) / 26 - 1
|
|
matchLetter += firstID[int(j+1) % 26-1]
|
|
matchLetter = matchLetter[::-1]
|
|
|
|
matchID = []
|
|
for ix in range(len(l1)-1):
|
|
matchID += [matchLetter+str(ix)]
|
|
|
|
rPlayers = list(zip(l1, l2))
|
|
TeamCnt = 0
|
|
for ID in matchID:
|
|
outID[ID] = {
|
|
"TEAM1": rPlayers[TeamCnt][0],
|
|
"TEAM2": rPlayers[TeamCnt][1],
|
|
"SCORE1": 0,
|
|
"SCORE2": 0,
|
|
"USERSCORE1": {"SCORE1": 0, "SCORE2": 0},
|
|
"USERSCORE2": {"SCORE1": 0, "SCORE2": 0}
|
|
}
|
|
|
|
TeamCnt += 1
|
|
|
|
# List of match ID's is now done
|
|
|
|
s += [matchID] # Schedule of matches
|
|
inlist.insert(1, inlist.pop())
|
|
|
|
outlist = [[], {}]
|
|
outlist[0] = s
|
|
outlist[1] = outID
|
|
# outlist[0] is list schedule of matches
|
|
# outlist[1] is dict data of matches
|
|
|
|
return outlist
|
|
|
|
def _rr_nextround(self, serverid, tID):
|
|
currFight = self._getfight(serverid, tID)
|
|
currRound = currFight["TYPEDATA"]["SCHEDULE"][currFight["TYPEDATA"]["ROUND"]]
|
|
|
|
for match in currRound:
|
|
if (currFight["TYPEDATA"]["MATCHES"][match]["SCORE1"] > currFight["RULES"]["BESTOF"]/2 or
|
|
currFight["TYPEDATA"]["MATCHES"][match]["SCORE2"] > currFight["RULES"]["BESTOF"]/2):
|
|
return False
|
|
return True
|
|
|
|
|
|
def check_folders():
|
|
if not os.path.exists("data/Fox-Cogs"):
|
|
print("Creating data/Fox-Cogs folder...")
|
|
os.makedirs("data/Fox-Cogs")
|
|
|
|
if not os.path.exists("data/Fox-Cogs/fight"):
|
|
print("Creating data/Fox-Cogs/fight folder...")
|
|
os.makedirs("data/Fox-Cogs/fight")
|
|
|
|
|
|
def check_files():
|
|
if not dataIO.is_valid_json("data/Fox-Cogs/fight/fight.json"):
|
|
dataIO.save_json("data/Fox-Cogs/fight/fight.json", {})
|
|
|
|
|
|
def setup(bot):
|
|
check_folders()
|
|
check_files()
|
|
bot.add_cog(Fight(bot))
|