def main():
op = -1
while op != 0:
print("Escolha uma opçao: \n")
print("1 - Entrar com o valores atraves de um arquivo '.txt' : \n")
print("2 - Entrar com os valores a partir do terminal: \n")
print("3 - para sair digite 0: \n")
op = int(input(""))
if op == 1:
print(
"-------------------------------------------------------------------"
)
#ler os numeros atraves do arquivo .txt
#caso queira mudar, basta alterar o arquivo
arq = open('./vetor.txt')
numeros = arq.read()
arq.close()
vetor = list(map(int, numeros.split(",")))
print("Vetor de Entrada -> " + str(vetor))
grafo = Service()
capital = grafo.verificaCapital(vetor)
if capital != False:
vetorDistancias = grafo.transformaVetor(vetor, capital)
print("\nvetor de Distancias -> " + str(vetorDistancias) +
"\n")
grafo.traduzVetor(vetorDistancias)
print("\n")
print(
"-------------------------------------------------------------------"
)
elif op == 2:
print(
"-------------------------------------------------------------------"
)
numeros = input(
"Digite uma sequencia de numeros no seguinte padrão: 'x,y,z,.., w,' :\n "
)
vetor = list(map(int, numeros.split(",")))
grafo = Service()
print("\nVetor de Entrada-> " + str(vetor) + "\n")
capital = grafo.verificaCapital(vetor)
if capital != False:
vetorDistancias = grafo.transformaVetor(vetor, capital)
print("vetor de Distancias -> " + str(vetorDistancias))
grafo.traduzVetor(vetorDistancias)
print("\n")
print(
"-------------------------------------------------------------------"
)
elif op == 0:
print("Exit\n")
op = 0
else:
print("Digite uma opcao valida\n")
