client.py

import requests
import sys
from pokemon import Pokemon
from lxml import etree
from battle import Battle
from ai import *

ai = None

if len(sys.argv) == 4:
    ai = AI()

if len(sys.argv) >= 3:
    poke = Pokemon(sys.argv[2])

elif len(sys.argv) == 2:
    poke = Pokemon(sys.stdin)

# começa a bataha


r = requests.post('http://localhost:5000/battle/', data = poke.to_XML(
                                    '<battle_state></battle_state>'))

while r.status_code != 200:
    print('status code: ', r.status_code)
    arq = input('Digite um nome de arquivo valido: ')
    poke = Pokemon(arq)
    r = requests.post('http://localhost:5000/battle/', data = poke.to_XML(
                                    '<battle_state></battle_state>'))

battle = Battle()

poke = Pokemon(r.text, xml = True)

battle_state = etree.fromstring(r.text)
poke1 = battle_state[0]
battle_state.remove(poke1)
pkmn = etree.tostring(battle_state, encoding = 'unicode')
 
poke2 = Pokemon(pkmn, xml = True)
 
while battle.all_alive(poke, poke2):

    atk_id = battle.make_choice(poke, poke2, ai) + 1
    r = requests.post('http://localhost:5000/battle/attack/' + str(atk_id))

    poke = Pokemon(r.text, xml = True)

    battle_state = etree.fromstring(r.text)
    poke1 = battle_state[0]
    battle_state.remove(poke1)
    pkmn = etree.tostring(battle_state, encoding = 'unicode')
 
    poke2 = Pokemon(pkmn, xml = True)

r = requests.post('http://localhost:5000/shutdown')