Python StringIO模塊實現在內存緩沖區中讀寫數據

2019-11-25 17:49:11

字體：大中小

來源：轉載

供稿：網友

模塊是用類編寫的，只有一個StringIO類，所以它的可用方法都在類中。
此類中的大部分函數都與對文件的操作方法類似。

例：

#coding=gbk
 
import StringIO, cStringIO, sys
 
s = StringIO.StringIO("JGood is a handsome boy")
s.write("JGood is a handsome boy /r/n")
s.write('okkkk中國')
s.seek(0)
print s.read()
 
#最后4個字節
s.seek(-4, 2)
print s.read()
 
#---- 結果 ----
#JGood is a handsome boy 
#okkkk中國
#中國

通過例子，我們看到了StringIO的行為，基本與file一致。StringIO提供了一個方法，可以方便的獲取其中的數據：StringIO. getvalue()。如果使用read方法獲取其中的數據，必須通過seek先設置"文件指針"的位置。

Python標準模塊中還提供了一個cStringIO模塊，它的行為與StringIO基本一致，但運行效率方面比StringIO更好。但使用 cStringIO模塊時，有幾個注意點： 1. cStringIO.StringIO不能作為基類被繼承；2. 創建cStringIO.StringIO對象時，如果初始化函數提供了初始化數據，新生成的對象是只讀的。所以下面的代碼是錯誤的：s = cStringIO.StringIO("JGood/n"); s.write("OOOKKK");

----------------------

復制代碼代碼如下:

s=StringIO.StrngIO([buf])

此實例類似于open方法，不同的是它并不會在硬盤中生成文件，而只寄存在緩沖區；可選參數buf是一個str或unicode類型。它將會與其他后續寫入的數據存放在一起。
----------------------
StringIO類中的方法：
● read
● readline
● readlines
● write
● writelines
● getvalue
● truncate
● tell
● seek
● close
● isatty
● flush
----------------------
s.read([n])
參數n限定讀取長度，int類型；缺省狀態為從當前讀寫位置讀取對象s中存儲的所有數據。讀取結束后，讀寫位置被移動。
----------------------
s.readline([length])
參數length限定讀取的結束位置，int類型，缺省狀態為None：從當前讀寫位置讀取至下一個以“/n”為結束符的當前行。讀寫位置被移動。
----------------------
s.readlines([sizehint])
參數sizehint為int類型，缺省狀態為讀取所有行并作為列表返回，除此之外從當前讀寫位置讀取至下一個以“/n”為結束符的當前行。讀寫位置被移動。
----------------------
s.write(s)
從讀寫位置將參數s寫入給對象s。參數s為str或unicode類型。讀寫位置被移動。
----------------------
s.writelines(list)
從讀寫位置將list寫入給對象s。參數list為一個列表，列表的成員為str或unicode類型。讀寫位置被移動。
----------------------
s.getvalue()
此函數沒有參數，返回對象s中的所有數據。
----------------------
s.truncate([size])
從讀寫位置起切斷數據，參數size限定裁剪長度，缺省值為None。
----------------------
s.tell()
返回當前讀寫位置。
----------------------
s.seek(pos[,mode])
移動當前讀寫位置至pos處，可選參數mode為0時將讀寫位置移動至pos處，為1時將讀寫位置從當前位置起向后移動pos個長度，為2時將讀寫位置置于末尾處再向后移動pos個長度；默認為0。
----------------------
s.close()
釋放緩沖區，執行此函數后，數據將被釋放，也不可再進行操作。
---------------------
s.isatty()
此函數總是返回0。不論StringIO對象是否已被close()。
----------------------
s.flush()
刷新內部緩沖區。
----------------------
dir(StringIO.StringIO)的返回值中還包含有一個test函數，不過不用理睬它，它沒有任何意義

=====================================================

StringIO經常被用來作為字符串的緩存，應為StringIO有個好處，他的有些接口和文件操作是一致的，也就是說用同樣的代碼，可以同時當成文件操作或者StringIO操作。比如：

復制代碼代碼如下:

import string, os, sys
import StringIO

def writedata(fd, msg):
fd.write(msg)

f = open('aaa.txt', 'w')

writedata(f, "xxxxxxxxxxxx")
f.close()

s = StringIO.StringIO()
writedata(s, "xxxxxxxxxxxxxx")

因為文件對象和StringIO大部分的方法都是一樣的，比如read, readline, readlines, write, writelines都是有的，這樣，StringIO就可以非常方便的作為"內存文件對象"。

復制代碼代碼如下:

import string
import StringIO

s = StringIO.StringIO()
s.write("aaaa")
lines = ['xxxxx', 'bbbbbbb']
s.writelines(lines)

s.seek(0)
print s.read()