『退屈なことはPythonのやらせよう』その7(9章: ファイル管理)

#! python3
# check09-1.py  -  ディレクトリツリーを渡り歩いて、.pdfや.jpgなどの特定の拡張子を持つファイルを見つけ
#                  新しいフォルダにコピーするプログラム
# Usage
# py check09_1.py <拡張子> <ファイルパス>

#ディレクトリツリーを渡り歩く
#.pdf, .jpgなどのファイルを見つける
# 新しいディレクトリを作り、そこに見つけたファイルを保存する

import os, sys, shutil, re
new_dir = os.mkdir("見つけたファイルの格納用のフォルダ(探索するフォルダと同じかそれより上の階層)")
regex_ext = re.compile(sys.argv[1])

for foldername, subfolders, filenames in os.walk(sys.argv[2]):
    print("The current folder is " + foldername)

    for subfolder in subfolders:
        print("SUBFOLDER OF " + foldername + ": " +subfolder)
    
    for filename in filenames:
        print("FILE INSIDE " + foldername + ": " + filename)
        mo = regex_ext.search(os.path.join(foldername,filename))
        if mo:
            shutil.copy(os.path.join(foldername,filename), ("見つけたファイルの格納用のフォルダ(探索するフォルダと同じかそれより上の階層)"))
    print("")

#! python3
# check09_2.py  -  ディレクトリツリーを渡りあるいて、100MB以上のファイルを見つける。
# 見つけたファイルは絶対パスを画面に表示する
# Usage
# py check09_2.py <path> <size>

import os, sys

for foldername, subfolders, filenames in os.walk(sys.argv[1]):

    for filename in filenames:
        #print("FILE INSIDE " + foldername + ": " + filename)
        if os.path.getsize(os.path.join(foldername, filename)) >= int(sys.argv[2]):
            print(os.path.join(foldername, filename))

#! python3
# check09_3_preparefile.py  -  fileの中に連番のファイルを作成する
# spam folderの中に、spam001 ~ spam020のテキストファイルを作る
# 3, 7, 13のファイルだけ作らない
# spamフォルダはコマンドラインで先に準備しておく

import os

path = os.getcwd()
file_path = os.path.join(path, "spam")

for i in range(1,21):
    if i == 3 or i == 7 or i ==13:
        continue
    created_file = open(os.path.join(file_path, "spam{:03}.txt".format(i)) , "w")
    created_file.write("neko{}".format(i))
    created_file.close()

#! python3
# check09_3.py  -  フォルダ中で、連番ファイルの抜けをみつけ、連番が飛んでいる箇所を見つける。
#                  番号の飛びを埋めるように、後に続くファイルの名前を変更する。
# Usage
# py check09_3.py <path> <pre>

import os, sys, re, shutil

#ファイル名を取得
dir_file = sys.argv[1]
pre = sys.argv[2]
file_dict = {}

file_list = os.listdir(dir_file)

for i in range(0, len(file_list)):
    file_dict[file_list[i]] = pre + "{:03}.txt".format(i+1)

#index的にはあるはずなのに、ファイル名として存在しないファイルを探す
def find_missing_file():
    for fname in file_dict.values():
        regex = re.compile(fname)
        mo = regex.search(", ".join(file_dict.keys()))
        if not mo:
            print(fname)


#元のファイル名から、連番の番号にする
def change_fname(change = False):
    if change == True:
        for k, v in file_dict.items():
            if k != v:
                print(k,v)
                shutil.move(os.path.join(dir_file, k), os.path.join(dir_file, v))
        
find_missing_file()

print("Do you want to change file name?:  y or [n]")
option = str(input())
if option == "y":
    change = True
else:
    change = False
change_fname(change)