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

StringIO的行為與file對象非常像，但它不是磁盤上文件，而是一個內存里的“文件”，我們可以像操作磁盤文件那樣來操作StringIO。這篇文章主要介紹了Python StringIO模塊，此模塊主要用於在內存緩沖區中讀寫數據。模塊中只有一個StringIO類，所以它的可用方法都在類中，此類中的大部分函數都與對文件的操作方法類似。

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

s=StringIO.StringIO([buf])
此實例類似於open方法，不同的是它並不會在硬盤中生成文件，而只寄存在緩沖區；可選參數buf是一個str或unicode類型。它將會與其他后續寫入的數據存放在一起（注意，若要在初始化數據之后繼續寫入數據，則在寫入數據之前，應先將讀寫位置移動到結尾，然后再寫入，否則，初始化數據會被覆蓋掉，因為讀寫位置默認是0）。

StringIO類中的方法：
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])
1》有size參數

無論讀寫位置在哪里，都從起始位置開始，裁剪size字節的數據。

2》不帶size參數

將當前讀寫位置之前的數據，裁剪下來。
----------------------
s.tell()
返回當前讀寫位置。
----------------------
s.seek(pos[,mode])
移動當前讀寫位置至pos處，可選參數mode為0時將讀寫位置移動至pos處，
為1時將讀寫位置從當前位置起向前或向后移動|pos|個長度，
為2時將讀寫位置置於末尾處再向前或向后移動|pos|個長度；

mode的默認值為0。
----------------------
s.close()
釋放緩沖區，執行此函數后，數據將被釋放，也不可再進行操作。
----------------------
s.isatty()
此函數總是返回0。
----------------------
s.flush()
刷新內部緩沖區。
----------------------

實例1：

 

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

f = open(r'C:\Users\91135\Desktop\test.txt', 'w')
writedata(f, "xxxxx!!!!")
f.close()
f = open(r'C:\Users\91135\Desktop\test.txt', 'r')
print f.read()
f.close()

實例2：

import StringIO
def writedata(file, msg):
    file.write(msg)

s = StringIO.StringIO('python')
print s.tell()#讀寫位置默認是0，因此，之后寫入的數據（"xxxxx!!!xxxxxx"）會將之前的數據（'python'）覆蓋掉
writedata(s, "xxxxx!!!xxxxxx")
print s.getvalue()

實例3：

 

import StringIO
s = StringIO.StringIO('python')
s.seek(0,2)#將讀寫位置移動到結尾
s.write("aaaa")
lines = ['xxxxx', 'bbbbbbb']
s.writelines(lines)
s.write("ttttttttt")
print s.getvalue()
#如果使用read方法獲取其中的數據，必須通過seek先設置"文件指針"的位置。
s.seek(0,0)#使用s.read()來讀取所有數據前，應將讀寫位置移動到開頭
print s.read()
print s.len

 

實例4：

 

 

import StringIO
s = StringIO.StringIO("python")
#讀寫位置默認是0，下面的語句在寫入數據時，並沒有移動讀寫位置，因此，之前的數據（"python"）會被覆蓋掉。
s.write("hello python!")
s.write('hello world!')
s.seek(0)
print s.read()
s.seek(-4,2)#移動讀寫位置，以便讀取最后4個字節
print s.read()

 

通過例子，我們看到了StringIO的行為，基本與file一致。StringIO提供了一個方法，無論讀寫位置在哪里，都可以方便的獲取其中的數據：StringIO.getvalue()。

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

 

import cStringIO
s = cStringIO.StringIO("python");
print s
print type(s)
print s.getvalue()
s.write("OOOKKK");#AttributeError: 'cStringIO.StringI' object has no attribute 'write'