2.10-finished-code.vy

# @version >=0.2.4 <0.3.0

DNA_DIGITS: constant(uint256) = 16
DNA_MODULUS: constant(uint256) = 10 ** DNA_DIGITS
HP_LIMIT: constant(uint256) = 1000
WILD_POKEMON: constant(address) = 0x66f4804E06007630e1aF0a7B0b279e6F27A3FdE5

struct Pokemon:
    name: String[32]
    dna: uint256
    HP: uint256
    matches: uint256
    wins: uint256

struct Trainer:
    name: String[32]

totalPokemonCount: public(uint256)
trainerPokemonCount: HashMap[address, uint256]

pokemonList: HashMap[uint256, Pokemon]
trainerList: HashMap[address, Trainer]
trainerToPokemon: HashMap[address, HashMap[uint256, Pokemon]]

event NewPokemonCreated:
    name: String[32]
    dna: uint256
    HP: uint256

event NewTrainerCreated:
    name: String[32]

interface WildPokemons:
    def battle(pokemon: Pokemon) -> (bool, String[32], uint256, uint256): nonpayable

@external
def battleWildPokemon(pokemonIndex: uint256):
    assert pokemonIndex < self.trainerPokemonCount[msg.sender], "Invalid Index Provided"
    
    battleResult: bool = empty(bool)
    newPokemonName: String[32] = empty(String[32])
    newPokemonDNA: uint256 = empty(uint256)
    newPokemonHP: uint256 = empty(uint256)

    battleResult, newPokemonName, newPokemonDNA, newPokemonHP = WildPokemons(WILD_POKEMON).battle(self.trainerToPokemon[msg.sender][pokemonIndex])

@pure
@internal
def _generateRandomDNA(_name: String[32]) -> uint256:
    random: uint256 = convert(keccak256(_name), uint256)
    return random % DNA_MODULUS

@internal
def _createPokemon(_name: String[32]) -> Pokemon:
    
    randomDNA: uint256 = self._generateRandomDNA(_name)

    randomHP: uint256 = randomDNA % HP_LIMIT

    newPokemon: Pokemon = Pokemon({
        name: _name,
        dna: randomDNA,
        HP: randomHP,
        matches: 0,
        wins: 0
    })

    self.pokemonList[self.totalPokemonCount] = newPokemon
    self.totalPokemonCount += 1

    log NewPokemonCreated(_name, randomDNA, randomHP)

    return newPokemon

@external
def createTrainer(trainerName: String[32], pokemonName: String[32]):
    
    newPokemon: Pokemon = self._createPokemon(pokemonName)

    newTrainer: Trainer = Trainer({
        name: trainerName
    })

    self.trainerList[msg.sender] = newTrainer
    self.trainerToPokemon[msg.sender][self.trainerPokemonCount[msg.sender]] = newPokemon
    self.trainerPokemonCount[msg.sender] += 1

    log NewTrainerCreated(trainerName)