python使用diskpart命令执行脚本

写自动化测试的过程中经常会有磁盘格式化的一些需求,之前用uiautomation实现了,比较复杂,于是换成diskpart命令,记录一下。

函数封装

    def execute_diskpart_cmd(cmd_list: iter):
        diskpart_cmd = '\n'.join(cmd_list)
        txt_file = 'diskpart_cmd.txt'
        with open(txt_file, 'w', encoding='utf-8') as f:
            f.write(diskpart_cmd)
        for i in os.popen('diskpart /s diskpart_cmd.txt'):
            print(i)
        os.remove(txt_file)

格式指定分区

execute_diskpart_cmd(cmd_list=[f'select volume E', 'format fs=exFAT label=AUTOTEST override quick'])

慢格式化指定分区

execute_diskpart_cmd(cmd_list=[f'select volume E', 'format fs=exFAT label=AUTOTEST override'])

清除磁盘只读属性

execute_diskpart_cmd(cmd_list=['select disk 1','attributes disk clear readonly'])
Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