Divide data in a folder into batches

FolderBatchMaker.py

# -*- coding: utf-8 -*-
"""
Created on Wed May 31 12:08:35 2017

@author: Qaisar.Tanvir
"""

import os
import sys
import shutil

main = r"folder"

count = len(os.listdir(main))
batchsize = 6
postfix = 0

for i,f in enumerate(os.listdir(main)):
    batchfolder = os.path.join(main,"batch "+ str(postfix))
    if not os.path.exists(batchfolder):
        os.makedirs(batchfolder)
    if(i % (count/batchsize) == 0):
        postfix += 1
        if not os.path.exists(os.path.join(main,"batch "+ str(postfix))):
            os.makedirs(os.path.join(main,"batch "+ str(postfix)))
            batchfolder = os.path.join(main,"batch "+ str(postfix))
            shutil.copy2(os.path.join(main,f),batchfolder)
    else:
        shutil.copy2(os.path.join(main,f),batchfolder)
    print i