#coding:utf-8 import logging class TestLog(object): ''' 封裝後的logging ''' def __init__(self , logger = None): ''' 指定保存日誌的文件路徑,日誌級別,以及調用文件 將日誌存入到指定的文件中 ''' # 建立一個logger self.logger = logging.getLogger(logger) self.logger.setLevel(logging.DEBUG) # 建立一個handler,用於寫入日誌文件 self.log_time = time.strftime("%Y_%m_%d_") self.log_path = "D:\\python\\workspace\\pythontest\\log\\" self.log_name = self.log_path + self.log_time + 'test.log' fh = logging.FileHandler(self.log_name, 'a') # 追加模式 這個是python2的 # fh = logging.FileHandler(self.log_name, 'a', encoding='utf-8') # 這個是python3的 fh.setLevel(logging.INFO) # 再建立一個handler,用於輸出到控制檯 ch = logging.StreamHandler() ch.setLevel(logging.INFO) # 定義handler的輸出格式 formatter = logging.Formatter('[%(asctime)s] %(filename)s->%(funcName)s line:%(lineno)d [%(levelname)s]%(message)s') fh.setFormatter(formatter) ch.setFormatter(formatter) # 給logger添加handler self.logger.addHandler(fh) self.logger.addHandler(ch) # 添加下面一句,在記錄日誌以後移除句柄 # self.logger.removeHandler(ch) # self.logger.removeHandler(fh) # 關閉打開的文件 fh.close() ch.close() def getlog(self): return self.logger
封裝後的logging代碼中format()中的自定義日誌格式,能夠根據喜愛更換:python
封裝python自帶的logging類,向Logger類中傳用例名稱,用法 log = Logger().getlog() #放在class上面 class ClassName() log.info("log message") 結果: [2018-01-17 22:45:05,447] test_mainrun.py test_run_mail line:31 [INFO]截圖保存成功,全路徑
具體用法:
代碼以下:web
1 #coding:utf-8 5 from selenium import webdriver 6 import unittest 7 from pythontest.commlib.baselib import TestLog 8 #自定義公共模塊 9 10 log = TestLog().getlog() 11 class testcals(unittest.TestCase): 12 u'''【調用】''' 13 def setUp(self): 14 self.driver = webdriver.Firefox() 15 self.base = Screen(self.driver) # 實例化自定義類commlib.baselib 16 17 def login(self): 18 url_login = "http://www.baidu.com" 19 self.driver.get(url_login) 20 21 def test_01_run_mail(self): 22 try: 26 self.login()28 log.info(self.img) 29 except Exception as msg: 30 log.error("異常緣由 [ %s ]" % msg)32 log.error(self.img) 33 raise 34 35 def test_02_case(self): 36 u'''【test_case】''' 37 log.error("首頁error 日誌") 38 log.debug("訂單頁debug 日誌") 39 log.info("活動頁info 日誌") 40 log.critical("支付critical 日誌") 4 43 44 def tearDown(self): 45 self.driver.quit() 46 47 if __name__ == "__main__": 48 unittest.main()