自己动手写一个U盘拷贝小工具

这是五一期间，参照知乎上一篇的文章《十行代码--用python写一个USB病毒》写成的，最初只是单纯的想写成死循环，直到文件占满硬盘为止，第一个遇到的问题是，拷贝到硬盘之后，由于要无限次拷贝，所以每次的文件夹的名字不能重复，对于这个问题，开始是采用追加的方式，比如第一个文件夹叫udisk_bakcup，第二个文件夹就叫udisk_backup_demo，第三个叫udisk_backup_demo_demo...看到的效果是：

这样肯定是不行的，因为文件夹名字的长度好像不能超过256个字节，受一位开发同学的启发，用时间戳来表示，看到的效果明显好多了

后来想，能不能不写死循环，而是根据用户的需要选择拷贝次数，能不能做成界面，一切在遇到tkinter和PyInstaller后得到了解决

界面化实现过程不太容易，遇到的问题参见：《解决: PyInstaller打包后exe文件打开时出现failed to execute script》

生成图标后，还有一个问题：打包后生成的exe文件打开后，工具左上角的图标并没有加载进来，而直接运行代码是ok的(见上图)

利用if判断，做了一定的容错处理，那时我不知道还可以用正则做这件事，如果知道，可能会简化一下代码

再后来发现当U盘中的目录很大时，几百M到几个G，拷贝容易出现卡死的情况，可能由于是单线程，所以想着用多线程或协程的方法优化一下，谁知后来越来越忙就放下了，等到有机会再试一把

这个工具还是很好用的，可以将U盘所有的目录和文件拷进硬盘，自动在硬盘上创建一个文件夹来存储这些拷贝数据，还可以按照需要拷贝多份(默认是拷贝一份)

