55 lines
1.4 KiB
Python
55 lines
1.4 KiB
Python
import os
|
||
import logging
|
||
import random
|
||
import numpy as np
|
||
import datetime
|
||
|
||
|
||
def create_logger(log_path):
|
||
"""
|
||
将日志输出到日志文件和控制台
|
||
"""
|
||
logger = logging.getLogger(__name__)
|
||
logger.setLevel(logging.INFO)
|
||
|
||
formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')
|
||
|
||
# 创建一个handler，用于写入日志文件
|
||
file_handler = logging.FileHandler(filename=log_path)
|
||
file_handler.setFormatter(formatter)
|
||
file_handler.setLevel(logging.INFO)
|
||
if file_handler not in logger.handlers:
|
||
logger.addHandler(file_handler)
|
||
|
||
# 创建一个handler，用于将日志输出到控制台
|
||
console = logging.StreamHandler()
|
||
console.setLevel(logging.DEBUG)
|
||
console.setFormatter(formatter)
|
||
if console not in logger.handlers:
|
||
logger.addHandler(console)
|
||
|
||
return logger
|
||
|
||
def get_file_name(fname):
|
||
"""
|
||
获取文件名
|
||
"""
|
||
return os.path.split(fname)[-1].split(".")[0]
|
||
|
||
def get_file_size(fname):
|
||
"""
|
||
获取文件大小MB
|
||
"""
|
||
fsize = os.path.getsize(fname)
|
||
fsize = fsize/float(1024 * 1024)
|
||
return round(fsize, 2)
|
||
|
||
def get_before_date(n):
|
||
"""
|
||
获取前N天的日期
|
||
"""
|
||
today = datetime.datetime.now()
|
||
# 计算偏移量
|
||
offset = datetime.timedelta(days=-n)
|
||
re_date = (today + offset).strftime('%Y-%m-%d')
|
||
return re_date |