如何使用python3.x使用（或更改）print写入文件

Question

我使用一个来自交互式代理的API来获取历史数据，我使用他们的代码来下载数据

该代码使用print函数将数据输出到终端，但我想将其重定向到一个文件（让我们称该文件为StockData.txt）

我使用的代码是：

from ibapi.client import EClient
from ibapi.wrapper import EWrapper
from ibapi.contract import Contract

class TestApp(EWrapper, EClient):
    def __init__(self):
        EClient.__init__(self, self)

    def error(self, reqId, errorCode, errorString):
        print("error: ", reqId, " ", errorCode, " ", errorString)

    def historicalData(self, reqId, bar):
        print("HistoricalData. ", reqId, " Date:", bar.date, "Open:", bar.open, "High:", bar.high, "Low:", bar.low, "Close:", bar.close, "Volume:", bar.volume)

def main():
    app = TestApp()

    app.connect("127.0.0.1", 7497, 0)

    contract = Contract ()
    contract.symbol = "AAPL"
    contract.secType = "STK"
    contract.exchange = "SMART"
    contract.currency = "USD"
    contract.primaryExchange = "NASDAQ"

    app.reqHistoricalData(0, contract, "", "1 D", "1 min", "TRADES", 0, 1, False, [])

    app.run()

if __name__ == "__main__":
    main()

正如我所说，我希望API能够写入StockData.txt，但我不确定如何做到这一点，因为这不是我的代码，而且我对Python也不是那么精通

有人能帮帮我吗？ 谢谢

Answer 1

虽然执行输出重定向可能更好：

python my_script.py > outputfile.txt

实际上覆盖任何函数都非常简单：

def print_to_file(*args):
    with open('text.txt', 'a') as fh:
        fh.write(' '.join(map(str,args)))
print = print_to_file

只需将上述代码行塞进脚本的某个较高位置
不过，这不是推荐的方法，因为它只会覆盖当前脚本中的print，而不会覆盖可能正在打印的导入模块中的print

更好的方法是更改stdout：

import sys
sys.stdout = open('text.txt', 'a')