Python - 检测U盘烧录小工具

自己不从事开发工作，方便自己工作时写了个小工具。

准备后面写成GUI图形化工具，先把代码保存在这里

待完善...

import os
import string
import time


# 获取磁盘列表，返回list
def get_disklist():
    disk_list = []
    for i in string.ascii_uppercase:
        disk = i + \':\'
        if os.path.isdir(disk):
            disk_list.append(disk)
    return disk_list


# 读取文件方法
def read_file():
    # 最后一个（新增的）盘符路径赋给t
    t = get_disklist()[-1]
    # sn为需要读取的文本名称（存放SN码的文件）
    c = t + r\'\sn\'
    with open(c, \'r\') as files:
        str_num = files.read()
    return str_num


# 加密方法
def author(xin):
    #公司机密，不方便透露
    return str1


if __name__ == \'__main__\':
    # 计数器
    a = 0
    while True:
        # 3为可变动数字，请改为未插入U盘前的磁盘分区数
        if len(get_disklist()) > 3:
            read_num = read_file()
            print(\'读取到的初始值为:{}\'.format(read_num))
            # 2d27572e
            if read_num == \'2d27572e\':
                print(\'sn为初始值，请使用树莓派启动此内存卡后再使用此脚本\')
            else:
                # 调用author方法得到需要的结果值
                write_num = author(read_num)
                # print(write_num)
                w_way = get_disklist()[-1] + r\'\author\'
                # print(w_way)
                # 写入文件
                with open(w_way, \'w\') as f:
                    f.write(write_num)
                    print(\'写入成功\')
                print(\'写入值为:{}\'.format(write_num))
                # 烧写成功后计数器+1
                a += 1
                print(\'刚刚烧录的是第{}张内存卡\'.format(a))
                while True:
                    if len(get_disklist()) > 3:
                        print(\'请拔出U盘\')
                        time.sleep(2)
                    else:
                        break
        else:
            print(\'请插入新的U盘\')

        time.sleep(2)