现将源码附上：

 '''
 version v1.0
 1. 实现了简单的pycharm运行功能
 2. 利用追加目录名的方法来实现添加n个目录，但受到目录名长度的限制
 
 version v2.0
 1. 利用时间戳表示目录名
 2. 利用tkinter实现了简单的GUI
 
 version v3.0
 1. 修改了部分GUI界面字符
 2. 利用datetime获取当前时间，并转化为字符串形式添加到目录
 
 version v4.0
 1. 修改了部分GUI界面
 
 version v5.0
 1. 限定了udisk_path, backup_path和num的范围
 2. 对以上输入进行容错处理
 3. 添加了计算程序运行时间的功能
 4. 添加了输入提示
 5. 不区分盘符的大小写
 
 version v6.0
 1. 更新了使用说明和字体、颜色
 2. 更新了软件图标
 3. 解决盘符为G:/的U盘不插入，不会弹出提示的问题。即在异常处理里增加了一个PermissionError的异常。
 
 version v6.1
 1. 解决了当U盘盘符为g:/, h:/时，弹出的错误信息
 2. 给拷贝次数一个默认值1
 3. 更新了使用说明
 '''
 
 import os, shutil, sys
 import time, datetime, tkinter
 import tkinter.messagebox as mb
 
 class Udisk:
 
     def __init__(self, backup_path = "D:/", udisk_path = "H:/", num = 1):
         self.backup_path = backup_path
         self.udisk_path = udisk_path
         self.num = num
 
     def resource_path(self, relative):
         if hasattr(sys, "_MEIPASS"):
             return os.path.join(sys._MEIPASS, relative)
         return os.path.join(relative)
 
     def copy_GUI(self):
         root = tkinter.Tk()
         root.title("U盘拷贝小工具 v6.1")
         root.geometry("400x260")
         icopath = self.resource_path(r"favicon-20180501101520441.ico")
         if os.path.exists(icopath):
             root.iconbitmap(icopath)
 
         tkinter.Label(root, text = "U盘盘符(如G:/)", font = ("华文中宋", 12)).grid(row = 0)
         tkinter.Label(root, text = "拷贝目录(如D:/)", font = ("华文中宋", 12)).grid(row = 1)
         tkinter.Label(root, text = "拷贝次数(如3)", font = ("华文中宋", 12)).grid(row = 2)
 
         self.input_1 = tkinter.Entry(root)
         self.input_2 = tkinter.Entry(root)
         self.input_3 = tkinter.Entry(root)
 
         self.input_3.insert(10, ")
 
         self.input_1.grid(row = 0, column = 1)
         self.input_2.grid(row = 1, column = 1)
         self.input_3.grid(row = 2, column = 1)
 
         tkinter.Button(root, text = "ok", font = ("Arial", 12), background = "MediumSlateBlue", command = self.run).grid(row=3,  pady=15, column=0)
         tkinter.Button(root, text = "exit", font = ("Arial", 12), background = "DarkGray", command = root.quit).grid(row=3, pady=15, column=1)
 
         tkinter.Label(root, text = "使用说明:\n1.本软件适用于windows系统;\n2.默认以/作为目录分隔符;\n3.目录名不能包含特殊字符\n4.默认拷贝次数为1", justify = "left", font = ("隶书", 12), pady = 20).grid(row = 4)
 
         root.mainloop()
 
     def dirExists(self):
         while True:
             if os.path.exists(self.backup_path) == True:
                 print("copytree方法的路径%s不能是已存在的!" % self.backup_path)
                 now = datetime.datetime.now()
                 self.backup_path += now.strftime('%Y-%m-%d %H-%M-%S.%f')
                 print("程序自动将路径重命名为: %s" % self.backup_path)
                 return self.dirExists()
             else:
                 break
 
     def copyUdisk(self):
         try:
             self.flag = True
             file_list = []
             content = os.listdir(self.udisk_path)
             self.dirExists()
         except FileNotFoundError:
             print("对不起，U盘不存在，请插入U盘重试! ")
             return False
         except PermissionError:
             print("对不起，U盘不存在，请插入U盘重试! ")
             self.flag = False
         else:
             for item_1 in content:
                 if os.path.isfile(os.path.join(self.udisk_path, item_1)) == False:
                     shutil.copytree(os.path.join(self.udisk_path, item_1), self.backup_path + "/" + item_1)
                 else:
                     file_list.append(item_1)
             for item_2 in file_list:
                 shutil.copy(os.path.join(self.udisk_path, item_2), self.backup_path + "/")
             print("已拷贝至", self.backup_path)
             return True
 
     def run(self):
         time_1 = time.time()
         try:
             self.udisk_path = self.input_1.get()
             self.backup_path = self.input_2.get()
             self.num = self.input_3.get()
             print(self.num)
             standard_backup_path = ["C:/", "c:/", "D:/", "d:/", "E:/", "e:/", "F:/", "f:/"]
             standard_udisk_path = ["G:/", "g:/", "H:/", "h:/"]
             if self.backup_path[0:3] in standard_backup_path and self.udisk_path[:] in standard_udisk_path and (type(int(self.num)) == int or self.num == ""):
                 for i in range(int(self.num)):
                     my_copy = Udisk(self.backup_path, self.udisk_path, self.num)
                     if my_copy.copyUdisk() == False:
                         mb.showerror("错误", "对不起，U盘不存在，请插入U盘重试! ")
                         break
                     elif my_copy.flag == False:
                         mb.showerror("错误", "对不起，U盘盘符不对应，请重试! ")
                         break
                     elif i == (int(self.num) - 1):
                         time_2 = time.time()
                         total_time = time_2 - time_1
                         output = "已拷贝成功:)"
                         mb.showinfo("结果", output)
                         mb.showinfo("共耗时", str(total_time)+"秒")
             elif self.udisk_path[:] not in standard_udisk_path:
                 if self.udisk_path == "":
                     mb.showerror("输入错误", "U盘盘符不能为空，请重试! ")
                 else:
                     mb.showerror("输入错误", "输入U盘盘符错误，请重试! ")
             elif self.backup_path[0:3] not in standard_backup_path:
                 if self.backup_path == "":
                     mb.showerror("输入错误", "拷贝目录不能为空，请重试! ")
                 else:
                     mb.showerror("输入错误", "输入拷贝目录错误，请重试! ")
         except ValueError:
                 mb.showerror("输入错误", "输入拷贝次数错误，请重试! ")
 
 mycopy = Udisk()
 mycopy.copy_GUI()