Làm cách nào để kiểm tra EOF bằng Python?

Question

Làm cách nào để kiểm tra EOF bằng Python? Tôi tìm thấy một lỗi trong mã của tôi, nơi khối văn bản cuối cùng sau khi dấu phân cách không được thêm vào danh sách trả về. Hoặc có thể có một cách tốt hơn để thể hiện chức năng này?

Dưới đây là mã của tôi:

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

Bạn có thể thấy việc giải quyết vấn đề này dễ dàng hơn bằng cách sử dụng 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] 

Một lựa chọn khác là sử dụng một regular expression để phù hợp với dải phân cách:

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

Các end-of-file điều kiện tổ chức ngay sau khi tuyên bố for chấm dứt - mà dường như cách đơn giản nhất để minorly sửa mã này (bạn có thể trích xuất text_block.getvalue() ở cuối nếu bạn muốn kiểm tra nó không có sản phẩm nào trước khi thêm nó).

Answer 3

Tại sao bạn cần StringIO ở đây?

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: Cố định chức năng, các đề xuất khác có thể tốt hơn, chỉ muốn viết chức năng tương tự như chức năng gốc.

EDIT: Giả tập tin bắt đầu với "- -", bằng cách thêm chuỗi rỗng vào danh sách bạn có thể "sửa chữa" các IndexError hoặc bạn có thể sử dụng cái này:

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 

Nhưng cả hai phiên bản tìm kiếm một bit xấu xí với tôi, phiên bản reg-ex sạch hơn nhiều.

Answer 4

Đây là vấn đề tiêu chuẩn với phát bộ đệm.

Bạn không phát hiện EOF - điều đó là không cần thiết. Bạn viết bộ đệm cuối cùng.

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

Đây là một cách nhanh chóng để xem nếu bạn có một tập tin rỗng:

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