Python StringIO模組實現在記憶體緩衝區中讀寫資料_python

模組是用類編寫的，只有一個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()

print s.getvalue()
s.write(" ttttttttt ")
s.seek(0)
print s.readlines()
print s.len

StringIO還有一個對應的c語言版的實現，它有更好的效能，但是稍有一點點的區別，cStringIO沒有len和pos屬性。（還有，cStringIO不支援Unicode編碼）