python程序怎么保存到u盘_Python 骚操作，自动拷贝U盘

阅读文本大概需要 6 分钟。

Python 这门语言有非常多有趣的内容，比如给微信好友自动发消息、查看微信好友撤回的消息、通过微信控制电脑等等。在我公众号上也分享过很多有趣的程序，近期我会去搜集一些骚操作的 Python 程序分享给大家，并命名「Python 骚操作」专栏，持续更新！文中一些程序可能来自别人的文章，但如果我觉得程序还有所改进的地方，我会继续添加一些代码，并且我会在文末标注原程序出处。

今天的「Python 骚操作」，分享的是一个自动拷贝电脑 U 盘内容的程序。当别人的 U 盘插入到你的电脑上时，Python 脚本会自动把 U 盘所有的内容，全部拷贝到你的电脑上。脚本每隔 5 秒扫描当前电脑是否有 U 盘插入，如果是已拷贝过的内容，则不会重复拷贝，每个相应的操作都会记录到日志文件中。

如果恰好这个 U 盘里有某些室友的「学习视频」，比如下的图片。

那骚气指数 10 颗星！

注意此程序只是用来学习交流，切勿违法使用。

Python 骚思路

在 windows 系统中，当有 U 盘插入到你的电脑上时，会给它分配一个磁盘名称，比如在我的电脑上分配的是「J」。所以我会先定义一个「USB_FILE」变量，表示 U 盘所在的目录。

我们利用 os 库来判断当前系统是否有「USB_FILE」，如果扫描到当前系统有这个目录，说明有 U 盘插入到当前电脑上。随后我们通过 shutil 库，把 U 盘里的内容全部拷贝到自己的电脑上。在每个操作之间我添加了 input_log() 方法，用来把当前相应的操作记录到日志文件中。

Python 骚源码

程序主要是由三个函数组成，get_dirs_size()、main()、input_log()。

get_dirs_size()

get_dirs_size() 主要是返回要拷贝 U 盘的长度，用来判断当前要拷贝的内容是否已经重复，如果重复则无需重新拷贝。

def get_dirs_size(dir):

size = 0

for root, dirs, files in os.walk(dir):

size += sum([getsize(join(root, name)) for name in files])

print(size)

return size

main()

main() 用来执行主逻辑代码，即每隔 5 秒检查当前电脑是否有 U 盘插入。如果检查到，则记录当前 U 盘的长度。在下一次检测的时候，如果当前的长度没有发生变化，则无需重新拷贝。每个操作都会记录到日志中，以便日后查看相应的拷贝信息。

def main():

old_size = 0

new_dirsize = 0

while(1):

if os.path.exists(USB_PATH):

print("检测到 U 盘")

input_log("检测到 U 盘")

new_dirsize = get_dirs_size(USB_PATH)

if old_size != new_dirsize:

for filename in os.listdir(USB_PATH):

print("拷贝了：" + filename)

input_log("拷贝了：" + filename)

shutil.copytree(USB_PATH, SAVE_PATH)

old_size = new_dirsize

else:

print("U 盘没有变化")

input_log("U 盘没有变化")

else:

print("暂时没有 U 判断")

print("开始休眠")

input_log("开始休眠")

sleep(5)

print("休眠结束，重新检测")

input_log("休眠结束，重新检测")

input_log() 利用 logging 库，来把相应的操作信息保存到一个日志文件中。

def input_log(message):

handler = logging.handlers.RotatingFileHandler(LOG_FILE, maxBytes=1024 * 1024, backupCount=5,

encoding='utf-8')  # 实例化handler

fmt = '%(asctime)s - %(levelname)s - %(message)s'

formatter = logging.Formatter(fmt)  # 实例化formatter

handler.setFormatter(formatter)  # 为handler添加formatter

logger = logging.getLogger('tst')  # 获取名为tst的logger

logger.addHandler(handler)  # 为logger添加handler

logger.setLevel(logging.DEBUG)

logger.info(message)

Python 骚教程

此次的程序使用非常简单，首先下载源码，后台回复「拷贝」即可获取源码。然后修改下 USB_PATH 和 SAVE_PATH，这两个参数表示的是你 U 盘所属的位置和你想要保存的目录。

LOG_FILE = "test.log"

USB_PATH = "J:\\"

SAVE_PATH = "D:\\disk_copy"

原程序参考来源：

本文首发于公众号「痴海」，后台回复「1024」即可获取最新编程资源。