ファイル数に応じてfor文を動かしたい

Question

任意の容量でファイル（写真or動画）をフォルダに分けたく以下のコードを書いてみました。

usern = 'YaKa'
mypath = 'C:/Users/'+ usern +'/Pictures/test-pic-folder/'
dvdsize = 10000000

import os
import shutil

for i in '123456789':
    os.mkdir(mypath + i)
    files = os.listdir(mypath)
    size = 0
    for f in files:
        if os.path.isfile(mypath + f):            
            size += os.path.getsize(mypath + f)
            if size < dvdsize:
                shutil.move(mypath + f, mypath + i)

一応目的どおり動くようですが、for i in '123456789': だとフォルダの数が限られるし、ファイル数によっては無駄なフォルダが生じます。
ファイル数に応じられる構造にしたいのですがどうすればよいでしょうか？

Answer 1

フォルダの数でループを回すんじゃなく、ファイルの残り容量でループを回す。
保存した容量を引いて行って、０以下になると終了になるようにすればいい

ああ、ファイルが複数あるのね
なら、ファイル単位でループ回して、保存容量に達するごとにカウンタをインクリメント、
最終的に残りファイルが０になったらループ終了

Comments

[YaKa]

ありがとうございます。
すぐにはつくりかえられませんが、トライしてみます。

Answer 2

ファイルでループを回すというアドバイスありがとうございます。保存容量に達するごとにカウンタをインクリメントというのが理解できなかったのですが、以下のコードはいかがでしょうか。（やはり最初の投稿のコードではうまく動いてなかったです。）

usern = 'YaKa'
mypath = 'C:/Users/'+ usern +'/Pictures/test-pic-folder/'
dvdsize = 11 * (10 ** 6)

import os
import shutil
    
# ファイルのリストを作る
files = os.listdir(mypath)

# フォルダ1を作る
i = 1
os.mkdir(mypath + str(i))
# 容量カウンタを設定する
size = 0

# ファイルでループ
for f in files:
    # 容量カウンタを増加させ、もしdvdsize未満だったらファイルを移動する。dvdsize以上だったらカウンタリセット＆i+1のフォルダをつくり、そこへ移動。
    size += os.path.getsize(mypath + f)
    if size < dvdsize:
        shutil.move(mypath + f, mypath + str(i))
    else:
        i += 1
        os.mkdir(mypath + str(i))
        # 容量カウンタはリセット
        size = os.path.getsize(mypath + f)
        shutil.move(mypath + f, mypath + str(i))