mirror of
https://github.com/NanjingForestryUniversity/supermachine-wood.git
synced 2025-11-08 18:23:54 +00:00
74 lines
1.8 KiB
Python
74 lines
1.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
Created on Nov 3 21:18:26 2020
|
|
|
|
@author: l.z.y
|
|
@e-mail: li.zhenye@qq.com
|
|
"""
|
|
import os
|
|
import shutil
|
|
import time
|
|
|
|
|
|
def mkdir_if_not_exist(dir_name, is_delete=False):
|
|
"""
|
|
创建文件夹
|
|
:param dir_name: 文件夹
|
|
:param is_delete: 是否删除
|
|
:return: 是否成功
|
|
"""
|
|
try:
|
|
if is_delete:
|
|
if os.path.exists(dir_name):
|
|
shutil.rmtree(dir_name)
|
|
print('[Info] 文件夹 "%s" 存在, 删除文件夹.' % dir_name)
|
|
|
|
if not os.path.exists(dir_name):
|
|
os.makedirs(dir_name)
|
|
print('[Info] 文件夹 "%s" 不存在, 创建文件夹.' % dir_name)
|
|
return True
|
|
except Exception as e:
|
|
print('[Exception] %s' % e)
|
|
return False
|
|
|
|
|
|
def create_file(file_name):
|
|
"""
|
|
创建文件
|
|
:param file_name: 文件名
|
|
:return: None
|
|
"""
|
|
if os.path.exists(file_name):
|
|
print("文件存在：%s" % file_name)
|
|
return False
|
|
# os.remove(file_name) # 删除已有文件
|
|
if not os.path.exists(file_name):
|
|
print("文件不存在，创建文件：%s" % file_name)
|
|
open(file_name, 'a').close()
|
|
return True
|
|
|
|
|
|
class Logger(object):
|
|
def __init__(self, is_to_file=False, path=None):
|
|
self.is_to_file = is_to_file
|
|
if path is None:
|
|
path = "wood.log"
|
|
self.path = path
|
|
create_file(path)
|
|
|
|
def log(self, content):
|
|
if self.is_to_file:
|
|
with open(self.path, "a") as f:
|
|
print(time.strftime("[%Y-%m-%d_%H-%M-%S]:"), file=f)
|
|
print(content, file=f)
|
|
else:
|
|
print(content)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
log = Logger(is_to_file=True)
|
|
log.log("nihao")
|
|
import numpy as np
|
|
a = np.ones((100, 100, 3))
|
|
log.log(a.shape)
|