3. Tabview

CustomTkinter

TabviewはCustomeTkinterにデフォルトで内蔵された便利な機能だが、タブボタンの名前を変更することができなかった

【解決策】タブ機能を自作でつくる

import subprocess
import sys
import os
import json
import sqlite3
import re
import customtkinter as ct
from datetime import datetime
from tkinter import messagebox 

CONFIG_FILE = "usb_auth.json"
DB_NAME = "passwords.db"


# ==========
# 挿入されているすべてのUSB情報を取得
# ==========
def get_usb_list():
    usb_dict = {}
    try:
        # PCに認識されている倫理ディスク(logicaldisk)を取得する
        # logicaldiskにはドライブレター(C:,D:,E:等)とその説明(LocalFixedDisk,RemovalDisk等)が書かれている
        # ここで「RemovalDisk」となっているものがUSBメモリやSDカードを意味する
        drives = subprocess.check_output(
            "wmic logicaldisk get name,description",
            shell=True
        ).decode("utf-8", errors="ignore").splitlines()
        # driveからRemovableを含む行(line)を探し、line.split()[0](最初のトークン、通常はE:のようなもの)を抜き出す
        removable_drives = []
        for line in drives:
            line = line.strip()
            if "Removable" in line:
                m = re.search(r"([A-Z]:)", line)
                if m:
                    removable_drives.append(m.group(1))
        # 初期値として「不明なUSB」をセット
        for drive in removable_drives:
            usb_dict[drive] = ("不明なUSB", "UNKNOWN")
        

        # 実際にPCに接続されているハードウェア情報(diskdrive)を取得する
        # 物理ディスク(接続ディスク)の製品名(Caption)とシリアル番号を取得
        result = subprocess.check_output(
            "wmic diskdrive get Caption,DeviceID,SerialNumber",
            shell=True
        ).decode("utf-8", errors="ignore").splitlines()   
        # resultからUSBを含む行(line)を探し、line.split()して分解
        for line in result:
            if "USB" in line:
                parts = line.split()
                if len(parts) >= 3:
                    caption = " ".join(parts[:-2]) #製品名
                    device_id = parts[-2] #DeviceID
                    serial = parts[-1] #シリアル番号

                    for d in usb_dict:
                        if usb_dict[d][1] == "UNKNOWN":
                            usb_dict[d] = (caption, serial)
                            break
    
    except Exception as e:
        print("USB取得エラー:", e)
    
    return usb_dict


# ==========
# 指定したserial(USBのシリアル番号)とドライブレターをJSONファイル(CONFIGファイル)に上書き保存する
# ==========
def save_usb(serial, drive):
    with open(CONFIG_FILE, "w") as f:
        json.dump({"serial": serial, "drive": drive}, f)



# ==========
# CONFIGファイルをロードして登録したUSBのシリアル番号を取得
# ==========
def load_usb():
    if os.path.exists(CONFIG_FILE):
        with open(CONFIG_FILE, "r") as f:
            return json.load(f) # dictを返す(configファイルがなければFalseを返す)



# ==========
# 挿したUSBのシリアル番号が登録したものと合致するか確認する
# ==========
def check_usb():
    # configは{"serial": "xxx", "drive": "E:"}のようなdictになる
    config = load_usb()
    # CONFIGファイルが空ならFalse
    if not config:
        return False
    # dictからシリアル番号を取り出す
    saved_serial = config.get("serial")
    usb_list = get_usb_list()
    return saved_serial in [serial for _, serial in usb_list.values()]



# ==========
# USBを登録していない場合は、選択画面を表示する
# ==========
def select_usb():
    usb_list = get_usb_list()
    if not usb_list:
        messagebox.showerror("エラー", "USBが見つかりません")
        sys.exit()
    win = ct.CTk()
    win.title("USB選択")
    ct.CTkLabel(win, text="起動に使うUSBを選択してください").pack(pady=10)

    #ボタンを押すとメッセージボックスを表示
    def on_select(serial, drive):
        save_usb(serial, drive)
        messagebox.showinfo("登録完了", f"USB({serial})が登録されました。もう一度アプリを起動させてください。")
        win.destroy()
    
    #Removable - BUFFALO USB - 0F7315812030 のようなボタンが表示される
    for drive, (caption, serial) in usb_list.items():
        text = f"{drive} - {caption} - {serial}"
        ct.CTkButton(win, text=text, command=lambda s=serial, d=drive: on_select(s, d)).pack(pady=5)
    
    win.mainloop()



# ==========
# DB初期化
# ==========
def init_db(db_path):
    conn = sqlite3.connect(db_path)
    c = conn.cursor()
    c.execute("""CREATE TABLE IF NOT EXISTS accounts (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        service TEXT,
        username TEXT,
        password TEXT,
        created_at TEXT,
        sort_order INTEGER,
        category TEXT
    )""")
    conn.commit()
    return conn


