Python批量整理文件夹，将文件按指定数目分文件夹存放

我是为了提高图片浏览效率，批量整理文件夹
将图像样本数多于2000个的文件夹按 2000个图像样本/一个子文件夹 存放
#子文件夹命名规则为01，02，03…

import os
import shutil
import glob


if __name__ == '__main__':
    # 需整理的文件夹
    srcpath = input("请输入需要整理的文件夹路径：" + "\n")

    for dirpath, dirname, filename in os.walk(srcpath):
        # 子文件夹数，子文件夹命名为‘01’，‘02’...
        n = 1

        for file in filename:
            # 子文件夹
            dstpath = srcpath + "//" + '0' + str(n)
            # 图像样本计数
            filenum = len(glob.glob(dstpath + "/*"))

            while os.path.exists(dstpath) and filenum >= 2000:
                n += 1
                dstpath = srcpath + "//" + '0' + str(n)
                filenum = len(glob.glob(dstpath + "/*"))

            #待整理样本文件数
            num = (len(glob.glob(srcpath + "/*"))) - n
            if num == 0:
                print("文件整理结束")
                quit()

            if not os.path.exists(dstpath):
                os.makedirs(dstpath)

            src_file = os.path.join(srcpath, file)
            shutil.move(src_file, dstpath)