Jak pętli do EOF w Pythonie?

Czy widziałeś jak iterację nad wierszami w pliku tekstowym?

for line in file_obj:
  use(line)

Możesz zrobić to samo z własnym generatorem:

def read_blocks(file_obj, size):
  while True:
    data = file_obj.read(size)
    if not data:
      break
    yield data

for block in read_blocks(file_obj, 4):
  use(block)

Zobacz też:

• Plik.Czytaj

Wolę już wspomniane rozwiązanie oparte na iteratorze, aby przekształcić to w pętlę for. Innym rozwiązaniem pisanym bezpośrednio jest "pętla-i-pół" Knutha

while 1:
    len_name = data.read(4)
    if not len_name:
        break
    names.append(data.read(len_name))

Możesz zobaczyć przez porównanie, jak łatwo jest go wciągnąć do własnego generatora i użyć jako pętli for.

Widzę, zgodnie z przewidywaniami, że typowa i najpopularniejsza odpowiedź wykorzystuje bardzo wyspecjalizowane generatory do "odczytu 4 bajtów na raz". Czasami ogólność nie jest trudniejsza (i znacznie bardziej satysfakcjonująca; -), więc zaproponowałem następujące bardzo ogólne rozwiązanie: {]}

import operator
def funlooper(afun, *a, **k):
  wearedone = k.pop('wearedone', operator.not_)
  while True:
    data = afun(*a, **k)
    if wearedone(data): break
    yield data

Teraz pożądany nagłówek pętli to po prostu: for len_name in funlooper(data.read, 4):.

Edit : made much more general by the wearedone idiom since a comment possigned my nieco mniej general previous version (hardcoding the exit test as if not data:) posiadania "ukrytej zależności", ze wszystkich rzeczy!-)

Zwykły szwajcarski nóż wojskowy pętli, itertools, też jest dobrze, oczywiście, jak zwykle:

import itertools as it

for len_name in it.takewhile(bool, it.imap(data.read, it.repeat(4))): ...

Lub, całkiem równoważnie:

import itertools as it

def loop(pred, fun, *args):
  return it.takewhile(pred, it.starmap(fun, it.repeat(args)))

for len_name in loop(bool, data.read, 4): ...

Znacznik EOF w Pythonie jest pustym ciągiem znaków, więc to, co masz, jest prawie najlepsze, co możesz uzyskać bez pisania funkcji do zawinięcia tego w iteratorze. Można by napisać w nieco bardziej pythoniczny sposób zmieniając while Jak:

while len_name:
    len_name = struct.unpack("<I", len_name)[0]
    names.append(data.read(len_name))
    len_name = data.read(4)

Wybrałbym funkcję Re sugestii Tendayi i iterator dla czytelności:

def read4():
    len_name = data.read(4)
    if len_name:
        len_name = struct.unpack("<I", len_name)[0]
        return data.read(len_name)
    else:
        raise StopIteration

for d in iter(read4, ''):
    names.append(d)