Pythonにおけるファイルのコピーとリネームの同時実行

Question

以下のように、ディレクトリ名とファイル名に基づいて、ファイルを1つのディレクトリに
コピー（移動）したいのですが、Pythonでどのように実装したらいいでしょうか？

'''現在（4000ファイル程度。スクリプトファイルは親ディレクトリに設置）
./TEXT1/AA/1.txt
./TEXT1/BB/2.txt
./TEXT1/BB/3.txt
./TEXT2/CC/1X.txt
'''

'''移動とリネーム後
./NEW_TEXT/AA1.txt
./NEW_TEXT/BB2.txt
./NEW_TEXT/BB2.txt
./NEW_TEXT/CC1X.txt
'''

os.copyとos.renameがあるのは把握しているのですが、
for文で回す時にコピーする際、ディレクトリ名を引用してファイルを同時にリネームする方法がわかりませんでした。お力添えいただけたら幸いです。

Answer 1

Pythonなど不要。そう、シェル芸ならね！

$ find . -type f -name '*.txt'|sed -n -E 's|./[^/]+/([^/]+)/([^/]+.txt)|cp -p & ./NEW_TEXT/\1\2|p'
cp -p ./TEXT1/AA/1.txt ./NEW_TEXT/AA1.txt
cp -p ./TEXT1/BB/2.txt ./NEW_TEXT/BB2.txt
cp -p ./TEXT1/BB/3.txt ./NEW_TEXT/BB3.txt
cp -p ./TEXT2/CC/1X.txt ./NEW_TEXT/CC1X.txt
$ !!|sh

Answer 2

Rubyだったらサクッと書けるんですが・・・

require 'fileutils'

files = Dir.glob("./TEXT*/*/*.txt")
files.each do |f|
  /TEXT.*?/(.*?)/(.*).txt/ =~ f
  after = "./NEW_TEXT/#{$~[1]}#{$~[2]}.txt"
  FileUtils.mv f, after
end

---

Pythonでも行けました。

import re
import glob
import shutil

files = glob.glob("./TEXT*/*/*.txt")
for f in files:
    m = re.match(r"./TEXT.*?/(.*?)/(.*).txt", f)
    after = "./NEW_TEXT/{0}{1}.txt".format(m.group(1), m.group(2))
    shutil.copy(f, after)

Comments

[MagMag]

ありがとうございました！

Answer 3

とりあえずglob.globでコピー元ファイルのパスのリストでも作ってください。

Pythonでフォルダ内のファイルリストを取得する - Qiita

宛先ファイル名は、まあos.pathとかに色々あるといえばありますが、とりあえず簡単にやりたいのなら文字列加工でこんな感じで作れば良いんじゃないでしょうか。

>>> "./NEW_TEXT/"+"".join("./TEXT1/AA/1.txt".split("/")[-2:])
'./NEW_TEXT/AA1.txt'

あとはコピー元ファイルパスのリストをforで回して、os.copyすればよいのでしょう。

Comments

[MagMag]

ありがとうございました。正規表現がわからない自分でも実装ができました！