【初心者歓迎】CSVファイル分割ツールを作ろう

import os
import customtkinter as ctk
from customtkintercontrols import App
from customtkintercontrols import FileSelectionControl
from customtkintercontrols import FolderSelectionControl
from customtkintercontrols import EntryControl
import csv_spliter as csvspl
from appconfig import AppConfig 
# -----------------------------------------------
# コールバック関数の定義 
# -----------------------------------------------

# ウィンドウを閉じる時に呼ばれる関数
def close_window():
    # 画面入力値を保存
    conf.set("file_selection",file_selection.get())
    conf.set("input_encoding_name",input_encoding_name.get())
    conf.set("folder",folder_selection.get())
    conf.set("output_encoding_name",output_encoding_name.get())
    conf.set("input_header_cnt",input_header_cnt.get())
    conf.set("split_count",split_count.get())
    conf.set("contain_words",contain_words.get())
    conf.set("not_contain_words",not_contain_words.get())
    conf.write()
    # プログラムの終了
    app.quit()

# 変換ボタンが押された時に呼ばれる関数
def split():
    csv_file = file_selection.get()
    in_encoder = input_encoding_name.get()
    out_folder = folder_selection.get()
    out_encoder = output_encoding_name.get()
    head_cnt = input_header_cnt.get()
    spl_cnt = split_count.get().strip()
    spl_cnt = "0" if spl_cnt == "なし" or spl_cnt == "" else spl_cnt
    varid_words = contain_words.get() if is_contain.get() != 0 else ""
    omit_wors = not_contain_words.get() if is_not_contain.get() != 0 else ""
    csvspl.split_file(csv_file,out_folder,int(head_cnt),int(spl_cnt),in_encoder,out_encoder,varid_words,omit_wors)
# -----------------------------------------------
# メインウインドウの定義
# -----------------------------------------------

# 前回入力値の保存・復元用インスタンスの生成
conf = AppConfig("./config.json")

# メインウィンドウの作成
app = App()
app.minsize(width=500, height=500)
app.geometry(f"800x550")
app.title("CSVファイル分割")

# customtkinterをダークモードに設定
ctk.set_appearance_mode("light")

# ウィンドウを閉じる時に呼びたい関数を登録
app.protocol("WM_DELETE_WINDOW", close_window)

# -----------------------------------------------
# 画面レイアウト（ウィジェット）の定義 
# -----------------------------------------------

# ファイル選択ダイアログを配置
file_selection = FileSelectionControl(app)
file_selection.label.configure(text=" 　　CSVファイル名")
file_selection.pack(padx=10,pady=10,fill="x")

# 文字コードとヘッダ行数の指定ウィジェットを配置
group = ctk.CTkFrame(app)
label = ctk.CTkLabel(group , text="  　　　　文字コード")
label.pack(padx=10,pady=10,side="left")

# 入力ファイル文字コードコンボボックスを配置
input_encoding_name = ctk.CTkComboBox(group , values=["SHIFT-JIS","UTF-8"])
input_encoding_name.pack(padx=10,pady=10,side="left")

# ヘッダ行数の選択コンボボックスを配置
label = ctk.CTkLabel(group , text="ヘッダ行数")
label.pack(padx=10,pady=10,side="left")
input_header_cnt = ctk.CTkComboBox(group , values=["0","1","2","3","4","5"])
input_header_cnt.set("1")
input_header_cnt.pack(padx=10,pady=10,side="left")
group.pack(padx=10,pady=10,fill="x")

# フォルダ選択ダイアログを配置
folder_selection = FolderSelectionControl(app)
folder_selection.label.configure(text="保存先フォルダ名")
folder_selection.pack(padx=10,pady=10,fill="x")

# 出力ファイル文字コードコンボボックスを配置
group = ctk.CTkFrame(app)
label = ctk.CTkLabel(group , text="  　　　　文字コード")
label.pack(padx=10,pady=10,side="left")
output_encoding_name = ctk.CTkComboBox(group , values=["SHIFT-JIS","UTF-8"])
output_encoding_name.pack(padx=10,pady=10,side="left")
group.pack(padx=10,pady=10,fill="x")


# 分割行数選択コンボボックスを配置
group = ctk.CTkFrame(app)
label = ctk.CTkLabel(group , text="  　　　　　 分割行数")
label.pack(padx=10,pady=10,side="left")
split_count = ctk.CTkComboBox(group , values=["なし","1000","2000","5000","10000","20000","50000"])
split_count.pack(pady=10,side="left")
group.pack(padx=10,pady=10,fill="x")

