【業務自動化】Chapter2-3：置換するファイル形式を選択式にしよう

pip install pyinputplus

import pyinputplus

# 整数入力（1～10の範囲）
age = pyinputplus.inputInt("年齢を入力してください (1-10): ", min=1, max=10)
print(f"あなたの年齢: {age}")

# 選択肢から入力
fruit = pyinputplus.inputChoice(['apple', 'banana', 'orange'])
print(f"選ばれたフルーツ: {fruit}")

# Yes/No 入力
proceed = pyinputplus.inputYesNo("続行しますか？ (yes/no): ")
if proceed == "yes":
    print("続行します！")
else:
    print("終了します。")

import csv

# CSVファイルを読み込む
with open("data.csv", newline="", encoding="utf-8-sig") as f:	# UTF-8（BOM付き）でファイルを開く
    reader = csv.reader(f)  # リーダーオブジェクトを作成
    rows = [row for row in reader]  # 行をリストに格納

print(rows)

import csv

# 書き込みたいデータ（2次元リスト）
rows = [
    ["名前", "年齢", "出身地"],
    ["太郎", "20", "東京"],
    ["花子", "25", "大阪"]
]

# CSVに書き込み
with open("output.csv", "w", newline="", encoding="utf-8-sig") as f:
    writer = csv.writer(f)   # writerオブジェクトを作成
    writer.writerows(rows)   # まとめて書き込み

import openpyxl     # xlsxを読み取るための外部モジュール
import pathlib      # ファイルやフォルダのパスを扱うための標準モジュール
import sys          # 実行環境を判定するために使用（py または exe）
import pyinputplus  # 入力値のチェックを行う外部モジュール
import csv          # CSVファイルを読み書きするための標準モジュール

# 実行環境に応じて基準ディレクトリを決定（exeかpyファイルか）
if getattr(sys, 'frozen', False):
    BASE_DIR = pathlib.Path(sys.executable).parent  # exe で動いている場合
else:
    BASE_DIR = pathlib.Path(__file__).parent        # pyファイルとして動いている場合
file_path = BASE_DIR / "workbooks"      # Excelファイルを格納するフォルダ
text_path = BASE_DIR / "input_files"    # 置換するファイルを格納するフォルダ

# Excelファイルを読み込み、シートを選択
book = openpyxl.load_workbook(file_path / "Chapter2-2.xlsx")
sheet = book.active

# 置換するファイル形式を選択
file_type = pyinputplus.inputChoice(["txt", "csv"],
    "置換するファイル形式を選択してください（txt/csv）"
    )

# 置換リスト（Excelから読み込む検索語・置換語）の定義
READ_START_ROW_NO = 3   # データ読み込みを開始する行
OLD_WORD_COL_NO = 2     # 検索する文字列が書かれた列
NEW_WORD_COL_NO = 3     # 置換後の文字列が書かれた列
old_words = []          # 検索する文字列のリスト
new_words = []          # 置換後の文字列のリスト

# Excelから置換リストを読み込む
for i in range(READ_START_ROW_NO, sheet.max_row+1): # データが書かれた全ての行に対して
    old_word = sheet.cell(i, OLD_WORD_COL_NO).value # 検索語を読み取り
    if old_word is None:            # 検索語が空白なら
        break                       # 繰り返しを終了
    old_words.append(str(old_word)) # 検索語をリストに追加

    new_word = sheet.cell(i, NEW_WORD_COL_NO).value # 置換語を読み取り
    if new_word is None:            # 置換語が空白なら
        new_word = old_word         # 検索語をそのまま使う
    new_words.append(str(new_word)) # 置換語をリストに追加

# テキストファイルを処理
file_counter = 0
if file_type == "txt":
    for target_file in text_path.glob(f"*.{file_type}"): # フォルダ内の全てのテキストファイルに対して
        # 改行して置換開始の宣言
        print()
        print(f'{target_file.name}のテキストを置換します')
        # テキストファイルを読み込み、検索語を置換語に置き換え
        file_text = target_file.read_text(encoding = "UTF-8")
        for i in range(len(old_words)):
            file_text = file_text.replace(old_words[i], new_words[i])
        # 新しいファイルを作成
        new_file_path = target_file.parent / f"【表記ゆれ修正済み】{target_file.stem}.{file_type}"
        new_file_path.write_text(file_text, encoding="UTF-8")
        file_counter += 1

# CSVファイルを処理
elif file_type == "csv":
    for target_file in text_path.glob(f"*.{file_type}"):  # フォルダ内の全てのCSVファイルに対して
        print()
        print(f'{target_file.name}のCSVを置換します')

        # CSVを読み込む
        with open(target_file, newline='', encoding="utf-8-sig") as f:
            reader = csv.reader(f)
            rows = [row for row in reader]

        # セル単位で置換処理
        for i in range(len(old_words)):
            rows = [[cell.replace(old_words[i], new_words[i]) for cell in row] for row in rows]

        # 新しいCSVファイルを作成
        new_file_path = target_file.parent / f"【表記ゆれ修正済み】{target_file.stem}.{file_type}"
        with open(new_file_path, "w", newline='', encoding="utf-8-sig") as f:
            writer = csv.writer(f)
            writer.writerows(rows)

        file_counter += 1

