import csv
import datetime
import os
import tkinter as tk
from tkinter import filedialog, messagebox, ttk


class App062:
    """習慣トラッカー"""

    def __init__(self, root):
        self.root = root
        self.root.title("習慣トラッカー")
        self.root.geometry("780x520")
        self.root.configure(bg="#f8f9fc")
        # 過去7日間 (今日を含む)
        today = datetime.date.today()
        self.days = [today - datetime.timedelta(days=6 - i) for i in range(7)]
        # habit name -> {date: bool}
        self.habits = {}
        self.vars = {}  # (habit, date) -> IntVar
        self._build_ui()

    def _build_ui(self):
        """UI を構築する"""
        title = tk.Frame(self.root, bg="#3776ab", pady=12)
        title.pack(fill=tk.X)
        tk.Label(title, text="習慣トラッカー", font=("Noto Sans JP", 16, "bold"),
                 bg="#3776ab", fg="white").pack()

        ctrl = ttk.LabelFrame(self.root, text="操作", padding=10)
        ctrl.pack(fill=tk.X, padx=15, pady=10)
        ttk.Label(ctrl, text="習慣:").grid(row=0, column=0)
        self.entry = ttk.Entry(ctrl, width=22)
        self.entry.grid(row=0, column=1, padx=6)
        self.entry.bind("<Return>", lambda e: self.add_habit())
        ttk.Button(ctrl, text="追加", command=self.add_habit).grid(row=0, column=2, padx=4)
        ttk.Button(ctrl, text="選択削除", command=self.remove_habit).grid(row=0, column=3, padx=4)
        ttk.Button(ctrl, text="CSV保存", command=self.save_csv).grid(row=0, column=4, padx=4)
        ttk.Button(ctrl, text="CSV読込", command=self.load_csv).grid(row=0, column=5, padx=4)

        self.grid_frame = ttk.LabelFrame(self.root, text="チェックリスト（過去7日間）", padding=10)
        self.grid_frame.pack(fill=tk.BOTH, expand=True, padx=15, pady=5)

        self.summary = tk.Label(self.root, text="", bg="#f8f9fc",
                                font=("Noto Sans JP", 11, "bold"), fg="#3776ab")
        self.summary.pack(pady=8)

        self._render_grid()

    def _render_grid(self):
        """チェックリストグリッドを再描画する"""
        for w in self.grid_frame.winfo_children():
            w.destroy()
        # ヘッダ
        ttk.Label(self.grid_frame, text="習慣", width=20, anchor="w",
                  font=("Noto Sans JP", 10, "bold")).grid(row=0, column=0, padx=4, pady=4)
        for i, d in enumerate(self.days):
            ttk.Label(self.grid_frame, text=d.strftime("%m/%d\n(%a)"),
                      font=("Noto Sans JP", 9, "bold")).grid(row=0, column=i + 1, padx=2)
        ttk.Label(self.grid_frame, text="達成率", font=("Noto Sans JP", 10, "bold")).grid(
            row=0, column=8, padx=6)

        self.selected = tk.StringVar(value="")
        for r, habit in enumerate(self.habits, start=1):
            ttk.Radiobutton(self.grid_frame, text=habit, value=habit,
                            variable=self.selected).grid(row=r, column=0, sticky="w", padx=4)
            for c, d in enumerate(self.days):
                v = tk.IntVar(value=1 if self.habits[habit].get(d.isoformat()) else 0)
                self.vars[(habit, d.isoformat())] = v
                ttk.Checkbutton(self.grid_frame, variable=v,
                                command=lambda h=habit, dt=d: self._on_check(h, dt)).grid(
                    row=r, column=c + 1)
            ttk.Label(self.grid_frame, text=self._rate(habit)).grid(row=r, column=8)

        self._update_summary()

    def _on_check(self, habit, date):
        """チェック状態を保存しサマリを更新する"""
        v = self.vars[(habit, date.isoformat())].get()
        self.habits[habit][date.isoformat()] = bool(v)
        self._render_grid()

    def _rate(self, habit) -> str:
        """指定習慣の週間達成率を文字列で返す"""
        done = sum(1 for d in self.days if self.habits[habit].get(d.isoformat()))
        return f"{done}/7 ({int(done / 7 * 100)}%)"

    def _update_summary(self):
        """全体達成率をサマリ欄に出す"""
        if not self.habits:
            self.summary.config(text="習慣を追加してください")
            return
        total = len(self.habits) * 7
        done = sum(1 for h in self.habits for d in self.days
                   if self.habits[h].get(d.isoformat()))
        self.summary.config(text=f"全体達成率: {done}/{total} ({int(done / total * 100)}%)")

    def add_habit(self):
        """習慣を追加する"""
        name = self.entry.get().strip()
        if not name:
            messagebox.showwarning("警告", "習慣名を入力してください")
            return
        if name in self.habits:
            messagebox.showinfo("情報", "既に登録されています")
            return
        self.habits[name] = {}
        self.entry.delete(0, tk.END)
        self._render_grid()

    def remove_habit(self):
        """選択中の習慣を削除する"""
        target = getattr(self, "selected", tk.StringVar()).get()
        if not target:
            messagebox.showinfo("情報", "削除する習慣を選択してください")
            return
        if messagebox.askyesno("確認", f"「{target}」を削除しますか？"):
            self.habits.pop(target, None)
            self._render_grid()

    def save_csv(self):
        """習慣記録をCSVに保存する"""
        path = filedialog.asksaveasfilename(defaultextension=".csv",
                                            filetypes=[("CSV", "*.csv")])
        if not path:
            return
        try:
            with open(path, "w", newline="", encoding="utf-8") as f:
                writer = csv.writer(f)
                writer.writerow(["habit"] + [d.isoformat() for d in self.days])
                for h in self.habits:
                    row = [h] + [int(bool(self.habits[h].get(d.isoformat()))) for d in self.days]
                    writer.writerow(row)
            messagebox.showinfo("保存完了", os.path.basename(path))
        except OSError as e:
            messagebox.showerror("エラー", str(e))

    def load_csv(self):
        """CSV から記録を読み込む"""
        path = filedialog.askopenfilename(filetypes=[("CSV", "*.csv")])
        if not path:
            return
        try:
            with open(path, "r", encoding="utf-8") as f:
                reader = csv.reader(f)
                header = next(reader)
                date_cols = header[1:]
                self.habits = {}
                for row in reader:
                    if not row:
                        continue
                    self.habits[row[0]] = {date_cols[i]: bool(int(v))
                                           for i, v in enumerate(row[1:]) if v}
            self._render_grid()
        except (OSError, ValueError, StopIteration) as e:
            messagebox.showerror("エラー", f"読込失敗: {e}")


if __name__ == "__main__":
    root = tk.Tk()
    app = App062(root)
    root.mainloop()