VnTrader增加日志引擎LogEngine

examples/VnTrader/VT_setting.json

@@ -7,5 +7,9 @@
 	"mongoLogging": true,
 
 	"darkStyle": true,
-	"language": "chinese"
+	"language": "chinese",
+
+	"logLevel": "info",
+	"logConsole": true,
+	"logFile": true
 }

vnpy/trader/VT_setting.json

@@ -7,5 +7,9 @@
 	"mongoLogging": true,
 
 	"darkStyle": true,
-	"language": "chinese"
+	"language": "chinese",
+
+	"logLevel": "debug",
+	"logConsole": true,
+	"logFile": true
 }

vnpy/trader/language/chinese/text.py

@@ -106,7 +106,7 @@ TRADE = u'成交'
 ORDER = u'委托'
 POSITION = u'持仓'
 ACCOUNT = u'账户'
-WORKING_ORDER = u'活动委托'
+WORKING_ORDER = u'可撤'
 
 SYSTEM = u'系统'
 CONNECT_DATABASE = u'连接数据库'

vnpy/trader/vtEngine.py

@@ -2,6 +2,7 @@
 
 import os
 import shelve
+import logging
 from collections import OrderedDict
 from datetime import datetime
 
@@ -47,6 +48,10 @@ class MainEngine(object):
         # 风控引擎实例（特殊独立对象）
         self.rmEngine = None
         
+        # 日志引擎实例
+        self.logEngine = None
+        self.initLogEngine()
+
     #----------------------------------------------------------------------
     def addGateway(self, gatewayModule):
         """添加底层接口"""
@@ -279,6 +284,36 @@ class MainEngine(object):
         """查询引擎中所有上层应用的信息"""
         return self.appDetailList
     
+    #----------------------------------------------------------------------
+    def initLogEngine(self):
+        """初始化日志引擎"""
+        # 创建引擎
+        self.logEngine = LogEngine()
+        
+        # 设置日志级别
+        levelDict = {
+            "debug": LogEngine.LEVEL_DEBUG,
+            "info": LogEngine.LEVEL_INFO,
+            "warn": LogEngine.LEVEL_WARN,
+            "error": LogEngine.LEVEL_ERROR,
+            "critical": LogEngine.LEVEL_CRITICAL,
+        }
+        level = levelDict.get(globalSetting["logLevel"], LogEngine.LEVEL_CRITICAL)
+        self.logEngine.setLogLevel(level)
+        
+        # 设置输出
+        if globalSetting['logConsole']:
+            self.logEngine.addConsoleHandler()
+            
+        if globalSetting['logFile']:
+            self.logEngine.addFileHandler()
+            
+        # 注册事件监听
+        self.eventEngine.register(EVENT_LOG, self.logEngine.processLogEvent)
+        
+        # 记录新启动日志引擎
+        self.logEngine.info(u"------日志引擎启动------")
+        
 
 ########################################################################
 class DataEngine(object):
@@ -377,4 +412,101 @@ class DataEngine(object):
         self.eventEngine.register(EVENT_ORDER, self.updateOrder)
 
 
+########################################################################
+class LogEngine(object):
+    """日志引擎"""
+    
+    # 日志级别
+    LEVEL_DEBUG = logging.DEBUG
+    LEVEL_INFO = logging.INFO
+    LEVEL_WARN = logging.WARN
+    LEVEL_ERROR = logging.ERROR
+    LEVEL_CRITICAL = logging.CRITICAL
+
+    #----------------------------------------------------------------------
+    def __init__(self):
+        """Constructor"""
+        self.logger = logging.getLogger()        
+        self.formatter = logging.Formatter('%(asctime)s  %(levelname)s: %(message)s')
+        self.level = self.LEVEL_CRITICAL
+        
+        self.consoleHandler = None
+        self.fileHandler = None
+        
+        # 添加NullHandler防止无handler的错误输出
+        nullHandler = logging.NullHandler()
+        self.logger.addHandler(nullHandler)    
+        
+        # 日志级别函数映射
+        self.levelFunctionDict = {
+            self.LEVEL_DEBUG: self.debug,
+            self.LEVEL_INFO: self.info,
+            self.LEVEL_WARN: self.warn,
+            self.LEVEL_ERROR: self.error,
+            self.LEVEL_CRITICAL: self.critical,
+        }
+        
+    #----------------------------------------------------------------------
+    def setLogLevel(self, level):
+        """设置日志级别"""
+        self.logger.setLevel(level)
+        self.level = level
+    
+    #----------------------------------------------------------------------
+    def addConsoleHandler(self):
+        """添加终端输出"""
+        if not self.consoleHandler:
+            self.consoleHandler = logging.StreamHandler()
+            self.consoleHandler.setLevel(self.level)
+            self.consoleHandler.setFormatter(self.formatter)
+            self.logger.addHandler(self.consoleHandler)
+            
+    #----------------------------------------------------------------------
+    def addFileHandler(self):
+        """添加文件输出"""
+        if not self.fileHandler:
+            filename = 'vt_' + datetime.now().strftime('%Y%m%d') + '.log'
+            filepath = getTempPath(filename)
+            self.fileHandler = logging.FileHandler(filepath)
+            self.fileHandler.setLevel(self.level)
+            self.fileHandler.setFormatter(self.formatter)
+            self.logger.addHandler(self.fileHandler)
+    
+    #----------------------------------------------------------------------
+    def debug(self, msg):
+        """开发时用"""
+        self.logger.debug(msg)
+        
+    #----------------------------------------------------------------------
+    def info(self, msg):
+        """正常输出"""
+        self.logger.info(msg)
+        
+    #----------------------------------------------------------------------
+    def warn(self, msg):
+        """警告信息"""
+        self.logger.warn(msg)
+        
+    #----------------------------------------------------------------------
+    def error(self, msg):
+        """报错输出"""
+        self.logger.error(msg)
+        
+    #----------------------------------------------------------------------
+    def exception(self, msg):
+        """报错输出+记录异常信息"""
+        self.logger.exception(msg)
+
+    #----------------------------------------------------------------------
+    def critical(self, msg):
+        """影响程序运行的严重错误"""
+        self.logger.critical(msg)
+
+    #----------------------------------------------------------------------
+    def processLogEvent(self, event):
+        """处理日志事件"""
+        log = event.dict_['data']
+        function = self.levelFunctionDict[log.logLevel]     # 获取日志级别对应的处理函数
+        function(log.logContent)
+    
     

vnpy/trader/vtObject.py

@@ -1,6 +1,7 @@
 # encoding: UTF-8
 
 import time
+from logging import INFO
 
 from vnpy.trader.vtConstant import (EMPTY_STRING, EMPTY_UNICODE, 
                                     EMPTY_FLOAT, EMPTY_INT)
@@ -236,6 +237,7 @@ class VtLogData(VtBaseData):
         
         self.logTime = time.strftime('%X', time.localtime())    # 日志生成时间
         self.logContent = EMPTY_UNICODE                         # 日志信息
+        self.logLevel = INFO                                    # 日志级别
 
 
 ########################################################################