Come verificare EOF in Python?

Question

Come controllo EOF in Python? Ho trovato un bug nel mio codice in cui l'ultimo blocco di testo dopo il separatore non viene aggiunto alla lista di ritorno. O forse c'è un modo migliore per esprimere questa funzione?

Ecco il mio codice:

def get_text_blocks(filename): 
    text_blocks = [] 
    text_block = StringIO.StringIO() 
    with open(filename, 'r') as f: 
     for line in f: 
      text_block.write(line) 
      print line 
      if line.startswith('-- -'): 
       text_blocks.append(text_block.getvalue()) 
       text_block.close() 
       text_block = StringIO.StringIO() 
    return text_blocks 

Answer 1

Potresti trovare più facile la soluzione utilizzando itertools.groupby.

def get_text_blocks(filename): 
    import itertools 
    with open(filename,'r') as f: 
     groups = itertools.groupby(f, lambda line:line.startswith('-- -')) 
     return [''.join(lines) for is_separator, lines in groups if not is_separator] 

Un'altra alternativa è usare un regular expression per abbinare i separatori:

def get_text_blocks(filename): 
    import re 
    seperator = re.compile('^-- -.*', re.M) 
    with open(filename,'r') as f: 
     return re.split(seperator, f.read()) 

Answer 2

L'end-of-file condizione è non appena la dichiarazione for termina - che sembra il modo più semplice per risolvere minorly questo codice (si può estrarre text_block.getvalue() alla fine se vuoi controllare che non sia vuoto prima di aggiungerlo).

Answer 3

Perché avete bisogno di StringIO qui?

def get_text_blocks(filename): 
    text_blocks = [""] 
    with open(filename, 'r') as f: 
     for line in f: 
      if line.startswith('-- -'): 
       text_blocks.append(line) 
      else: text_blocks[-1] += line   
    return text_blocks 

EDIT: Fisso la funzione, altri suggerimenti potrebbe essere migliore, volevo solo scrivere una funzione simile a quella originale.

EDIT: Assunta il file inizia con "- -", con l'aggiunta di stringa vuota alla lista è possibile "fissare" l'IndexError o si potrebbe utilizzare questo:

def get_text_blocks(filename): 
    text_blocks = [] 
    with open(filename, 'r') as f: 
     for line in f: 
      if line.startswith('-- -'): 
       text_blocks.append(line) 
      else: 
       if len(text_blocks) != 0: 
        text_blocks[-1] += line   
    return text_blocks 

Ma entrambe le versioni guarda un un po 'brutto per me, la versione reg-ex è molto più pulita.

Answer 4

Questo è il problema standard emettono buffer.

Non si rileva EOF: è inutile. Scrivi l'ultimo buffer.

def get_text_blocks(filename): 
    text_blocks = [] 
    text_block = StringIO.StringIO() 
    with open(filename, 'r') as f: 
     for line in f: 
      text_block.write(line) 
      print line 
      if line.startswith('-- -'): 
       text_blocks.append(text_block.getvalue()) 
       text_block.close() 
       text_block = StringIO.StringIO() 
     ### At this moment, you are at EOF 
     if len(text_block) > 0: 
      text_blocks.append(text_block.getvalue()) 
     ### Now your final block (if any) is appended. 
    return text_blocks 

Answer 5

def get_text_blocks(filename): 
    text_blocks = [] 
    text_block = StringIO.StringIO() 
    with open(filename, 'r') as f: 
     for line in f: 
      text_block.write(line) 
      print line 
      if line.startswith('-- -'): 
       text_blocks.append(text_block.getvalue()) 
       text_block.close() 
       text_block = StringIO.StringIO() 
     ### At this moment, you are at EOF 
     if len(text_block) > 0: 
      text_blocks.append(text_block.getvalue()) 
     ### Now your final block (if any) is appended. 
    return text_blocks 

Answer 6

questo è un modo veloce per vedere se si dispone di un file vuoto:

if f.read(1) == '': 
print "EOF" 
f.close()