Python2删除指定目录文件与日志记录

前言

由于公司近期搭建了SMB服务器,用作来不同部门之间的数据交换用,但其中的临时文件夹需要每日清空,本打算使用执行使用rm -rf来完成,但考虑到需要记录所删除的文件,因此借着这个需求来复习下logging模块的使用。

使用方式

利用crontab创建定时任务,设置每天凌晨3点执行一次,日志存在py文件目录下

1
00 03 * * * /usr/bin/python /root/remove_file.py

日志

源码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
# -*- coding: utf-8 -*-
# -*- python2.7 -*-
import os
import sys
import logging
reload(sys)
sys.setdefaultencoding('utf8')


DIR = r'/smb/anonymous/___数据交换___(每日清理)/'
#DIR = r'E:\test'
class Remove_File(object):
def __init__(self):
'''
开启日志
'''
self.logger = logging.getLogger()
self.logger.setLevel(logging.INFO)
self.file_logs = logging.FileHandler('remove_file_logs.log', encoding='utf-8')
self.format_logs = logging.Formatter('%(asctime)s %(filename)s[line:%(lineno)d] %(message)s')
self.file_logs.setFormatter(self.format_logs)
self.logger.addHandler(self.file_logs)

def dir_list(self):
'''
递归获取文件与文件夹
'''
for root, dirs, files in os.walk(DIR, topdown=False):
for i in root:
for i in files:
file = os.path.join(root, i)
self.logger.info(file)
for i in root:
for i in dirs:
dir = os.path.join(root, i)
self.logger.info(dir)

def del_files(self):
'''
递归删除文件与文件夹
'''
for root, dirs, files in os.walk(DIR, topdown=False):
for name in files:
os.remove(os.path.join(root, name))
for name in dirs:
os.rmdir(os.path.join(root, name))

def main(self):
self.dir_list()
self.del_files()

if __name__ == '__main__':
rm = Remove_File()
rm.main()