python 監控windows磁盤空間和備份大小

#!/usr/bin/env python
# Version = 3.5.2
# __auth__ = '無名小妖'
import os
import time
import sendmail
import psutil
import collections

disk_used = collections.OrderedDict()
cur_time = time.time()
# current_day = cur_time - cur_time % 86400
root_dir = ["D:\\backup", "G:\db_backup"]
log_name = 'result.log'


def get_disk_info():
    """
    查看磁盤屬性信息
    :return: 磁盤使用率和剩余空間
    """
    for id in psutil.disk_partitions():
        if 'cdrom' in id.opts or id.fstype == '':
            continue
        disk_name = id.device.split(':')
        s = disk_name[0]
        disk_info = psutil.disk_usage(id.device)
        disk_used[s + '盤使用率：'] = '{}%'.format(disk_info.percent)
        disk_used[s + '剩余空間：'] = '{}GB'.format(disk_info.free // 1024 // 1024 // 1024)
    return disk_used


if __name__ == '__main__':
    with open(log_name, 'w') as f:
        ret = get_disk_info()
        for k, v in ret.items():
            f.write('{}{}。\n'.format(k, v))

        for item in root_dir:
            print(item)
            all_size = 0
            for parent, dir_names, file_names in os.walk(item):  # 遍歷文件夾
                for filename in file_names:
                    path = os.path.join(parent, filename)  # 獲取文件路徑
                    stat_info = os.stat(path)  # 讀取文件屬性
                    upload_time = time.ctime(stat_info.st_mtime)

                    if stat_info.st_mtime > cur_time - 86400:
                        all_size += stat_info.st_size
                        if stat_info.st_size // 1024 <= 5:
                            f.write('\n{}\n\t大小：{}KB,  上傳時間：{}。'.format(path, stat_info.st_size // 1024, upload_time))

            f.write('\n\n {}本日備份大小：{}GB.\n'.format(item, all_size // 1024 // 1024 // 1024))

    with open(log_name, 'rb') as logfile:
        sendmail.sendmail('204-備份監控', str(logfile.read(), encoding='GBK'))

　　

sendmail.py

#!/usr/bin/env python
# Version = 3.5.2
# __auth__ = '無名小妖'
import smtplib
import email.mime.multipart
import email.mime.text


def sendmail(sub, con):
    msg = email.mime.multipart.MIMEMultipart()
    msg['from'] = "[e-mail]"
    msg['to'] = "[e-mail]"
    msg['subject'] = sub
    content = con
    txt = email.mime.text.MIMEText(content)
    msg.attach(txt)
    smtp = smtplib.SMTP()
    smtp.connect('smtp.mail.haoyisheng.com', '25')
    smtp.login('[e-mail]', 'xxxxxx')
    smtp.sendmail('[e-mail]', '[e-mail]',str(msg))
    smtp.quit()