python file.seek() 文件指针

当我们读取文件内容时，并不能重复的读取，比如一个blogCblog.txt文件里有blogCblog内容，用两个read()方法读取blogCblog.txt的内容，会发现，第一个返回文件内容，第二个返回‘’。并不能重复读取，如果我想重复读取呢？怎么解决。

    其实这跟文件指针有关，当我们对文件操作时，文件内部会有一个文件指针来定位当前位置，如图：

 

当open时文件指针是在初始位置1，当读取4个字节内容是文件指针位置就在4，再write时，文件指针从4移到8，所以，只要控制了文件指针，就能重复的读取了！

 操作文件指针可以通过seek方法：

　　seek(offset, whence)：移动文件指针

　　offset：偏移量，可为负数

　　whence：偏移相对位置

 

偏移相对位置为os模块当中的SEEK_SET、SEEK_CUR、SEEK_END：

　　os.SEEK_SET：表示文件的相对起始位置

　　os.SEEK_CUR：表示文件的相对当前位置

　　os.SEEK_END：表示文件的相对结束位置

 

来看下实例代码（一个blogCblog.txt文件里有blogCblog内容）：

f = open('blogCblog.txt')  #首先先创建一个文件对象
print f.read(3)  #用read()方法读取并打印
print f.tell()  #打印出文件指针的位置
f.close()  #关闭文件

#打印结果：blo
#         3

上面可以看到用read(3)读取3个字节的内容，而tell()方法返回文件指针的位置。下面来操作文件指针：

import os  #导入os模块
f = open('blogCblog.txt')  #首先先创建一个文件对象，打开方式为w
print f.read(3)  #用read()方法读取并打印
print f.tell()  #打印出文件指针的位置
f.seek(0, os.SEEK_SET)  #用seek()方法操作文件指针（把文件指针移到文件起始位置并移动0）
print f.read(3)  #用read()方法再次读取并打印
print f.tell()  #打印出文件指针的位置
f.close()  #关闭文件

#打印结果：blo
#         3
#        blo
#         3

上面的代码就重复读取了，如果把第五行注释掉，那么第6、7行代码打印的结果为gCb和6，这是因为文件指针在第3，当再次读取时就在当前的文件指针向后移，这也就是为什么readline()会记住读取位置的原因了。而第5行代码的作用就是把文件指针移到初始位置！就实现了重复读取。

 

当你移到指针时，偏移量大于字符串长度时就会报IOError，需要注意！

 

项目实战

# 生成Excel导出
def basic_export_excel(file_addr=None, data=None, data_key=None, header=None, sheet_name=None):
    fp = io.BytesIO()
    book = Workbook(file_addr if file_addr is not None else fp)
    sheet = book.add_worksheet(sheet_name)
    for idx, head in enumerate(header):
        sheet.write(0, idx, head)
    for row_number, row_data in enumerate(data):
        for col, key in enumerate(data_key):
            insert_val = row_data[key]
            if isinstance(insert_val, datetime):
                insert_val = insert_val.strftime('%Y-%m-%d %H:%M:%S')
            elif isinstance(insert_val, date):
                insert_val = insert_val.strftime('%Y-%m-%d')
            elif isinstance(insert_val, list):
                insert_val = str(insert_val)
            sheet.write(row_number + 1, col, insert_val)
    book.close()
    fp.seek(0)
    return fp
# 开始调用
fp = basic_export_excel(data=hardware_data, data_key=fields, header=field_des)
# 给前端发送文件
return send_file(filename_or_fp=fp, attachment_filename=sheet_title, as_attachment=True)