diff --git a/data/v2/build.py b/data/v2/build.py index a9b7e5e17..76561b22f 100644 --- a/data/v2/build.py +++ b/data/v2/build.py @@ -1388,6 +1388,7 @@ def csv_record_to_objects(info): dream_world = "other/dream-world/" home = "other/home/" official_art = "other/official-artwork/" + showdown = "other/showdown/" gen_i = "versions/generation-i/" gen_ii = "versions/generation-ii/" gen_iii = "versions/generation-iii/" @@ -1438,6 +1439,32 @@ def csv_record_to_objects(info): poke_sprites + official_art + "shiny/", info, "png" ), }, + "showdown": { + "front_default": try_image_names( + poke_sprites + showdown, info, "gif" + ), + "front_shiny": try_image_names( + poke_sprites + showdown + "shiny/", info, "gif" + ), + "front_female": try_image_names( + poke_sprites + showdown + "female/", info, "gif" + ), + "front_shiny_female": try_image_names( + poke_sprites + showdown + "shiny/female/", info, "gif" + ), + "back_default": try_image_names( + poke_sprites + showdown + "back/", info, "gif" + ), + "back_shiny": try_image_names( + poke_sprites + showdown + "back/shiny/", info, "gif" + ), + "back_female": try_image_names( + poke_sprites + showdown + "back/female/", info, "gif" + ), + "back_shiny_female": try_image_names( + poke_sprites + showdown + "back/shiny/female", info, "gif" + ), + }, }, "versions": { "generation-i": { diff --git a/pokemon_v2/tests.py b/pokemon_v2/tests.py index 32d406d2c..57ef4a14b 100644 --- a/pokemon_v2/tests.py +++ b/pokemon_v2/tests.py @@ -1675,6 +1675,32 @@ def setup_pokemon_sprites_data( back_shiny_female=False, ): sprite_path = "https://raw.githubusercontent.com/PokeAPI/sprites/master/sprites/pokemon/%s.png" + showdown_path = "https://raw.githubusercontent.com/PokeAPI/sprites/master/sprites/pokemon/other/showdown/%s.png" + + showdown = { + "front_default": showdown_path % pokemon.id if front_default else None, + "front_female": showdown_path % f"female/{pokemon.id}" + if front_female + else None, + "front_shiny": showdown_path % f"shiny/{pokemon.id}" + if front_shiny + else None, + "front_shiny_female": showdown_path % f"shiny/female/{pokemon.id}" + if front_shiny_female + else None, + "back_default": showdown_path % f"back/{pokemon.id}" + if back_default + else None, + "back_female": showdown_path % f"back/female/{pokemon.id}" + if back_female + else None, + "back_shiny": showdown_path % f"back/shiny/{pokemon.id}" + if back_shiny + else None, + "back_shiny_female": showdown_path % f"back/shiny/female/{pokemon.id}" + if back_shiny_female + else None, + } sprites = { "front_default": sprite_path % pokemon.id if front_default else None, @@ -1692,7 +1718,8 @@ def setup_pokemon_sprites_data( } pokemon_sprites = PokemonSprites.objects.create( - pokemon=pokemon, sprites=json.dumps(sprites) + pokemon=pokemon, + sprites=json.dumps(sprites | {"other": {"showdown": showdown}}), ) pokemon_sprites.save() @@ -5025,6 +5052,7 @@ def test_pokemon_api(self): ) sprites_data = json.loads(pokemon_sprites.sprites) + response_sprites_data = json.loads(response.data["sprites"]) # sprite params self.assertEqual( @@ -5033,6 +5061,15 @@ def test_pokemon_api(self): ) self.assertEqual(sprites_data["back_default"], None) + self.assertEqual( + sprites_data["other"]["showdown"]["front_default"], + response_sprites_data["other"]["showdown"]["front_default"], + ) + self.assertEqual( + sprites_data["other"]["showdown"]["back_default"], + response_sprites_data["other"]["showdown"]["back_default"], + ) + def test_pokemon_form_api(self): pokemon_species = self.setup_pokemon_species_data() pokemon = self.setup_pokemon_data(pokemon_species=pokemon_species)