# ==========
# GUI
# ==========
def run_app(conn):
    global service_entry, user_entry, pass_entry, category_menu, tabview, list_frames, root, categories

    root = ct.CTk()
    root.title("パスワード管理アプリ")
    root.geometry("800x600")

    # 横並び用フレーム
    # pack()はDIVタグのようなものであり、この中に横並びさせたい要素をいれる
    input_frame = ct.CTkFrame(root)
    input_frame.pack(pady=10)

    # 入力欄とボタンを横並びに配置
    service_entry = ct.CTkEntry(input_frame, placeholder_text="サービス名", width=150)
    service_entry.grid(row=0, column=0, padx=5, pady=5)

    user_entry = ct.CTkEntry(input_frame, placeholder_text="ユーザー名", width=150)
    user_entry.grid(row=0, column=1, padx=5, pady=5)

    pass_entry = ct.CTkEntry(input_frame, placeholder_text="パスワード", width=150)
    pass_entry.grid(row=0, column=2, padx=5, pady=5)

    # カテゴリ選択
    categories = ["タブ1", "タブ2", "タブ3"]
    category_menu = ct.CTkOptionMenu(input_frame, values=categories)
    category_menu.set(categories[0])
    category_menu.grid(row=0, column=3, padx=5, pady=5)

    add_btn = ct.CTkButton(input_frame, text="追加", command=lambda: add_account(conn))
    add_btn.grid(row=0, column=4, padx=5, pady=5)
    
    # タブビュー
    tabview = ct.CTkTabview(root, width=800, height=500)
    tabview.pack(pady=10, fill="both", expand=True)

    # 一覧表示(Treeviewだとセル単位で編集できない、デザインがダサいので使わない)
    list_frames = {}
    for cat in categories:
        tab = tabview.add(cat)

        # 
        tab_top_frame = ct.CTkFrame(tab)
        tab_top_frame.pack(fill="x", pady=5)

        title_label = ct.CTkLabel(tab_top_frame, text=cat, font=ct.CTkFont(size=16, weight="bold"))
        title_label.pack(side="left", padx=5)

        def make_rename_func(old_name, label):
            def rename_tab_local():
                new_name = ct.CTkInputDialog(text="新しいタブ名を入力").get_input()
                if new_name:
                    #tabview.tab(old_name, text=new_name)
                    label.configure(text=new_name)
                    idx = categories.index(old_name)
                    categories[idx] = new_name
                    category_menu.configure(values=categories)
            return rename_tab_local
        
        rename_btn = ct.CTkButton(tab_top_frame, text="変更", width=60, command=make_rename_func(cat, title_label))
        rename_btn.pack(side="left", padx=5)


        # スクロールフレーム
        list_frames[cat] = ct.CTkScrollableFrame(tab, width=750, height=400)
        list_frames[cat].pack(fill="both", expand=True)


    # USB変更ボタン
    ct.CTkButton(root, text="USBを更新する", command=change_usb).pack(pady=10)

    # 最初の表示
    show_account(conn)

    root.mainloop()


# ==========
# アカウントをデータベースに追加
# ==========
def add_account(conn):
    service = service_entry.get()
    user = user_entry.get()
    password = pass_entry.get()
    category = category_menu.get()
    now = datetime.now().strftime("%Y/%m/%d %H:%M:%S")

    if not service or not user or not password:
        messagebox.showerror("エラー", "すべて入力してください")
        return
    
    # sort_orderの最大値を探して+1する
    c = conn.cursor()
    c.execute("SELECT MAX(sort_order) FROM accounts WHERE category=?", (category,))
    max_order = c.fetchone()[0]
    if max_order is None:
        max_order = 0
    sort_order = max_order + 1

    c.execute("INSERT INTO accounts (service, username, password, created_at, sort_order, category) VALUES (?, ?, ?, ?, ?, ?)",
              (service, user, password, now, sort_order, category))
    conn.commit()
    messagebox.showinfo("保存", f"{category}にデータを保存しました")
    show_account(conn)



