Function returns value only the first time it is executed

Question

This is my code:

import conector_modbus as conector_modbus
import leitor_csv as leitor_csv

address_file = 'address.csv'
dados = leitor_csv.leitor(address_file,"dicionario")

def coletar_dados():
    for dado in dados:
        if dado['var'] == "int":
            dado['valor'] = conector_modbus.ler_dado(int(dado['reg']),"int")
            yield dado
        if dado['var'] == "float":
            dado['valor'] = conector_modbus.ler_dado(int(dado['reg']),"float")
            yield dado

def preparar_dados(dados_modbus,maquina,num):
    registradores = {}
    for dado in dados_modbus:
        if dado['maquina'] == maquina and dado['num'] == num:
            registradores.update({dado['nome']:dado['valor']})
    return registradores


if conector_modbus.status_modbus("10.123.1.5",502) == True:
    dados_modbus = coletar_dados()
    dados_br1 = preparar_dados(dados_modbus,"br","1")
    print(dados_br1)

    dados_br1 = preparar_dados(dados_modbus,"br","1")
    print(dados_br1)

And this is the result:

root@jonatas-A530:/media/jonatas/Documentos/Projetos/orange_modbus# python3 teste.py
{'producao_ton': 1056.0025634765625, 'fluxo_ton_h': 0, 'corrente_amp': 32767, 'horimetro_h': 1064.68994140625, 'tempo_sob_carga_h': 4.1541666984558105, 'tempo_ocioso_h': 0.466552734375}
{}

The function was called 2 times, but in the second it does not return value. I would like to understand why this occurs.

Answer 1

Its function coletar_dados() is a generator, that is, after all the data has been consumed its purpose has been fulfilled.

To use it again you need another instance of Generator.

In your case just create a Generator new before using it again.

if conector_modbus.status_modbus("10.123.1.5",502) == True:
    dados_modbus = coletar_dados()
    dados_br1 = preparar_dados(dados_modbus,"br","1")
    print(dados_br1)

    dados_modbus = coletar_dados()  # criando outro generator
    dados_br1 = preparar_dados(dados_modbus,"br","1")
    print(dados_br1)

Answer 2

Jonp, tries to make a loop in his function to collect data, because in it you, make a Yield in data, it is with Yield your function becomes interable, having the need to interact in it to obtain the data. Try this:

if conector_modbus.status_modbus("10.123.1.5",502) == True:
    dados_modbus = [ dados for dados in coletar_dados()]
    dados_br1 = preparar_dados(dados_modbus,"br","1")
    print(dados_br1)

    dados_br1 = preparar_dados(dados_modbus,"br","1")
    print(dados_br1)

Answer 3

Performing a survey I found the following link

One of the answers was the following:

• Store Generator results in a data structure in memory or disk that you can iterate again:

  dados_modbus = list(coletar_dados())
  for dado in dados_modbus: print(dado)
  for dado in dados_modbus: print(dado)`
  

I will use this solution, although the other previous suggestions are correct as well.