背景如下,想在析构函数__del__里面做一些退出前的操作:
第一版代码如下:
def __del__(self):
try:
if self.monitor_info['IterSpeed'] > 0 and self.monitor_info[
'FullTime'] > 0:
benchmark_inserter = insertdata.DataInseter()#写入数据库
atexit.register(benchmark_inserter.insert, **self.monitor_info)
except:
with open(os.getcwd() + '/auto_benchmark/fail_models.sh',
"a+") as f:
f.writelines(self.monitor_info['ModelName'] + '\n')
logger.info(f'{model_info} failed to write to db!')
思路如下:
加try 是想防止self.monitor_info等原因不存在报错,就想直接把错误模型信息写入文件。
发现报错主要两方面:
- open不存在
- os.getcwd() 为none
第一次用这个方法,不熟悉,猜测可能是import的库被析构了,谷歌一下,看到这个回答:
https://blog.csdn.net/nankai0912678/article/details/105269848
也因为这个回答,
https://blog.csdn.net/mz02230909mz/article/details/111594937
“导入atexit模块,在类初始化函数中注册一个析构函数,拥有与__del__函数同样的效果”
发现自己使用有误,上传数据库的操作和else里面的导入文件的操作,其实可以一起放到析构里面来做。而不是一个在析构,另一个不在。
后面倒入文件的操作也是在析构时候做就可以,所以改成这样:
def __init__(self):
atexit.register(self.test)
def test(self):
if self.monitor_info['IterSpeed'] > 0 and self.monitor_info[
'FullTime'] > 0:
benchmark_inserter = insertdata.DataInseter()
benchmark_inserter.insert(**self.monitor_info)
else:
with open(os.getcwd() + '/auto_benchmark/fail_models.sh',
"a+") as f:
f.writelines(self.monitor_info['ModelName'] + '\n')
logger.info(f'{model_info} failed to write to db!')
成功~
所以,写在__del__里面的操作,其实都可以用atexit来做,反而麻烦更少~