# ==========
# アカウントの表示(カテゴリごと)
# ==========
def show_account(conn):

    # 既存のフレームを削除
    for frame in list_frames.values():
        for widget in frame.winfo_children():
            widget.destroy()
    
    c = conn.cursor()
    c.execute("SELECT id, service, username, password, created_at, sort_order, category FROM accounts ORDER BY sort_order ASC")
    rows = c.fetchall()

    for row in rows:
        account_id, service, username, password, created_at, sort_order, category = row
        frame = list_frames.get(category)
        if not frame:
            continue
        idx = len(frame.winfo_children()) // 8 # 1行の要素数に応じて計算

        service_entry = ct.CTkEntry(frame, width=150)
        service_entry.insert(0, service)
        service_entry.grid(row=idx, column=0, padx=5, pady=2)

        user_entry = ct.CTkEntry(frame, width=150)
        user_entry.insert(0, username)
        user_entry.grid(row=idx, column=1, padx=5, pady=2)

        pass_entry = ct.CTkEntry(frame, width=50)
        pass_entry.insert(0, password)
        pass_entry.grid(row=idx, column=2, padx=5, pady=2)

        created_entry = ct.CTkEntry(frame, width=160)
        created_entry.insert(0, created_at or "")
        created_entry.grid(row=idx, column=3, padx=5, pady=2)

        # 更新ボタン
        update_btn = ct.CTkButton(
            frame, text = "更新",
            command=lambda i=account_id, s=service_entry, u=user_entry, p=pass_entry: update_account(conn, i, s, u, p),
            width=60
        )
        update_btn.grid(row=idx, column=4, padx=5)

        # 削除ボタン
        delete_btn = ct.CTkButton(
            frame, text = "削除",
            command=lambda i=account_id: delete_account(conn, i),
            width=60
        )
        delete_btn.grid(row=idx, column=5, padx=5)

        # 行入れ替えボタン
        upper_btn = ct.CTkButton(frame, text="↑", command=lambda i=account_id: move_account(conn, i, -1), width=30)
        upper_btn.grid(row=idx, column=6, padx=2)
        lower_btn = ct.CTkButton(frame, text="↓", command=lambda i=account_id: move_account(conn, i, +1), width=30)
        lower_btn.grid(row=idx, column=7, padx=2)



# ==========
# アカウント更新
# ==========
def update_account(conn, account_id, service_entry, user_entry, pass_entry):
    service = service_entry.get()
    user = user_entry.get()
    password = pass_entry.get()
    now = datetime.now().strftime("%Y/%m/%d %H:%M:%S")

    c = conn.cursor()
    c.execute("UPDATE accounts SET service=?, username=?, password=?, created_at=? WHERE id=?",
              (service, user, password, now, account_id))
    conn.commit()
    messagebox.showinfo("更新", "データを更新しました")

    # 表示更新する
    show_account(conn)



# ==========
# アカウント削除
# ==========
def delete_account(conn, account_id):
    if messagebox.askyesno("確認", "本当に削除しますか?"):
        c = conn.cursor()
        c.execute("DELETE FROM accounts WHERE id=?", (account_id,))
        conn.commit()
        show_account(conn)


# ==========
# アカウントの行入れ替え
# ==========
def move_account(conn, account_id, direction):
    # 現在のsort_orderの数値を取得
    c = conn.cursor()
    c.execute("SELECT sort_order FROM accounts WHERE id=?", (account_id,))
    current_order = c.fetchone()[0]

    # 隣接している行を取得
    if direction == -1:
        c.execute("SELECT id, sort_order FROM accounts WHERE sort_order < ? ORDER BY sort_order DESC LIMIT 1", (current_order,))
    else:
        c.execute("SELECT id, sort_order FROM accounts WHERE sort_order > ? ORDER BY sort_order ASC LIMIT 1", (current_order,))
    neighbor = c.fetchone()
    
    # 隣接行と入れ替える
    if neighbor:
        neighbor_id, neighbor_order = neighbor
        c.execute("UPDATE accounts SET sort_order=? WHERE id=?", (neighbor_order, account_id))
        c.execute("UPDATE accounts SET sort_order=? WHERE id=?", (current_order, neighbor_id))
        conn.commit()
        show_account(conn)


# ==========
# タブ名変更
# ==========




# ==========
# USBの変更
# ==========
def change_usb():
    select_usb()
    messagebox.showinfo("USB変更", "USBが変更されました。次回起動から有効です。")




# ==========
# アプリを起動したときの処理
# ==========
# configファイルからdictを取得
config = load_usb()

if not config:
    # CONFIGファイルがない初回起動ではUSB選択画面を表示する
    select_usb()
    # 一度システム終了する
    sys.exit()

# 登録USBを挿入していなければエラーを表示する
if not check_usb():
    messagebox.showerror("エラー", "認証USBが見つかりません。アプリを終了します。")
    sys.exit()

# CONFIGファイルがある場合は、dictからドライブレターを取得
drive = config.get("drive")

# 念の為、PATHが存在するか確認
#usb_path = drive + os.sep
#if not os.path.exists(usb_path):
#    messagebox.showerror("エラー", f"USBドライブ{drive}が見つかりません")
#    sys.exit()

# DBファイルの絶対パスを作成(os.sepは"/"と同じ意味)
db_path = os.path.join(drive + os.sep, DB_NAME)

# USB内のDBを初期化する
conn = init_db(db_path)

#画面表示
ct.set_appearance_mode("dark")
ct.set_default_color_theme("blue")
run_app(conn)

BACK