316 lines
11 KiB
Python
316 lines
11 KiB
Python
import random
|
|
import re
|
|
import string
|
|
import json
|
|
import os
|
|
|
|
from . import swchar
|
|
from funcs import logThis
|
|
|
|
with open("resources/swskills.json", "r") as f:
|
|
skillData = json.load(f)
|
|
|
|
def roll(abi : int = 1, prof : int = 0, dif : int = 3, cha : int = 0, boo : int = 0, setb : int = 0, force : int = 0):
|
|
result = ""
|
|
diceResult = []
|
|
for x in range(abi):
|
|
choice = random.choice(["","S","S","SS","A","A","SA","AA"])
|
|
result += choice
|
|
diceResult.append("abi"+choice)
|
|
|
|
for x in range(prof):
|
|
choice = random.choice(["","S","S","SS","SS","A","SA","SA","SA","AA","AA","R"])
|
|
result += choice
|
|
diceResult.append("prof"+choice)
|
|
|
|
for x in range(dif):
|
|
choice = random.choice(["","F","FF","H","H","H","HH","FH"])
|
|
result += choice
|
|
diceResult.append("dif"+choice)
|
|
|
|
for x in range(cha):
|
|
choice = random.choice(["","F","F","FF","FF","H","H","FH","FH","HH","HH","D"])
|
|
result += choice
|
|
diceResult.append("cha"+choice)
|
|
|
|
for x in range(boo):
|
|
choice = random.choice(["","","S","SA","AA","A"])
|
|
result += choice
|
|
diceResult.append("boo"+choice)
|
|
|
|
for x in range(setb):
|
|
choice = random.choice(["","","F","F","H","H"])
|
|
result += choice
|
|
diceResult.append("setb"+choice)
|
|
|
|
for x in range (force):
|
|
choice = random.choice(["B","B","B","B","B","B","BB","L","L","LL","LL","LL"])
|
|
result += choice
|
|
diceResult.append("force"+choice)
|
|
|
|
return result, diceResult
|
|
|
|
def simplify(result : str):
|
|
logThis("Simplifying "+result)
|
|
simp = ""
|
|
success = (result.count('S') + result.count('R')) - (result.count('F') + result.count('D'))
|
|
advantage = result.count('A') - result.count('H')
|
|
result = re.sub("S|A|F|H","",result)
|
|
|
|
if success > 0:
|
|
for x in range(success):
|
|
simp += "S"
|
|
elif success < 0:
|
|
for x in range(abs(success)):
|
|
simp += "F"
|
|
|
|
if advantage > 0:
|
|
for x in range(advantage):
|
|
simp += "A"
|
|
elif advantage < 0:
|
|
for x in range(abs(advantage)):
|
|
simp += "H"
|
|
|
|
simp += result
|
|
|
|
return simp
|
|
|
|
def diceResultToEmoji(diceResults : list):
|
|
emoji = ""
|
|
for result in diceResults:
|
|
if result == "abiA":
|
|
emoji += "<:abil1a:695267684476125264> "
|
|
if result == "abiSA":
|
|
emoji += "<:abil1a1s:695267684484513842> "
|
|
if result == "abiS":
|
|
emoji += "<:abil1s:695267684514005013> "
|
|
if result == "abiAA":
|
|
emoji += "<:abil2a:695267684547428352> "
|
|
if result == "abiSS":
|
|
emoji += "<:abil2s:695267684761206914> "
|
|
if result == "abi":
|
|
emoji += "<:abilbla:695267684660674602> "
|
|
|
|
if result == "profA":
|
|
emoji += "<:prof1a:695267685361123338> "
|
|
if result == "profSA":
|
|
emoji += "<:prof1a1s:695267685067653140> "
|
|
if result == "profR":
|
|
emoji += "<:prof1r:695267685067522088> "
|
|
if result == "profS":
|
|
emoji += "<:prof1s:695267684899881012> "
|
|
if result == "profAA":
|
|
emoji += "<:prof2a:695267684996218982> "
|
|
if result == "profSS":
|
|
emoji += "<:prof2s:695267684878647327> "
|
|
if result == "prof":
|
|
emoji += "<:profbla:695267684698292235> "
|
|
|
|
if result == "difF":
|
|
emoji += "<:dif1f:695267684924915804> "
|
|
if result == "difH":
|
|
emoji += "<:dif1h:695267684908138506> "
|
|
if result == "difFH":
|
|
emoji += "<:dif1h1f:695267684908269678> "
|
|
if result == "difFF":
|
|
emoji += "<:dif2f:695267684924784680> "
|
|
if result == "difHH":
|
|
emoji += "<:dif2h:695267685071585340> "
|
|
if result == "dif":
|
|
emoji += "<:difbla:695267685000544276> "
|
|
|
|
if result == "chaD":
|
|
emoji += "<:cha1d:695267684962533447> "
|
|
if result == "chaF":
|
|
emoji += "<:cha1f:695267684601954346> "
|
|
if result == "chaH":
|
|
emoji += "<:cha1h:695267685046681620> "
|
|
if result == "chaFH":
|
|
emoji += "<:cha1h1f:695267685063327784> "
|
|
if result == "chaFF":
|
|
emoji += "<:cha2f:695267684832641097> "
|
|
if result == "chaHH":
|
|
emoji += "<:cha2h:695267684631183381> "
|
|
if result == "cha":
|
|
emoji += "<:chabla:695267684895686787> "
|
|
|
|
if result == "booA":
|
|
emoji += "<:boo1a:695267684975116329> "
|
|
if result == "booSA":
|
|
emoji += "<:boo1a1s:695267684970922024> "
|
|
if result == "booS":
|
|
emoji += "<:boo1s:695267684979441714> "
|
|
if result == "booAA":
|
|
emoji += "<:boo2a:695267685100945488> "
|
|
if result == "boo":
|
|
emoji += "<:boobla:695267684757012550> "
|
|
|
|
if result == "setbF":
|
|
emoji += "<:set1f:695267685054939197> "
|
|
if result == "setbH":
|
|
emoji += "<:set1h:695267685147082802> "
|
|
if result == "setb":
|
|
emoji += "<:setbla:695267685151408169> "
|
|
|
|
if result == "forceB":
|
|
emoji += "<:for1b:695267684593434677> "
|
|
if result == "forceL":
|
|
emoji += "<:for1l:695267684606148640> "
|
|
if result == "forceBB":
|
|
emoji += "<:for2b:695267684903944303> "
|
|
if result == "forceLL":
|
|
emoji += "<:for2l:695267684992024626> "
|
|
|
|
return emoji
|
|
|
|
def resultToEmoji(result : str):
|
|
emoji = ""
|
|
for char in result:
|
|
if char == 'S':
|
|
emoji += "<:success:690971244971163718> "
|
|
if char == 'A':
|
|
emoji += "<:advantage:690970761611051079> "
|
|
if char == 'R':
|
|
emoji += "<:swtriumph:690971267486187643> "
|
|
if char == 'F':
|
|
emoji += "<:failure:690970957786906664> "
|
|
if char == 'H':
|
|
emoji += "<:threat:690971009469382656> "
|
|
if char == 'D':
|
|
emoji += "<:despair:690971200163414238> "
|
|
if char == 'L':
|
|
emoji += "<:light:691010089905029171>"
|
|
if char == 'B':
|
|
emoji += "<:dark:691010101901000852>"
|
|
|
|
return emoji
|
|
|
|
def emojiToResult(emoji : str):
|
|
result = ""
|
|
for char in emoji:
|
|
if char == "<:light:691010089905029171>":
|
|
emoji += 'L'
|
|
if char == "<:dark:691010101901000852>":
|
|
emoji += 'B'
|
|
|
|
return result
|
|
|
|
def diceToEmoji(dice : list):
|
|
emoji = ""
|
|
|
|
for x in range(dice[0]):
|
|
emoji += "<:ability:690974213397282826> "
|
|
for x in range(dice[1]):
|
|
emoji += "<:proficiency:690973435354153071> "
|
|
for x in range(dice[2]):
|
|
emoji += "<:difficulty:690973992470708296> "
|
|
for x in range(dice[3]):
|
|
emoji += "<:challenge:690973419906400306> "
|
|
for x in range(dice[4]):
|
|
emoji += "<:boost:690972178216386561> "
|
|
for x in range(dice[5]):
|
|
emoji += "<:setback:690972157890658415> "
|
|
for x in range(dice[6]):
|
|
emoji += "<:force:690973451883774013> "
|
|
|
|
return emoji
|
|
|
|
def obligationRoll():
|
|
logThis("Rolling for obligation")
|
|
with open("resources/swcharacters.json", "r") as f:
|
|
data = json.load(f)
|
|
|
|
table = []
|
|
|
|
for character in data:
|
|
for obligation in data[character]["Obligations"]:
|
|
for x in range(data[character]["Obligations"][obligation]):
|
|
table.append(data[character]["Name"]+", "+obligation)
|
|
|
|
while len(table) < 100:
|
|
table.append("Nothing")
|
|
|
|
return random.choice(table)
|
|
|
|
def parseRoll(user : str,cmd : str = ""):
|
|
cmd = re.sub(' +',' ',cmd.upper()) + " "
|
|
if cmd[0] == " ":
|
|
cmd = cmd[1:]
|
|
cmd = swchar.replaceSpaces(cmd.capitalize())
|
|
commands = cmd.split(" ")
|
|
if commands[0] == "":
|
|
rollParameters = [1,0,3,0,0,0,0]
|
|
else:
|
|
rollParameters = [0,0,0,0,0,0,0]
|
|
|
|
if string.capwords(commands[0]) == "Obligations":
|
|
return obligationRoll()
|
|
|
|
elif string.capwords(commands[0]) in skillData:
|
|
logThis("Oh look! This guy has skills!")
|
|
if swchar.userHasChar:
|
|
logThis("They have a character. That much we know")
|
|
skillLevel = swchar.charData(user,"Skills " + string.capwords(commands[0]))
|
|
|
|
if string.capwords(commands[0]) == "Lightsaber":
|
|
logThis("The skill is lightsaber")
|
|
charLevel = swchar.charData(user,"Characteristics " + swchar.lightsaberChar(user))
|
|
else:
|
|
charLevel = swchar.charData(user,"Characteristics " + skillData[string.capwords(commands[0])])
|
|
|
|
abilityDice = abs(charLevel-skillLevel)
|
|
proficiencyDice = min(skillLevel,charLevel)
|
|
|
|
commands = [str(abilityDice)] + [str(proficiencyDice)] + commands[1:]
|
|
logThis("Converted skill to dice")
|
|
else:
|
|
logThis("Okay, no they don't i guess")
|
|
return "You don't have a user. You can make one with !swchar"
|
|
|
|
elif string.capwords(commands[0]) in ["Ranged","Piloting"]:
|
|
logThis("They fucked up writing the name of a ranged or piloting skill")
|
|
if string.capwords(commands[0]) == "Ranged":
|
|
return "Did you mean \"Ranged - Heavy\" or \"Ranged - Light\""
|
|
else:
|
|
return "Did you mean \"Piloting - Planetary\" or \"Piloting - Space\""
|
|
|
|
try:
|
|
logThis("Converting commands to dice")
|
|
for x in range(len(commands)-1):
|
|
if commands[x] != "":
|
|
commands[x] = commands[x].upper()
|
|
if commands[x][0] == "A":
|
|
rollParameters[0] = int(commands[x].replace("A",""))
|
|
elif commands[x][0] == "P":
|
|
rollParameters[1] = int(commands[x].replace("P",""))
|
|
elif commands[x][0] == "D":
|
|
rollParameters[2] = int(commands[x].replace("D",""))
|
|
elif commands[x][0] == "C":
|
|
rollParameters[3] = int(commands[x].replace("C",""))
|
|
elif commands[x][0] == "B":
|
|
rollParameters[4] = int(commands[x].replace("B",""))
|
|
elif commands[x][0] == "S":
|
|
rollParameters[5] = int(commands[x].replace("S",""))
|
|
elif commands[x][0] == "F":
|
|
rollParameters[6] = int(commands[x].replace("F",""))
|
|
else:
|
|
rollParameters[x] = int(commands[x])
|
|
except:
|
|
logThis("Someone fucked u-up! (it was them)")
|
|
return "Invalid input!"
|
|
|
|
logThis("Rolling "+str(rollParameters))
|
|
rollResults, diceResults = roll(rollParameters[0],rollParameters[1],rollParameters[2],rollParameters[3],rollParameters[4],rollParameters[5],rollParameters[6])
|
|
|
|
simplified = simplify(rollResults)
|
|
|
|
name = swchar.getName(user)
|
|
|
|
logThis("Returns results and simplified results")
|
|
|
|
if simplified == "":
|
|
return name + " rolls: " + "\n" + diceResultToEmoji(diceResults) + "\nEverything cancels out!"
|
|
else:
|
|
return name + " rolls: " + "\n" + diceResultToEmoji(diceResults) + "\n" + resultToEmoji(simplified)
|
|
|