Python：如何動態寫入多個文件

Question

我順序記錄來自多個流的數據包。 根據每個數據包中的數據，我需要將數據寫入特定的日志文件。 我將一次最多打開8個文件以滿足最多8個並發數據流日志文件在新數據流的開始和結束時打開和關閉。

我已經提出以下代碼來處理寫操作。

#Identify where the data needs to be stored
filePointer = unpack_from('!B',payload, 4)

#Grab the data          
capData = unpack_from('!160s', payload, 10)         

#Store the data
if filePointer[0] == 1: Logfile1.write(str(capData[0]))
elif filePointer[0] == 2: Logfile2.write(str(capData[0]))
elif filePointer[0] == 3: Logfile3.write(str(capData[0]))
elif filePointer[0] == 4: Logfile4.write(str(capData[0]))
elif filePointer[0] == 5: Logfile5.write(str(capData[0]))
elif filePointer[0] == 6: Logfile6.write(str(capData[0]))
elif filePointer[0] == 7: Logfile7.write(str(capData[0]))
elif filePointer[0] == 8: Logfile8.write(str(capData[0]))

有沒有更好的pythony方式來做到這一點？

變量可以以某種方式用於組成文件句柄或返回所需的文件句柄嗎？

干杯

POB

Answer 1

您可以構建文件列表字典並使用它來動態選擇文件：

d = {1: Logfile1, 2: Logfile2, ...}
curr_file = d[filePointer[0]]
curr_file.write(str(capData[0]))

Answer 2

你可以使用一個列表，比如說“logfiles”來存儲你的“logfileN”文件對象。

或者dict，代碼示例：

logfiles = {}
for i in range(1, 9):
    logfiles[i] = open("Mylog_%s.log"%(i,), "w")


for filePointer, capData in yourLoop: 
    logfiles[filePointer[0]].write(str(capData[0]))

（根據需要添加錯誤控制）

Answer 3

使用dict，將filePointer [0]映射到logfile

Answer 4

我認為pythonic方式是這樣的：

logFileHandlers = {
    1: Logfile1,
    2: Logfile2,
    3: Logfile3,
    4: Logfile4,
    5: Logfile5,
    6: Logfile6,
    7: Logfile7,
    8: Logfile8,
}
try:
    fileHandler = logFileHandlers[filePointer[0]]
    fileHandler.write(str(capData[0]))
except KeyError as e:
    print e # handle exception: unrecognized data from stream