フォルダ及びファイルの名前の最大値を抜き出して管理したい

Question

[]はフォルダです。

A:\というフォルダ内に
[data]
data.mp4

Bというフォルダ内に

[data-01]
[data-02]
[data-03]
data-01.mp4
data-02.mp4
data-03.mp4

という構造の時に

A内の[data]とdata.mp4を
[data-04]
data-04.mp4
にリネームしてBに移動するプログラムを書きたいのです。

リネームする-??の数値は、B内に数あるdata-??の最大値に+1してリネームされるようにしたいです。

なお、名前のdataの部分は他のプログラムとの兼ね合いがあるので、いじらない形で管理したいです。

リネームも移動も書けるのですが、フォルダリストから最大値を抜き出すという段階でつまづいてしまいました。

追記及び修正

import shutil
import os
from pathlib import Path

output_dir = Path("A:/output")
input_dir = Path("A:/input")

dataList = []

for path in output_dir.glob('*.*'):
    dataName =  path.name.split("-")[1]
    dataNumber = os.path.splitext(dataName)[0]

    dataList.append(dataNumber)
    
RenameNumber = int(max(dataList)) + 1
RenameNumberName = 'data-' + str(RenameNumber).zfill(2)

shutil.move(str(input_dir) + "/data.txt", str(output_dir) + "/" +  RenameNumberName + ".txt")
shutil.move(str(input_dir) + "/data", str(output_dir) + "/" + RenameNumberName )

以上でなんとか動くようにはなりました…。

質問は2点です。
①上記コードの綺麗な書き方
②上記コードでinput_dirにpathを書いているのですが、実際はこのpyファイルの実行場所を取得したいので、その書き方

よろしくお願い致します。

Comments

[meg_]

複数の数字の中から最大値を求めることは出来ますか？

[gentaro]

「pythonまたはバッチファイルでお願い致します」っていうのは、制作依頼ですか？
ガイドライン違反ですが。
https://teratail.com/help/avoid-asking

Answer 1

①上記コードの綺麗な書き方

綺麗かどうかは微妙ですが、こんな感じでしょうか。

• できるだけpathlibの機能だけを使う
• globパターンの記述は決め打てる限り詳細に
• 変数の名前はPythonの慣習に従って lower_snake に統一
• 読める範囲(個人差有り)で内包表記やジェネレータ式を積極的に使う

import pathlib


output_dir = pathlib.Path(r'A:\output')
input_dir = pathlib.Path(r'A:\input')

# 0123456789
# data-XX.mp4
new_num = 1 + max(
    int(n) for n in map(
        lambda p: p.stem[5:], output_dir.glob('data-[0-9][0-9].mp4')
    )
)

(input_dir / 'data.mp4').rename(
    output_dir / f'data-{new_num:02d}.mp4'
)
(input_dir / 'data').rename(
    output_dir / f'data-{new_num:02d}'
)

正しく動作する保証は無いので、ダミーデータを使うか／コピーを取ってから試してみて下さい。

---

②上記コードでinput_dirにpathを書いているのですが、実際はこのpyファイルの実行場所を取得したいので、その書き方

• カレントディレクトリという意味なら

input_dir = pathlib.Path('.')

• 実行ファイルのあるディレクトリという意味なら

input_dir = pathlib.Path(__file__).parent

Comments

[SILASSETH]

ありがとうございます。勉強になります。

Answer 2

たんに01から始めて、一致したファイル名があれば、数字を＋１して、一致したファイルが存在しなくなるまで繰り返せばいいだけでは

Comments

[SILASSETH]

こんな感じで書いてみたのですが、このコードは間違っており動きません…。

x = 1
for F in os.path.exists( str(output_dir) + "/" + 'data-' + str(x).zfill(2) +'.txt') :
if os.path.exists( str(output_dir) + "/" + 'data-' + str(x).zfill(2) +'.txt') :
x = int(x) +1
continue

else:
移動リネーム
break

for F in os.path.exists( str(output_dir) + "/" + 'data-' + str(x).zfill(2) +'.txt') :
の部分の書き方が分からないので、アドバイスいただきたいです。

[y_waiwai]

x=1
while true:
fname=ファイル名生成
if not os.path.exist(fname) :
break
x++;

というふーにして、fnameをプリントするなりしてどういう内容になってるのか確認してみよう

[SILASSETH]

無事動くコードが書けました。ありがとうございます。

Answer 3

①上記コードの綺麗な書き方

特に問題ないと思います。
強いて言えば、

RenameNumberName = f'data-{RenameNumber:02}'

でしょうか。
あと、splitextを先にやってからsplitをするほうが自然でしょう。

実際はこのpyファイルの実行場所を取得したい

「このpyファイルの格納場所」ですよね？

input_dir = Path(os.path.dirname(__file__))

です。

Comments

[SILASSETH]

ありがとうございます。無事、思い通りの処理ができるようになった上、勉強にもなりました。