adicionago type hint para parametro 'agente'
This commit is contained in:
@ -4,10 +4,11 @@ import pygame as pg
|
||||
from modules.agente import Agente
|
||||
from modules.mapa import Mapa
|
||||
import modules.estrategia as ModEstrategia
|
||||
|
||||
flags = 0
|
||||
|
||||
|
||||
def devolveCor(agente: Agente) -> tuple[int, int, int, int]:
|
||||
def devolveCor(agente: Agente | None) -> tuple[int, int, int, int]:
|
||||
match agente.estrategia:
|
||||
case "Neutro":
|
||||
return (255, 255, 255, 0)
|
||||
@ -17,6 +18,8 @@ def devolveCor(agente: Agente) -> tuple[int, int, int, int]:
|
||||
return (255, 0, 0, 0)
|
||||
case "Lunatico":
|
||||
return (0, 0, 255, 0)
|
||||
case None:
|
||||
return (0, 0, 0, 0)
|
||||
case _:
|
||||
return (0, 0, 0, 0)
|
||||
|
||||
@ -24,7 +27,7 @@ def devolveCor(agente: Agente) -> tuple[int, int, int, int]:
|
||||
def criarTabuleiro(
|
||||
surface: pg.Surface,
|
||||
dimensao_tabuleiro: tuple[int, int],
|
||||
mapa: Mapa,
|
||||
mapa: Mapa | None,
|
||||
) -> None:
|
||||
dimensoes_surface_principal: tuple[int, int] = surface.get_size()
|
||||
max_altura: int = int(
|
||||
@ -119,4 +122,4 @@ def main(mapa: Mapa | None):
|
||||
|
||||
if __name__ == '__main__':
|
||||
mapa_mundo: Mapa = Mapa((10, 10))
|
||||
app = App(mapa_mundo)
|
||||
app = main(mapa_mundo)
|
||||
|
||||
Reference in New Issue
Block a user