# 将malware2中的恶意代码以md5值命名后，整理到一个文件夹中
import os
import shutil

DIR = 'G:/malware2-laiyao'
MOVE_DIR = 'G:/malware_part1'

dirs = os.listdir(DIR)
print('dirs', dirs)

for dir in dirs:
    if not os.path.isdir(os.path.join(DIR, dir) ):
        continue
    files = os.listdir(os.path.join(DIR, dir))
    print('dir', dir)
    print('files', files)
    for file in files:
        md5 = file.split('=')[-1]
        shutil.copy(os.path.join(DIR, dir, file), os.path.join(MOVE_DIR, md5))
