phixxy/sparkytron3000
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

changed pokemon into a cog

Browse source
This commit is contained in:
phixxy 2024-01-20 17:39:19 -08:00
parent 275c5b7ee2
commit 2b0ec090c4
2 changed files with 253 additions and 249 deletions
Download patch file Download diff file Expand all files Collapse all files

46
extensions/pokemon.py
View file

@ -1,5 +1,4 @@
#plugin file for sparkytron 3000 #plugin file for sparkytron 3000
from discord.ext import commands from discord.ext import commands
import discord import discord
import random import random
@ -9,9 +8,15 @@ import math
import time import time
import aiohttp import aiohttp
class Pokemon(commands.Cog):
def __init__(self, bot):
self.bot = bot
self.working_dir = "tmp/pokemon/"
self.data_dir = "data/pokemon/"
self.folder_setup()
async def get_json(url): async def get_json(self, url):
http_session = aiohttp.ClientSession() http_session = aiohttp.ClientSession()
async with http_session.get(url) as resp: async with http_session.get(url) as resp:
json_data = await resp.json() json_data = await resp.json()
@ -24,10 +29,10 @@ async def get_json(url):
aliases=['pkmn'], aliases=['pkmn'],
hidden=True hidden=True
) )
async def pokemon(ctx, arg1=None, arg2=None, arg3=None, arg4=None): async def pokemon(self, ctx, *args):
async def starter_picker(id): #id = pokedex number async def starter_picker(id): #id = pokedex number
url = "https://pokeapi.co/api/v2/pokemon-species/" + str(id) url = "https://pokeapi.co/api/v2/pokemon-species/" + str(id)
json_data = await get_json(url) json_data = await self.get_json(url)
if (json_data["evolves_from_species"] == None) and (not json_data['is_mythical']) and (not json_data['is_legendary']): if (json_data["evolves_from_species"] == None) and (not json_data['is_mythical']) and (not json_data['is_legendary']):
return True return True
else: else:
@ -59,7 +64,7 @@ async def pokemon(ctx, arg1=None, arg2=None, arg3=None, arg4=None):
async def generate_starter(discord_id): async def generate_starter(discord_id):
random.seed(discord_id) random.seed(discord_id)
json_data = await get_json('https://pokeapi.co/api/v2/pokemon-species/') json_data = await self.get_json('https://pokeapi.co/api/v2/pokemon-species/')
pokemon_count = json_data['count'] pokemon_count = json_data['count']
base_pokemon = False base_pokemon = False
while not base_pokemon: while not base_pokemon:
@ -70,7 +75,7 @@ async def pokemon(ctx, arg1=None, arg2=None, arg3=None, arg4=None):
async def get_pkmn_from_id(id): async def get_pkmn_from_id(id):
url = 'https://pokeapi.co/api/v2/pokemon/' + str(id) url = 'https://pokeapi.co/api/v2/pokemon/' + str(id)
json_data = await get_json(url) json_data = await self.get_json(url)
return json_data return json_data
async def give_buddy_food(pkmn_data): async def give_buddy_food(pkmn_data):
@ -128,7 +133,7 @@ async def pokemon(ctx, arg1=None, arg2=None, arg3=None, arg4=None):
embed.add_field(name="Types", value=type_str, inline=False) embed.add_field(name="Types", value=type_str, inline=False)
return embed return embed
if arg1=='start': if args[0]=='start':
if not os.path.isdir("databases/pokemon/"): if not os.path.isdir("databases/pokemon/"):
os.makedirs("databases/pokemon/") os.makedirs("databases/pokemon/")
if not os.path.isfile("databases/pokemon/"+str(ctx.author.id)+'.json'): if not os.path.isfile("databases/pokemon/"+str(ctx.author.id)+'.json'):
@ -137,7 +142,7 @@ async def pokemon(ctx, arg1=None, arg2=None, arg3=None, arg4=None):
json_data = await get_pkmn_from_id(starter_id) json_data = await get_pkmn_from_id(starter_id)
is_shiny = await shiny_roll() is_shiny = await shiny_roll()
nature = random.randint(0,19) nature = random.randint(0,19)
nature_data = await get_json('https://pokeapi.co/api/v2/nature/') nature_data = await self.get_json('https://pokeapi.co/api/v2/nature/')
nature = nature_data['results'][nature]['name'] nature = nature_data['results'][nature]['name']
json_data['shiny'] = is_shiny json_data['shiny'] = is_shiny
json_data['nickname'] = None json_data['nickname'] = None
@ -155,8 +160,8 @@ async def pokemon(ctx, arg1=None, arg2=None, arg3=None, arg4=None):
await ctx.channel.send("You already have a pokemon!") await ctx.channel.send("You already have a pokemon!")
return return
elif arg1 == 'nick' or arg1 == 'nickname': elif args[0] == 'nick' or args[0] == 'nickname':
nickname = arg2 nickname = args[1]
json_data = await load_pokemon(ctx.author.id) json_data = await load_pokemon(ctx.author.id)
json_data['nickname'] = nickname json_data['nickname'] = nickname
await save_pokemon(ctx.author.id, json_data) await save_pokemon(ctx.author.id, json_data)
@ -164,15 +169,15 @@ async def pokemon(ctx, arg1=None, arg2=None, arg3=None, arg4=None):
await ctx.channel.send(message) await ctx.channel.send(message)
return return
elif arg1 == 'feed': elif args[0] == 'feed':
json_data = await load_pokemon(ctx.author.id) json_data = await load_pokemon(ctx.author.id)
json_data, fed = await give_buddy_food(json_data) json_data, fed = await give_buddy_food(json_data)
if fed: if fed:
await save_pokemon(ctx.author.id, json_data) await save_pokemon(ctx.author.id, json_data)
if json_data['nickname']: if json_data['nickname']:
message = "You " + arg1 + ' ' + json_data['nickname'] message = "You " + args[0] + ' ' + json_data['nickname']
else: else:
message = "You " + arg1 + ' ' + json_data['name'] message = "You " + args[0] + ' ' + json_data['name']
await ctx.channel.send(message) await ctx.channel.send(message)
return return
else: else:
@ -183,15 +188,15 @@ async def pokemon(ctx, arg1=None, arg2=None, arg3=None, arg4=None):
await ctx.channel.send(message) await ctx.channel.send(message)
return return
elif arg1 == 'hug': elif args[0] == 'hug':
json_data = await load_pokemon(ctx.author.id) json_data = await load_pokemon(ctx.author.id)
json_data, hugged = await give_buddy_affection(json_data) json_data, hugged = await give_buddy_affection(json_data)
if hugged: if hugged:
await save_pokemon(ctx.author.id, json_data) await save_pokemon(ctx.author.id, json_data)
if json_data['nickname']: if json_data['nickname']:
message = "You " + arg1 + ' ' + json_data['nickname'] message = "You " + args[0] + ' ' + json_data['nickname']
else: else:
message = "You " + arg1 + ' ' + json_data['name'] message = "You " + args[0] + ' ' + json_data['name']
await ctx.channel.send(message) await ctx.channel.send(message)
return return
else: else:
@ -219,7 +224,7 @@ async def pokemon(ctx, arg1=None, arg2=None, arg3=None, arg4=None):
aliases=['pdex'], aliases=['pdex'],
hidden=False hidden=False
) )
async def pokedex(ctx): async def pokedex(self, ctx):
pokemon = ctx.message.content.split(" ", maxsplit=1)[1] pokemon = ctx.message.content.split(" ", maxsplit=1)[1]
try: try:
shiny = False shiny = False
@ -229,7 +234,7 @@ async def pokedex(ctx):
url = "https://pokeapi.co/api/v2/pokemon/" + pokemon url = "https://pokeapi.co/api/v2/pokemon/" + pokemon
dex_url = "https://pokeapi.co/api/v2/pokemon-species/" + pokemon dex_url = "https://pokeapi.co/api/v2/pokemon-species/" + pokemon
#try: #try:
data = await get_json(url) data = await self.get_json(url)
name = data['name'] name = data['name']
height_str = str(int(data['height'])/10) + 'm' height_str = str(int(data['height'])/10) + 'm'
weight_str = str(int(data['weight'])/10) + 'kg' weight_str = str(int(data['weight'])/10) + 'kg'
@ -243,7 +248,7 @@ async def pokedex(ctx):
sprite = data["sprites"]["front_default"] sprite = data["sprites"]["front_default"]
if shiny: if shiny:
sprite = data["sprites"]["front_shiny"] sprite = data["sprites"]["front_shiny"]
dex_data = await get_json(dex_url) dex_data = await self.get_json(dex_url)
generation = dex_data['generation']['name'].upper().replace("GENERATION","Generation") generation = dex_data['generation']['name'].upper().replace("GENERATION","Generation")
for entry in dex_data['flavor_text_entries']: for entry in dex_data['flavor_text_entries']:
if entry['language']['name'] == 'en': if entry['language']['name'] == 'en':
@ -270,5 +275,4 @@ async def pokedex(ctx):
await ctx.channel.send(message) await ctx.channel.send(message)
async def setup(bot): async def setup(bot):
bot.add_command(pokedex) bot.add_cog(Pokemon)
bot.add_command(pokemon)

2
sparkytron3000.py
View file

@ -8,7 +8,6 @@ import time
import os import os
import asyncio import asyncio
from dotenv import load_dotenv from dotenv import load_dotenv
import aiohttp import aiohttp
#Stable Diffusion #Stable Diffusion
@ -44,6 +43,7 @@ async def handle_error(error):
f.write(log_line) f.write(log_line)
return error return error
def create_channel_config(filepath): def create_channel_config(filepath):
config_dict = { config_dict = {
"personality":"average", "personality":"average",