shutilでファイル数が少ないフォルダを一括削除する

クロール先のデータがまとまった量で欲しい場合は、中途半端に取得されたフォルダは逆に邪魔になるので一括で削除したい。使いやすいように関数にしてまとめました。

使用環境

windows10
Anaconda 3.6.1

shutil

shutilはpython標準ライブラリの一つ。高度なディレクトリ操作が可能。
[公式レファレンス]https://docs.python.jp/3/library/shutil.html

コード

delete_folder.py

import os
import shutil

#デフォルトでは実行フォルダ内の要素３つ以下のフォルダを削除
def delete_folder( directory_dir= os.getwsd() ,size=3):

    folder_list = os.listdir(directory_dir)
    folder_dir = [os.path.join(directory_dir,i) for i in folder_list if len(os.listdir(os.path.join(directory_dir,i))) <= size ]
    for folder in folder_dir:
        print(folder+'を削除します')
        shutil.rmtree(folder)

    print('完了')