Python获取磁盘、文件夹大小信息（附邮件发送）（二）

def sendmail(sub, con):
    msg = email.mime.multipart.MIMEMultipart()
    msg['from'] = "xxx.com" # 发邮件的邮箱地址
    msg['to'] = "yyy.com" # 收邮件的邮箱地址
    msg['subject'] = sub # 邮件标题
    content = con # 邮件内容
    txt = email.mime.text.MIMEText(content)
    msg.attach(txt)
    smtp = smtplib.SMTP()
    smtp.connect('smtp.126.com') #使用的邮件服务
    smtp.login('xxx.com', 'zzz') # 登录发邮件的邮箱账号和密码
    smtp.sendmail(msg['from'], msg['to'],str(msg)) # 发邮件格式
    smtp.quit()
    
sendmail("当前磁盘的运行状态", str(ddddd), encoding='GBK' ) # str(ddddd)需要发送的内容

"""
Author:NoamaNelson
Date:2019-11-19
Discription:Get the size of D:\\image, and then this folder will keep writing files. 
When the remaining space of D disk is less than 5GB, a warning message will be given
"""

import os
import os.path
import smtplib
import email.mime.multipart
import email.mime.text
import sendmail
import psutil
import collections


def get_size(path):
    list1 = []
    fileList = os.listdir(path)  # 获取path目录下所有文件
    for filename in fileList:
        pathTmp = os.path.join(path,filename)  # 获取path与filename组合后的路径
        if os.path.isdir(pathTmp):   # 判断是否为目录
            get_size(pathTmp)        # 是目录就继续递归查找
        elif os.path.isfile(pathTmp):  # 判断是否为文件
            filesize = os.path.getsize(pathTmp)  # 如果是文件，则获取相应文件的大小
            #print('目录中的子文件大小：%d字节' % filesize)
            list1.append(filesize)      # 将文件的大小添加到列表
    print('%s 目录中的文件总大小：%d 字节' % (path, sum(list1)))
    print('%s 目录中的文件总大小: %.4f MB' % (path, (sum(list1)/1024/1024)))
    print('%s 目录中的文件总大小: %.4f GB' % (path, (sum(list1)/1024/1024/1024)))
    return sum(list1)
    
def get_disk_info():
    disk_used = {}
    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)
        #print(disk_info)
        disk_used[s + '盘使用率：'] = '{}%'.format(disk_info.percent)
        disk_used[s + '剩余空间：'] = '{}GB'.format(disk_info.free // 1024 // 1024 // 1024)
    print("sdsds:%s"%disk_used)
    return disk_used

def sendmail(sub, con):
    msg = email.mime.multipart.MIMEMultipart()
    msg['from'] = "xxx.com" # 发邮件的邮箱地址
    msg['to'] = "yyy.com" # 收邮件的邮箱地址
    msg['subject'] = sub # 邮件标题
    content = con # 邮件内容
    txt = email.mime.text.MIMEText(content)
    msg.attach(txt)
    smtp = smtplib.SMTP()
    smtp.connect('smtp.126.com') #使用的邮件服务
    smtp.login('xxx.com', 'zzz') # 登录发邮件的邮箱账号和密码
    smtp.sendmail(msg['from'], msg['to'],str(msg)) # 发邮件格式
    smtp.quit()

if __name__ == "__main__":   
    #path= input("输入路径：").strip()  #指定文件路径
    log_name = 'result.log'
    with open(log_name, 'w') as f:
        path = r"D:\\image"
        disk_used1 = get_disk_info()
        for k, v in disk_used1.items():
            f.write('{}{}。\n'.format(k, v))
        list2 = get_size(path)
        f.write('%s 目录中的文件总大小：%d 字节\n' % (path, list2))
        f.write('%s 目录中的文件总大小: %.4f MB\n' % (path, (list2/1024/1024)))
        f.write('%s 目录中的文件总大小: %.4f GB\n' % (path, (list2/1024/1024/1024)))
        disk_used2 = disk_used1['D剩余空间：']
        disk_used3 = int(disk_used2.split('G')[0])
        #intlist2 = intlist1/1024/1024/1024
        if disk_used3 > 20:
            print("D剩余空间：%s"%disk_used1['D剩余空间：'])
            f.write("D剩余空间：%s"%disk_used1['D剩余空间：'])
        elif disk_used3 <= 20:
            print("D盘剩余空间为%dGB,建议停止数据存储"%disk_used3)  
            f.write("D盘剩余空间为%dGB,建议停止数据存储"%disk_used3)
    with open(log_name, 'rb') as logfile:
        sendmail("磁盘当前存储状态", str(logfile.read(), encoding='GBK'))