QMT中print内容同时输出到控制台和日志文件

pengxuan

已于 2023-04-17 09:44:24 修改

阅读量784

点赞数 1

CC 4.0 BY-SA版权

分类专栏： python量化文章标签： python numpy 开发语言

于 2023-03-31 11:51:27 首次发布

本文链接：https://blog.csdn.net/pengxuan/article/details/129876754

python量化专栏收录该内容

12 篇文章

订阅专栏

文章介绍了如何在QMT中使用自定义的Logger类，将print输出同时记录到控制台和日志文件，以便于长时间运行程序时保存完整日志信息。通过定义Logger类，重写write方法，实现了日志的双重记录。在回测结束时，会关闭日志文件并恢复控制台输出。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在QMT中使用print可以打印出执行的日志，但是有时候执行程序时间长，打印的内容比较多，控制台中就要覆盖前面的内容，不能够完整的保留下来做分析，下面示例可以在print到控制台的同时记录到日志文件中。

#encoding:gbk
 
import pandas as pd
import numpy as np
import talib
import sys
import time

#Logger类的定义
class Logger(object):
	def __init__(self,filename="Default.log"):
		self.terminal=sys.stdout
		self.log=open(filename,"w",encoding="utf-8") #防止编码错误

	def write(self, message):
		self.terminal.write(message)
		self.log.write(message)

	def flush(self):
		pass

#log初始化函数,在函数中初始化log对象
def loginit():
	t=time.strftime("-%Y%m%d-%H%M%S",time.localtime()) #时间戳
	filename='d:\stock\log\log'+t+'.txt'
	log=Logger(filename)
	sys.stdout=log
 
def init(ContextInfo):
	ContextInfo.stock_code_list=['000001.SZ']
	ContextInfo.accID='18510385'
	ContextInfo.set_universe(ContextInfo.stock_code_list)
	ContextInfo.set_account(ContextInfo.accID)
	loginit() #初始化log对象
	
def handlebar(ContextInfo):
	timetag=ContextInfo.get_bar_timetag(ContextInfo.barpos)
	dt=timetag_to_datetime(timetag,'%Y-%m-%d %H:%M%S')
	print(dt,'-'*30)
	if ContextInfo.is_last_bar() or dt[:10]==ContextInfo.end[:10]:
		sys.stdout.log.close() #关闭文件
		sys.stdout=sys.stdout.terminal #恢复输出的控制台引用
		print('回测完成') #最后一句没有写到日志文件中