app.py

"""A small desktop app for recording experiments."""
from __future__ import annotations
from email import header
from tkinter import messagebox, ttk

from tracker import Experiment, ExperimentStore, STATUSES, matches_search, today


import tkinter as tk
from pathlib import Path
from tkinter import messagebox, ttk


DATA_FILE = Path(__file__).with_name("experiments.json")


class ExperimentTrackerApp(tk.Tk):
    def __init__(self) -> None:
        super().__init__()
        style = ttk.Style(self)
        style.theme_use("clam")
        style.configure(".", background="#1e1e1e", foreground="#ffffff")
        style.configure("TFrame", background="#1e1e1e")
        style.configure("TLabel", background="#1e1e1e", foreground="#ffffff")
        style.configure("TButton", background="#2d2d2d", foreground="#ffffff")
        style.configure("TEntry", fieldbackground="#2d2d2d", foreground="#ffffff")
        style.configure("TCombobox", fieldbackground="#2d2d2d", foreground="#ffffff")
        self.title("Experiment Tracker")
        self.geometry("1050x700")
        self.minsize(900, 600)
        self.store = ExperimentStore(DATA_FILE)
        try:
            self.experiments = self.store.load()
        except ValueError as error:
            messagebox.showwarning("Data issue", f"{error}\nStarting with an empty list.")
            self.experiments = []
        self.selected_id: str | None = None
        self.search_var = tk.StringVar()
        self.fields: dict[str, tk.Widget] = {}
        self._build_interface()
        self.refresh_list()

    def _build_interface(self) -> None:
        self.columnconfigure(0, weight=1)
        style = ttk.Style()
        style.theme_use("clam")
        style.configure("Dark.TEntry", fieldbackground="#222222", foreground="white")
        style.map(
    "Dark.TCombobox",
    fieldbackground=[("readonly", "#222222")],
    foreground=[("readonly", "white")]
)
        self.rowconfigure(1, weight=1)

        header = ttk.Frame(self, padding=(16, 14, 16, 8))
        header.grid(row=0, column=0, sticky="ew")
        header.columnconfigure(1, weight=1)
        ttk.Label(header, text="Experiment Tracker", font=("Segoe UI", 18, "bold")).grid(row=0, column=0, sticky="w")
        search_area = ttk.Frame(header)
        search_area.grid(row=0, column=1, sticky="ew", padx=(30, 6))
        search_area.columnconfigure(1, weight=1)
        ttk.Label(search_area, text="Search:").grid(row=0, column=0, padx=(0, 6))
        ttk.Entry(search_area, textvariable=self.search_var).grid(row=0, column=1, sticky="ew")
        ttk.Button(header, text="Clear search", command=self.clear_search).grid(row=0, column=2)
        self.search_var.trace_add("write", lambda *_: self.refresh_list())
        self.stats_var = tk.StringVar(value="Total: 0    Planned: 0    Running: 0    Completed: 0")
        ttk.Label(
        header,
        textvariable=self.stats_var,
    font=("Segoe UI", 10, "bold")
).grid(row=1, column=0, columnspan=3, sticky="w", pady=(8, 0))

        content = ttk.PanedWindow(self, orient=tk.HORIZONTAL)
        content.grid(row=1, column=0, sticky="nsew", padx=16, pady=(0, 16))

        left = ttk.Frame(content, padding=8)
        content.add(left, weight=1)
        ttk.Label(left, text="Your experiments", font=("Segoe UI", 12, "bold")).pack(anchor="w", pady=(0, 6))
        style.configure(
        "Dark.Treeview",
        background="#222222",
        foreground="white",
        fieldbackground="#222222",
        rowheight=28
)

        style.map(
        "Dark.Treeview",
        background=[("selected", "#444444")],
        foreground=[("selected", "white")]
)         
        self.tree = ttk.Treeview(
    left,
    columns=("status", "start", "end"),
    show="tree headings",
    selectmode="browse",
    style="Dark.Treeview"
)
        self.tree.heading("#0", text="Experiment")
        self.tree.heading("status", text="Status")
        self.tree.heading("start", text="Started")
        self.tree.heading("end", text="Ended")
        self.tree.column("#0", width=220)
        self.tree.column("status", width=90, anchor="center")
        self.tree.column("start", width=90, anchor="center")

        self.tree.pack(fill="both", expand=True)
        self.tree.bind("<<TreeviewSelect>>", self.load_selected)
        self.tree.tag_configure("planned", foreground="#B8860B")
        self.tree.tag_configure("running", foreground="#1E6BB8")
        self.tree.tag_configure("completed", foreground="#228B22")
        self.tree.tag_configure("cancelled", foreground="#C62828")
        self.tree.column("end", width=90, anchor="center")
        right = ttk.Frame(content, padding=8)
        content.add(right, weight=2)
        ttk.Label(right, text="Experiment details", font=("Segoe UI", 12, "bold")).grid(row=0, column=0, columnspan=2, sticky="w", pady=(0, 10))
        right.columnconfigure(1, weight=1)
        self._add_entry(right, "Name", "name", 1)
        self._add_text(right, "Hypothesis", "hypothesis", 2)
        self._add_entry(right, "Independent variable", "independent_variable", 3)
        self._add_entry(right, "Dependent variable", "dependent_variable", 4)
        self._add_text(right, "Controls / constants", "controls", 5)
        self._add_text(right, "Results / observations", "results", 6)
        self._add_combo(right, "Status", "status", 7)
        self._add_entry(right, "Start date (YYYY-MM-DD)", "start_date", 8)
        self._add_entry(right, "End date (YYYY-MM-DD)", "end_date", 9)

        buttons = ttk.Frame(right)
        buttons.grid(row=10, column=0, columnspan=2, sticky="e", pady=(14, 0))
        ttk.Button(buttons, text="New", command=self.new_experiment).pack(side="left", padx=4)
        ttk.Button(buttons, text="Save", command=self.save_experiment).pack(side="left", padx=4)
        ttk.Button(buttons, text="Delete", command=self.delete_experiment).pack(side="left", padx=4)
        self.new_experiment()

    def _add_entry(self, parent: ttk.Frame, label: str, key: str, row: int) -> None:
        ttk.Label(parent, text=label).grid(row=row, column=0, sticky="nw", padx=(0, 10), pady=4)
        widget = ttk.Entry(parent, style="Dark.TEntry")
        widget.grid(row=row, column=1, sticky="ew", pady=4)
        self.fields[key] = widget

    def _add_text(self, parent: ttk.Frame, label: str, key: str, row: int) -> None:
        ttk.Label(parent, text=label).grid(row=row, column=0, sticky="nw", padx=(0, 10), pady=4)
        widget = tk.Text(parent, height=3, wrap="word", font=("Segoe UI", 10), bg="#2d2d2d", fg="#ffffff", insertbackground="#ffffff")
        widget.grid(row=row, column=1, sticky="ew", pady=4)
        self.fields[key] = widget

    def _add_combo(self, parent: ttk.Frame, label: str, key: str, row: int) -> None:
        ttk.Label(parent, text=label).grid(row=row, column=0, sticky="nw", padx=(0, 10), pady=4)
        widget = ttk.Combobox(parent, values=STATUSES, state="readonly", style="Dark.TCombobox")
        widget.grid(row=row, column=1, sticky="ew", pady=4)
        self.fields[key] = widget

    def value(self, key: str) -> str:
        widget = self.fields[key]
        return widget.get("1.0", "end-1c").strip() if isinstance(widget, tk.Text) else widget.get().strip()

    def set_value(self, key: str, value: str) -> None:
        widget = self.fields[key]
        if isinstance(widget, tk.Text):
            widget.delete("1.0", "end")
            widget.insert("1.0", value)
        elif isinstance(widget, ttk.Combobox):
            widget.set(value)
        else:
            widget.delete(0, "end")
            widget.insert(0, value)

    def new_experiment(self) -> None:
        self.selected_id = None
        for key in self.fields:
            self.set_value(key, "")
        self.set_value("status", "Planned")
        self.set_value("start_date", today())
        self.tree.selection_remove(self.tree.selection())

    def current_experiment(self) -> Experiment:
        values = {key: self.value(key) for key in self.fields}
        return Experiment.new(**values) if self.selected_id is None else Experiment(id=self.selected_id, **values)

    def save_experiment(self) -> None:
        if self.value("status") == "Completed" and not self.value("end_date"):
         self.set_value("end_date", today())
        self.selected_id = self.selected_id or self.current_experiment().id
        experiment = self.current_experiment()
        if not experiment.name:
            messagebox.showinfo("Name needed", "Give the experiment a name before saving it.")
            return
        old = next((i for i, item in enumerate(self.experiments) if item.id == experiment.id), None)
        if old is None:
            self.experiments.append(experiment)
        else:
            self.experiments[old] = experiment
        self.store.save(self.experiments)
        self.selected_id = experiment.id
        self.refresh_list(select_id=experiment.id)
        messagebox.showinfo("Saved", "Your experiment was saved on this computer.")

    def delete_experiment(self) -> None:
        if self.selected_id is None:
            messagebox.showinfo("Nothing selected", "Choose an experiment to delete.")
            return
        if not messagebox.askyesno("Delete experiment", "Delete this experiment permanently?"):
            return
        self.experiments = [item for item in self.experiments if item.id != self.selected_id]
        self.store.save(self.experiments)
        self.new_experiment()
        self.refresh_list()
    def update_stats(self) -> None:
        total = len(self.experiments)
        planned = sum(item.status == "Planned" for item in self.experiments)
        running = sum(item.status == "Running" for item in self.experiments)
        completed = sum(item.status == "Completed" for item in self.experiments)

        self.stats_var.set(
            f"Total: {total}    Planned: {planned}    "
            f"Running: {running}    Completed: {completed}"
        )
    def refresh_list(self, select_id: str | None = None) -> None:
        self.update_stats()
        self.tree.delete(*self.tree.get_children())
        query = self.search_var.get()
        for experiment in self.experiments:
            if matches_search(experiment, query):
                self.tree.insert(
    "",
    "end",
    iid=experiment.id,
    text=experiment.name,
    values=(experiment.status, experiment.start_date ,experiment.end_date),
    tags=(experiment.status.lower(),),
)
        if select_id and self.tree.exists(select_id):
            self.tree.selection_set(select_id)
            self.tree.focus(select_id)

            

    def clear_search(self) -> None:
        self.search_var.set("")

    def load_selected(self, _event: object = None) -> None:
        selected = self.tree.selection()
        if not selected:
            return
        self.selected_id = selected[0]
        experiment = next(item for item in self.experiments if item.id == self.selected_id)
        for key in self.fields:
            self.set_value(key, getattr(experiment, key))


if __name__ == "__main__":
    ExperimentTrackerApp().mainloop()