# 抽出対象のキーワード入力欄を配置
group = ctk.CTkFrame(app)
is_contain = ctk.CTkCheckBox(group,text="キーワードを含む行を抽出",width=90)
is_contain.pack(padx=10,pady=10,side="left")
contain_words = EntryControl(group)
contain_words.pack(padx=10,pady=10,fill="x", expand=True)
group.pack(padx=10,pady=10,fill="x")

# 除外対象のキーワード入力欄を配置
group = ctk.CTkFrame(app)
is_not_contain = ctk.CTkCheckBox(group,text="キーワードを含む行を除外",width=90)
is_not_contain.pack(padx=10,pady=10,side="left")
not_contain_words = EntryControl(group)
not_contain_words.pack(padx=10,pady=10,fill="x", expand=True)
group.pack(padx=10,pady=10,fill="x")

# 実行ボタンの配置
split_button = ctk.CTkButton(app,text="CSVファイルの分割",height=40,width=300,command=split)
split_button.pack(padx=20,pady=10,side="top", anchor="e")


# -----------------------------------------------
# 前回実行時の画面入力値を復元
# -----------------------------------------------
file_selection.set(conf.get("file_selection",""))
input_encoding_name.set(conf.get("input_encoding_name","SHIFT-JIS"))
input_header_cnt.set(conf.get("input_header_cnt","1"))
output_encoding_name.set(conf.get("output_encoding_name","SHIFT-JIS"))
split_count.set(conf.get("split_count",""))
contain_words.set(conf.get("contain_words",""))
not_contain_words.set(conf.get("not_contain_words",""))

# CSVファイル名を復元
folder_selection.set(conf.get("csvfile"))

# 保存先フォルダの値を復元
folder_selection.set(conf.get("folder"))
if folder_selection.get() == "":
    folder_selection.set(os.getcwd())


# -----------------------------------------------
# CustomTkinterの起動 
# -----------------------------------------------

# メインループを開始
app.mainloop()

import os

def split_file(input_file, output_folder, header_count=1,line_max=10000,
               input_encoding="shift-jis",output_encoding="shift-jis",valid_words="",omit_words=""):
    """
    指定されたファイルを指定された行数ごとに分割します。

    Args:
        source_file (str): 分割対象のファイルパス
        output_dir (str): 出力フォルダパス
        header_rows (int, optional): ヘッダー行数 (デフォルト: 0)
        lines_per_file (int, optional): 1ファイルあたりの最大行数 (デフォルト: 10000)
        source_encoding (str, optional): 入力ファイルのエンコーディング (デフォルト: "shift-jis")
        output_encoding (str, optional): 出力ファイルのエンコーディング (デフォルト: "shift-jis")
        valid_patterns (str, optional): 継続行として扱う文字列 (デフォルト: "")
        omit_patterns (str, optional): 除外する文字列 (デフォルト: "")

    Returns:
        None
    """

    # 出力ファイル名とファイルポインタを作成する関数
    def get_new_fp(file,no,output_folder,enc_name):
        name,ext = os.path.splitext(file)
        path = os.path.join(output_folder,os.path.basename(name))
        return  open(f"{path}_{no}{ext}", 'w', encoding=enc_name)

    # １行の中に指定したキーワードが含まれているかをチェックする関数
    def match(line,words):
        return any(word in line for word in words.split(','))

    headers = [] # ヘッダーを格納するリスト
    file_no = 0  # 出力ファイルの連番
    line_no = 0  # 行番号
    body_count = 0 # ボディの行数カウンタ
    output_fp = get_new_fp(input_file,file_no,output_folder,output_encoding) # 出力ファイルポインタ

    with open(input_file, 'r', encoding=input_encoding) as input_fp: 
        for line in input_fp:
            # ヘッダをリストに保存する
            if line_no < header_count :
                headers.append(line)
            else :
                # 指定したキーワードが含まれていなければ読み飛ばす
                if valid_words != "" and not match(line,valid_words):
                    continue
                # 指定したキーワードが含まれていたら読み飛ばす
                if omit_words != "" and match(line,omit_words):
                    continue
                               
                # ボディの行数が分割数に達すると新しいファイルを作成
                if line_max > 0 and body_count >= line_max :
                    output_fp.close()
                    output_fp = get_new_fp(input_file,file_no,output_folder,output_encoding) 
                    # ヘッダを書き込む
                    output_fp.writelines(headers)
                    file_no += 1
                    body_count = 0
                # ボディのカウントを１つ増やす
                body_count += 1
            # ファイルに書き込む
            output_fp.write(line)
            # 行番号を１つ増やす
            line_no += 1

        if output_fp is not None:
            output_fp.close()