adicionar gestão de erros à função 'devolveCor()'

This commit is contained in:
2024-09-20 12:50:03 +01:00
parent d80b44e40d
commit 4adfb19893

View File

@ -1,5 +1,6 @@
import sys import sys
import random import random
from typing import type_check_only
import pygame as pg import pygame as pg
from modules.agente import Agente from modules.agente import Agente
from modules.mapa import Mapa from modules.mapa import Mapa
@ -9,6 +10,9 @@ flags = 0
def devolveCor(agente: Agente | None) -> tuple[int, int, int, int]: def devolveCor(agente: Agente | None) -> tuple[int, int, int, int]:
if type(agente) is None:
return (0, 0, 0, 0)
else:
match agente.estrategia: match agente.estrategia:
case "Neutro": case "Neutro":
return (255, 255, 255, 0) return (255, 255, 255, 0)
@ -18,8 +22,6 @@ def devolveCor(agente: Agente | None) -> tuple[int, int, int, int]:
return (255, 0, 0, 0) return (255, 0, 0, 0)
case "Lunatico": case "Lunatico":
return (0, 0, 255, 0) return (0, 0, 255, 0)
case None:
return (0, 0, 0, 0)
case _: case _:
return (0, 0, 0, 0) return (0, 0, 0, 0)