# 改行して終了
print()
print(f'{file_counter}ファイル置換完了')
input("Enterキーでプログラムを終了します")

import openpyxl     # xlsxを読み取るための外部モジュール
import pathlib      # ファイルやフォルダのパスを扱うための標準モジュール
import sys          # 実行環境を判定するために使用（py または exe）
import pyinputplus  # 入力値のチェックを行う外部モジュール
import csv          # CSVファイルを読み書きするための標準モジュール

# 実行環境に応じて基準ディレクトリを決定（exeかpyファイルか）
if getattr(sys, 'frozen', False):
    BASE_DIR = pathlib.Path(sys.executable).parent  # exe で動いている場合
else:
    BASE_DIR = pathlib.Path(__file__).parent        # pyファイルとして動いている場合
file_path = BASE_DIR / "workbooks"      # Excelファイルを格納するフォルダ
text_path = BASE_DIR / "input_files"    # 置換するファイルを格納するフォルダ

# Excelファイルを読み込み、シートを選択
book = openpyxl.load_workbook(file_path / "Chapter2-2.xlsx")
sheet = book.active

# 置換するファイル形式を選択
file_type = pyinputplus.inputChoice(["txt", "csv"],
    "置換するファイル形式を選択してください（txt/csv）"
    )

# 置換リスト（Excelから読み込む検索語・置換語）の定義
READ_START_ROW_NO = 3   # データ読み込みを開始する行
OLD_WORD_COL_NO = 2     # 検索する文字列が書かれた列
NEW_WORD_COL_NO = 3     # 置換後の文字列が書かれた列
old_words = []          # 検索する文字列のリスト
new_words = []          # 置換後の文字列のリスト

# Excelから置換リストを読み込む
for i in range(READ_START_ROW_NO, sheet.max_row+1): # データが書かれた全ての行に対して
    old_word = sheet.cell(i, OLD_WORD_COL_NO).value # 検索語を読み取り
    if old_word is None:            # 検索語が空白なら
        break                       # 繰り返しを終了
    old_words.append(str(old_word)) # 検索語をリストに追加

    new_word = sheet.cell(i, NEW_WORD_COL_NO).value # 置換語を読み取り
    if new_word is None:            # 置換語が空白なら
        new_word = old_word         # 検索語をそのまま使う
    new_words.append(str(new_word)) # 置換語をリストに追加

# テキストファイルを処理
file_counter = 0
if file_type == "txt":
    for target_file in text_path.glob(f"*.{file_type}"): # フォルダ内の全てのテキストファイルに対して
        # 改行して置換開始の宣言
        print()
        print(f'{target_file.name}のテキストを置換します')
        # テキストファイルを読み込み、検索語を置換語に置き換え
        file_text = target_file.read_text(encoding = "UTF-8")
        for i in range(len(old_words)):
            file_text = file_text.replace(old_words[i], new_words[i])
        # 新しいファイルを作成
        new_file_path = target_file.parent / f"【表記ゆれ修正済み】{target_file.stem}.{file_type}"
        new_file_path.write_text(file_text, encoding="UTF-8")
        file_counter += 1

# CSVファイルを処理
elif file_type == "csv":
    for target_file in text_path.glob(f"*.{file_type}"):  # フォルダ内の全てのCSVファイルに対して
        print()
        print(f'{target_file.name}のCSVを置換します')

        # CSVを読み込む
        with open(target_file, newline='', encoding="utf-8-sig") as f:  # UTF-8（BOM付き）でファイルを開く
            reader = csv.reader(f)          # CSVリーダーを作成
            rows = [row for row in reader]  # 各行をリストに格納（2次元リスト）

        # セル単位で置換処理
        for i in range(len(old_words)):
            # 行ごと(row)にセル(cell)を走査して文字列置換
            rows = [[cell.replace(old_words[i], new_words[i]) for cell in row] for row in rows]

        # 新しいCSVファイルを作成
        new_file_path = target_file.parent / f"【表記ゆれ修正済み】{target_file.stem}.{file_type}"
        with open(new_file_path, "w", newline='', encoding="utf-8-sig") as f:
            writer = csv.writer(f)  # CSVライターを作成
            writer.writerows(rows)  # 置換後のデータを書き込み
        file_counter += 1

# 改行して終了
print()
print(f'{file_counter}ファイル置換完了')
input("Enterキーでプログラムを終了します")

pyinstaller --onefile Chapter2-3.py