changed pokemon game to mostly be a cog and commands
This commit is contained in:
phixxy
parent
022d9afbee
commit
9e1624eb6b
1 changed files with 171 additions and 180 deletions
|
|
@ -12,6 +12,8 @@ async def get_json(self, url):
|
||||||
json_data = await resp.json()
|
json_data = await resp.json()
|
||||||
return json_data
|
return json_data
|
||||||
|
|
||||||
|
#@commands.cooldown(1, 60, commands.BucketType.user) Cooldowns....
|
||||||
|
|
||||||
class Pokemon:
|
class Pokemon:
|
||||||
async def __init__(self, id: int, generate: bool = True) -> None:
|
async def __init__(self, id: int, generate: bool = True) -> None:
|
||||||
if generate: # Should I even do it this way?
|
if generate: # Should I even do it this way?
|
||||||
|
|
@ -39,193 +41,182 @@ class PokemonGame(commands.Cog):
|
||||||
except:
|
except:
|
||||||
self.bot.logger.exception("PokemonGame failed to make directories")
|
self.bot.logger.exception("PokemonGame failed to make directories")
|
||||||
|
|
||||||
@commands.command(
|
async def starter_picker(self, id): #id = pokedex number
|
||||||
description="Pokemon",
|
url = "https://pokeapi.co/api/v2/pokemon-species/" + str(id)
|
||||||
help="Pokemon game",
|
json_data = await get_json(url)
|
||||||
brief="Pokemon Game",
|
if (json_data["evolves_from_species"] == None) and (not json_data['is_mythical']) and (not json_data['is_legendary']):
|
||||||
aliases=['pkmn'],
|
|
||||||
hidden=True
|
|
||||||
)
|
|
||||||
async def pokemon(self, ctx, *args):
|
|
||||||
async def starter_picker(id): #id = pokedex number
|
|
||||||
url = "https://pokeapi.co/api/v2/pokemon-species/" + str(id)
|
|
||||||
json_data = await get_json(url)
|
|
||||||
if (json_data["evolves_from_species"] == None) and (not json_data['is_mythical']) and (not json_data['is_legendary']):
|
|
||||||
return True
|
|
||||||
else:
|
|
||||||
return False
|
|
||||||
|
|
||||||
async def shiny_roll():
|
|
||||||
roll = random.randint(0,2047)
|
|
||||||
return not roll
|
|
||||||
|
|
||||||
async def save_pokemon(discord_id, pokemon_dict):
|
|
||||||
path = self.data_dir+str(discord_id)+".json"
|
|
||||||
pokemon_dict = json.dumps(pokemon_dict)
|
|
||||||
with open(path, 'w') as f:
|
|
||||||
f.writelines(pokemon_dict)
|
|
||||||
return True
|
return True
|
||||||
|
else:
|
||||||
async def load_pokemon(discord_id):
|
return False
|
||||||
if os.path.isfile(self.data_dir+str(discord_id)+".json"):
|
|
||||||
with open(self.data_dir+str(discord_id)+".json", 'r') as f:
|
async def shiny_roll(self):
|
||||||
json_data = json.loads(f.readline())
|
roll = random.randint(0,2047)
|
||||||
return json_data
|
return not roll
|
||||||
else:
|
|
||||||
return False
|
async def save_pokemon(self, discord_id, pokemon_dict):
|
||||||
|
path = self.data_dir+str(discord_id)+".json"
|
||||||
async def generate_starter(discord_id):
|
pokemon_dict = json.dumps(pokemon_dict)
|
||||||
random.seed(discord_id)
|
with open(path, 'w') as f:
|
||||||
json_data = await get_json('https://pokeapi.co/api/v2/pokemon-species/')
|
f.writelines(pokemon_dict)
|
||||||
pokemon_count = json_data['count']
|
return True
|
||||||
base_pokemon = False
|
|
||||||
while not base_pokemon:
|
|
||||||
starter_id = random.randint(1,pokemon_count)
|
|
||||||
base_pokemon = await starter_picker(starter_id)
|
|
||||||
random.seed()
|
|
||||||
return starter_id
|
|
||||||
|
|
||||||
async def get_pkmn_from_id(id):
|
async def load_pokemon(self, discord_id):
|
||||||
url = 'https://pokeapi.co/api/v2/pokemon/' + str(id)
|
if os.path.isfile(self.data_dir+str(discord_id)+".json"):
|
||||||
json_data = await get_json(url)
|
with open(self.data_dir+str(discord_id)+".json", 'r') as f:
|
||||||
|
json_data = json.loads(f.readline())
|
||||||
return json_data
|
return json_data
|
||||||
|
else:
|
||||||
async def give_buddy_food(pkmn_data):
|
return False
|
||||||
try:
|
|
||||||
last_food = pkmn_data['last_food']
|
|
||||||
except:
|
|
||||||
last_food = 0
|
|
||||||
this_food = time.time()
|
|
||||||
if (this_food - last_food) >= 1800:
|
|
||||||
pkmn_data['last_food'] = this_food
|
|
||||||
level = await calc_pkmn_buddy_level(pkmn_data)
|
|
||||||
pkmn_data['buddy_xp'] += (4*level)
|
|
||||||
return pkmn_data, True
|
|
||||||
else:
|
|
||||||
return pkmn_data, False
|
|
||||||
|
|
||||||
async def give_buddy_affection(pkmn_data):
|
async def generate_starter(self, discord_id):
|
||||||
try:
|
random.seed(discord_id)
|
||||||
last_hug = pkmn_data['last_hug']
|
json_data = await get_json('https://pokeapi.co/api/v2/pokemon-species/')
|
||||||
except:
|
pokemon_count = json_data['count']
|
||||||
last_hug = 0
|
base_pokemon = False
|
||||||
this_hug = time.time()
|
while not base_pokemon:
|
||||||
if (this_hug - last_hug) >= 600:
|
starter_id = random.randint(1,pokemon_count)
|
||||||
pkmn_data['last_hug'] = this_hug
|
base_pokemon = await self.starter_picker(starter_id)
|
||||||
level = await calc_pkmn_buddy_level(pkmn_data)
|
random.seed()
|
||||||
pkmn_data['buddy_xp'] += (3*level)
|
return starter_id
|
||||||
return pkmn_data, True
|
|
||||||
else:
|
async def get_pkmn_from_id(self, id):
|
||||||
return pkmn_data, False
|
url = 'https://pokeapi.co/api/v2/pokemon/' + str(id)
|
||||||
|
json_data = await get_json(url)
|
||||||
async def calc_pkmn_buddy_level(pkmn_json): #this uses the 'fast' xp rate
|
return json_data
|
||||||
buddy_xp = pkmn_json['buddy_xp']
|
|
||||||
return min(math.floor(((5*buddy_xp)/4)**(1/3)),100)
|
async def give_buddy_food(self, pkmn_data):
|
||||||
|
|
||||||
async def make_pmkn_embed(pkmn_dict):
|
|
||||||
if pkmn_dict['nickname']:
|
|
||||||
title = pkmn_dict['nickname'] + ' (' + pkmn_dict['name'].capitalize() + ')'
|
|
||||||
else:
|
|
||||||
title = pkmn_dict['name'].capitalize()
|
|
||||||
embed=discord.Embed(title=title)
|
|
||||||
if pkmn_dict['shiny']:
|
|
||||||
embed.set_image(url=pkmn_dict['sprites']['front_shiny'])
|
|
||||||
else:
|
|
||||||
embed.set_image(url=pkmn_dict['sprites']['front_default'])
|
|
||||||
nature = pkmn_dict['nature']
|
|
||||||
buddy_level = await calc_pkmn_buddy_level(pkmn_dict)
|
|
||||||
buddy_xp = pkmn_dict['buddy_xp']
|
|
||||||
types = []
|
|
||||||
for key in pkmn_dict['types']:
|
|
||||||
types.append(key['type']['name'].capitalize())
|
|
||||||
type_str = ', '.join(types)
|
|
||||||
embed.add_field(name="Nature", value=nature.capitalize(), inline=False)
|
|
||||||
embed.add_field(name="Buddy Level", value=buddy_level , inline=True)
|
|
||||||
embed.add_field(name="Buddy XP", value=buddy_xp, inline=True)
|
|
||||||
embed.add_field(name="Types", value=type_str, inline=False)
|
|
||||||
return embed
|
|
||||||
try:
|
try:
|
||||||
if args[0]=='start':
|
last_food = pkmn_data['last_food']
|
||||||
if not os.path.isfile(self.data_dir+str(ctx.author.id)+'.json'):
|
|
||||||
uniq_id = time.time()
|
|
||||||
starter_id = await generate_starter(ctx.author.id)
|
|
||||||
json_data = await get_pkmn_from_id(starter_id)
|
|
||||||
is_shiny = await shiny_roll()
|
|
||||||
nature = random.randint(0,19)
|
|
||||||
nature_data = await get_json('https://pokeapi.co/api/v2/nature/')
|
|
||||||
nature = nature_data['results'][nature]['name']
|
|
||||||
json_data['shiny'] = is_shiny
|
|
||||||
json_data['nickname'] = None
|
|
||||||
json_data['unique_id'] = uniq_id
|
|
||||||
json_data['nature'] = nature
|
|
||||||
json_data['buddy_level'] = 1
|
|
||||||
json_data['buddy_xp'] = 0
|
|
||||||
json_data['last_food'] = 0
|
|
||||||
json_data['last_hug'] = 0
|
|
||||||
await save_pokemon(ctx.author.id, json_data)
|
|
||||||
embed = await make_pmkn_embed(json_data)
|
|
||||||
await ctx.channel.send(embed=embed)
|
|
||||||
return
|
|
||||||
else:
|
|
||||||
await ctx.channel.send("You already have a pokemon!")
|
|
||||||
return
|
|
||||||
|
|
||||||
elif args[0] == 'nick' or args[0] == 'nickname':
|
|
||||||
nickname = args[1]
|
|
||||||
json_data = await load_pokemon(ctx.author.id)
|
|
||||||
json_data['nickname'] = nickname
|
|
||||||
await save_pokemon(ctx.author.id, json_data)
|
|
||||||
message = "You gave " + nickname + ' a new name!'
|
|
||||||
await ctx.channel.send(message)
|
|
||||||
return
|
|
||||||
|
|
||||||
elif args[0] == 'feed':
|
|
||||||
json_data = await load_pokemon(ctx.author.id)
|
|
||||||
json_data, fed = await give_buddy_food(json_data)
|
|
||||||
if fed:
|
|
||||||
await save_pokemon(ctx.author.id, json_data)
|
|
||||||
if json_data['nickname']:
|
|
||||||
message = "You " + args[0] + ' ' + json_data['nickname']
|
|
||||||
else:
|
|
||||||
message = "You " + args[0] + ' ' + json_data['name']
|
|
||||||
await ctx.channel.send(message)
|
|
||||||
return
|
|
||||||
else:
|
|
||||||
if json_data['nickname']:
|
|
||||||
message = "Your " + json_data['nickname'] + " isn't hungry!"
|
|
||||||
else:
|
|
||||||
message = "Your " + json_data['name'] + " isn't hungry!"
|
|
||||||
await ctx.channel.send(message)
|
|
||||||
return
|
|
||||||
|
|
||||||
elif args[0] == 'hug':
|
|
||||||
json_data = await load_pokemon(ctx.author.id)
|
|
||||||
json_data, hugged = await give_buddy_affection(json_data)
|
|
||||||
if hugged:
|
|
||||||
await save_pokemon(ctx.author.id, json_data)
|
|
||||||
if json_data['nickname']:
|
|
||||||
message = "You " + args[0] + ' ' + json_data['nickname']
|
|
||||||
else:
|
|
||||||
message = "You " + args[0] + ' ' + json_data['name']
|
|
||||||
await ctx.channel.send(message)
|
|
||||||
return
|
|
||||||
else:
|
|
||||||
if json_data['nickname']:
|
|
||||||
message = "You hugged " + json_data['nickname'] + " but " + json_data['nickname'] + " has been hugged recently."
|
|
||||||
else:
|
|
||||||
message = "You hugged " + json_data['name'] + " but " + json_data['name'] + " has been hugged recently."
|
|
||||||
await ctx.channel.send(message)
|
|
||||||
return
|
|
||||||
except:
|
except:
|
||||||
#Default !pokemon behavior (Load and show pokemon embed)
|
last_food = 0
|
||||||
discord_id = ctx.author.id
|
this_food = time.time()
|
||||||
buddy_json = await load_pokemon(discord_id)
|
if (this_food - last_food) >= 1800:
|
||||||
if not buddy_json:
|
pkmn_data['last_food'] = this_food
|
||||||
await ctx.channel.send("You don't have a buddy yet. Type ```!pokemon start``` to start your Pokemon journey!")
|
level = await self.calc_pkmn_buddy_level(pkmn_data)
|
||||||
|
pkmn_data['buddy_xp'] += (4*level)
|
||||||
|
return pkmn_data, True
|
||||||
|
else:
|
||||||
|
return pkmn_data, False
|
||||||
|
|
||||||
|
async def give_buddy_affection(self, pkmn_data):
|
||||||
|
try:
|
||||||
|
last_hug = pkmn_data['last_hug']
|
||||||
|
except:
|
||||||
|
last_hug = 0
|
||||||
|
this_hug = time.time()
|
||||||
|
if (this_hug - last_hug) >= 600:
|
||||||
|
pkmn_data['last_hug'] = this_hug
|
||||||
|
level = await self.calc_pkmn_buddy_level(pkmn_data)
|
||||||
|
pkmn_data['buddy_xp'] += (3*level)
|
||||||
|
return pkmn_data, True
|
||||||
|
else:
|
||||||
|
return pkmn_data, False
|
||||||
|
|
||||||
|
async def calc_pkmn_buddy_level(self, pkmn_json): #this uses the 'fast' xp rate
|
||||||
|
buddy_xp = pkmn_json['buddy_xp']
|
||||||
|
return min(math.floor(((5*buddy_xp)/4)**(1/3)),100)
|
||||||
|
|
||||||
|
async def make_pmkn_embed(self, pkmn_dict):
|
||||||
|
if pkmn_dict['nickname']:
|
||||||
|
title = pkmn_dict['nickname'] + ' (' + pkmn_dict['name'].capitalize() + ')'
|
||||||
|
else:
|
||||||
|
title = pkmn_dict['name'].capitalize()
|
||||||
|
embed=discord.Embed(title=title)
|
||||||
|
if pkmn_dict['shiny']:
|
||||||
|
embed.set_image(url=pkmn_dict['sprites']['front_shiny'])
|
||||||
|
else:
|
||||||
|
embed.set_image(url=pkmn_dict['sprites']['front_default'])
|
||||||
|
nature = pkmn_dict['nature']
|
||||||
|
buddy_level = await self.calc_pkmn_buddy_level(pkmn_dict)
|
||||||
|
buddy_xp = pkmn_dict['buddy_xp']
|
||||||
|
types = []
|
||||||
|
for key in pkmn_dict['types']:
|
||||||
|
types.append(key['type']['name'].capitalize())
|
||||||
|
type_str = ', '.join(types)
|
||||||
|
embed.add_field(name="Nature", value=nature.capitalize(), inline=False)
|
||||||
|
embed.add_field(name="Buddy Level", value=buddy_level , inline=True)
|
||||||
|
embed.add_field(name="Buddy XP", value=buddy_xp, inline=True)
|
||||||
|
embed.add_field(name="Types", value=type_str, inline=False)
|
||||||
|
return embed
|
||||||
|
|
||||||
|
@commands.command()
|
||||||
|
async def pkmn_start(self, ctx):
|
||||||
|
if not os.path.isfile(self.data_dir+str(ctx.author.id)+'.json'):
|
||||||
|
uniq_id = time.time()
|
||||||
|
starter_id = await self.generate_starter(ctx.author.id)
|
||||||
|
json_data = await self.get_pkmn_from_id(starter_id)
|
||||||
|
is_shiny = await self.shiny_roll()
|
||||||
|
nature = random.randint(0,19)
|
||||||
|
nature_data = await get_json('https://pokeapi.co/api/v2/nature/')
|
||||||
|
nature = nature_data['results'][nature]['name']
|
||||||
|
json_data['shiny'] = is_shiny
|
||||||
|
json_data['nickname'] = None
|
||||||
|
json_data['unique_id'] = uniq_id
|
||||||
|
json_data['nature'] = nature
|
||||||
|
json_data['buddy_level'] = 1
|
||||||
|
json_data['buddy_xp'] = 0
|
||||||
|
json_data['last_food'] = 0
|
||||||
|
json_data['last_hug'] = 0
|
||||||
|
await self.save_pokemon(ctx.author.id, json_data)
|
||||||
|
embed = await self.make_pmkn_embed(json_data)
|
||||||
|
await ctx.channel.send(embed=embed)
|
||||||
|
return
|
||||||
|
else:
|
||||||
|
await ctx.channel.send("You already have a pokemon!")
|
||||||
|
return
|
||||||
|
|
||||||
|
|
||||||
|
@commands.command()
|
||||||
|
async def pkmn_nick(self, ctx, nickname):
|
||||||
|
json_data = await self.load_pokemon(ctx.author.id)
|
||||||
|
json_data['nickname'] = nickname
|
||||||
|
await self.save_pokemon(ctx.author.id, json_data)
|
||||||
|
message = "You gave " + nickname + " a new name!"
|
||||||
|
await ctx.channel.send(message)
|
||||||
|
|
||||||
|
|
||||||
|
@commands.command()
|
||||||
|
@commands.cooldown(1, 3600, commands.BucketType.user)
|
||||||
|
async def pkmn_feed(self, ctx):
|
||||||
|
discord_id = ctx.author.id
|
||||||
|
json_data = await self.load_pokemon(discord_id)
|
||||||
|
json_data, fed = await self.give_buddy_food(json_data)
|
||||||
|
if fed:
|
||||||
|
await self.save_pokemon(ctx.author.id, json_data)
|
||||||
|
if json_data['nickname']:
|
||||||
|
message = "You Feed " + json_data['nickname']
|
||||||
else:
|
else:
|
||||||
embed = await make_pmkn_embed(buddy_json)
|
message = "You Feed " + json_data['name']
|
||||||
message = await ctx.channel.send(embed=embed)
|
await ctx.channel.send(message)
|
||||||
return
|
return
|
||||||
|
|
||||||
|
|
||||||
|
@commands.command()
|
||||||
|
@commands.cooldown(1, 1800, commands.BucketType.user)
|
||||||
|
async def pkmn_hug(self, ctx):
|
||||||
|
discord_id = ctx.author.id
|
||||||
|
json_data = await self.load_pokemon(discord_id)
|
||||||
|
json_data, hugged = await self.give_buddy_affection(json_data)
|
||||||
|
if hugged:
|
||||||
|
await self.save_pokemon(discord_id, json_data)
|
||||||
|
if json_data['nickname']:
|
||||||
|
message = "You " + "hugged" + ' ' + json_data['nickname']
|
||||||
|
else:
|
||||||
|
message = "You " + "hugged" + ' ' + json_data['name']
|
||||||
|
await ctx.channel.send(message)
|
||||||
|
|
||||||
|
|
||||||
|
@commands.command()
|
||||||
|
async def pokemon(self, ctx):
|
||||||
|
discord_id = ctx.author.id
|
||||||
|
buddy_json = await self.load_pokemon(discord_id)
|
||||||
|
if not buddy_json:
|
||||||
|
await ctx.channel.send("You don't have a buddy yet. Type ```!pokemon start``` to start your Pokemon journey!")
|
||||||
|
else:
|
||||||
|
embed = await self.make_pmkn_embed(buddy_json)
|
||||||
|
message = await ctx.channel.send(embed=embed)
|
||||||
|
return
|
||||||
|
|
||||||
|
|
||||||
async def pkmn_msg(self, discord_id):
|
async def pkmn_msg(self, discord_id):
|
||||||
path = self.data_dir+str(discord_id)+'.json'
|
path = self.data_dir+str(discord_id)+'.json'
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue