diff --git a/modules/interaccoes.py b/modules/interaccoes.py index 5471d6f..f8cb424 100644 --- a/modules/interaccoes.py +++ b/modules/interaccoes.py @@ -1,12 +1,13 @@ +from hashlib import blake2b + from modules.agente import Agente from modules.mapa import Mapa -from hashlib import blake2b def assinatura_interaccao(agente1: Agente, agente2: Agente) -> str: - assin = blake2b(digest_size=8) # usar 64 bits para performance - assin.update(f"{id(agente1)}+{id(agente2)}".encode("utf-8")) - return assin.hexdigest() + assinatura = blake2b(digest_size=8) # usar 64 bits para performance + assinatura.update(f"{id(agente1)}+{id(agente2)}".encode()) + return assinatura.hexdigest() def criarDicionarioInteraccoes(mapa: Mapa) -> dict[str, list[Agente]]: @@ -20,8 +21,8 @@ def criarDicionarioInteraccoes(mapa: Mapa) -> dict[str, list[Agente]]: ass_inversa = assinatura_interaccao(agente_vizinho, agente_actual) # confirmar se interaccao já existe no dicionario if ( - ass_normal in dicionario_interacoes.keys() - or ass_inversa in dicionario_interacoes.keys() + ass_normal in dicionario_interacoes + or ass_inversa in dicionario_interacoes ): continue # se não existe, vamos adicionar interaccao ao dicionario