封装log方法

 1 import logging
 2 import os.path
 3 import time
 4 
 5 
 6 project_path = 'selenium_framework_demo_lianxi'  #定义项目目录
 7 
 8 class Logger(object):
 9     def __init__(self, logger):
10         """
11         指定保存日志的文件路径，日志级别，以及调用文件
12         将日志存入到指定的文件中
13         :param logger:
14         """
15 
16         # 创建一个logger
17         self.logger = logging.getLogger(logger)
18         self.logger.setLevel(logging.DEBUG)
19 
20         # 返回当前时间
21         current_time = time.strftime('%Y%m%d%H%M', time.localtime(time.time()))
22         # 返回当前目录
23         current_path = os.path.dirname(os.path.abspath(project_path))
24         # 指定分隔符对字符串进行切片
25         path1 = current_path.split(project_path)
26         path2 = [path1[0], project_path]
27         path3 = ''
28         # 在该路径下新建下级目录
29         new_name = path3.join(path2) + '/logs/'
30         # 返回当前时间的年月日作为目录名称
31         dir_time = time.strftime('%Y%m%d', time.localtime(time.time()))
32         # 判断该目录是否存在
33         isExists = os.path.exists(new_name + dir_time)
34         if not isExists:
35             os.makedirs(new_name + dir_time)
36             print(new_name + dir_time + "目录创建成功")
37 
38         else:
39             # 如果目录存在则不创建，并提示目录已存在
40             print(new_name + "目录 %s 已存在" % dir_time)
41 
42         try:
43             # 创建一个handler，用于写入日志文件
44 
45             # 如果case组织结构式 /testsuit/featuremodel/xxx.py ， 那么得到的相对路径的父路径就是项目根目录
46             log_name = new_name + dir_time + '/' + current_time + '.log'  # 定义日志文件的路径以及名称
47 
48             fh = logging.FileHandler(log_name)
49             fh.setLevel(logging.INFO)
50 
51             # 再创建一个handle，用于输出到控制台
52             ch = logging.StreamHandler()
53             ch.setLevel(logging.INFO)
54 
55             # 定义handle的输出格式
56             formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
57             fh.setFormatter(formatter)
58             ch.setFormatter(formatter)
59 
60             # 给logger添加handler
61             self.logger.addHandler(fh)
62             self.logger.addHandler(ch)
63         except Exception as e:
64             print("输出日志失败！ %s" % e)
65 
66     # 日志接口，用户只需调用这里的接口即可，这里只定位了INFO, WARNING, ERROR三个级别的日志，可根据需要定义更多接口
67     def info(cls, msg):
68         cls.log.info(msg)
69         return
70 
71     def warning(cls, msg):
72         cls.log.warning(msg)
73         return
74 
75     def error(cls, msg):
76         cls.log.error(msg)
77         return
78 
79     # 定义一个函数，回调logger实例
80     def getlog(self):
81         return self.logger
82 
83 # 用于调试
84 # if __name__ == '__main__':
85 #     logger = Logger()
86 #     logger.info('This is info')
87 #     logger.warning('This is warning')
88 #     logger.error('This is error')