Python3 Tkinter Treeviewでソートしたい

Question

CSVのデータをTreeviewで表示し、
ヘッダ列をクリックしたら昇順、もう一度クリックしたら降順、もう一度クリックしたら昇順、･･･、としたいです。
（一応この段階ではもとの順番に戻さなくていいです。）

クリックしたヘッダが何列目かまでは取得できましたが、クリックした列の各行のデータの取得と各行の入れ替え方法がわかりません。

一応、コメントアウトしてあるやつを有効化したときと同じ動作を期待しているのですが、最初のforの前にあるやつがよくわかりませんでした。文法的な意味で です。

以下ソースコードです。すべて同じ階層に配置。

main.py

from tkinter import ttk
from tkinter.font import Font
import tkinter as tk

import read


def headerSelected(tree, reverse_flag):
    """ Memo
        winfo_pointerx()    マウスカーソルのx座標
        winfo_rootx()       ディスプレイ上でのx座標
        identify_column(x)  x座標のセルの識別子（文字列 "#n"。 nは1以上の整数）を返す
    """
    # クリックされたヘッダ列取得（http://notwodaily.hatenablog.com/entry/2018/06/15/003411）
    x = tree.winfo_pointerx() - tree.winfo_rootx()
    select_column_str = tree.identify_column(x)
    select_column_int = int(select_column_str[1:]) - 1
    print(select_column_int)    # 1列目クリックで0, 2列目クリックで1、･･･、n列目クリックでn-1

    # クリックした列の各行のデータ取得
    print(tree.get_children())  # ('I001', 'I002', 'I003', 'I004', 'I005', 'I006')

    
    """ ソート（https://stackoverrun.com/ja/q/396512）

    l = [(tree.set(k, select_column_int), k) for k in tree.get_children("")]
    l.sort(reverse=reverse_flag)
    
    for index, (val, k) in enumerate(l):
        tree.move(k, "", index)

    tree.heading(select_column_int, command=lambda:headerSelected(tree, not reverse_flag))
    """


win = tk.Tk()

win.rowconfigure(0, weight=1)
win.columnconfigure(0, weight=1)

s = ttk.Style()
s.theme_use("clam")
fontsize = 30
myfont = Font(size=fontsize)
s.configure("Treeview.Heading", font=("", fontsize))
s.configure("Treeview", font=("", fontsize),
            rowheight=myfont.metrics()["linespace"])
s.configure("Vertical.TScrollbar", arrowsize=30)


datas = read.read()

columns = 0
for data in datas:
    if columns < len(data):
        columns = len(data)

tree = ttk.Treeview(win)
tree.grid(row=0, column=0, sticky="nsew", padx=(5, 0), pady=(5, 5))
tree["columns"] = tuple(range(columns))
tree["show"] = "headings"
for i in range(columns):
    tree.heading(i, text="Data"+str(i+1), command=lambda:headerSelected(tree, False))
for data in datas:
    tree.insert("", "end", value=(data))

ysb = ttk.Scrollbar(win, orient="vertical", command=tree.yview)
tree.configure(yscrollcommand=ysb.set)
ysb.grid(row=0, column=1, sticky="nsew", padx=(0, 5), pady=5)


win.mainloop()

read.py

import csv
import os
import sys

def read():
    data = []
    SRC = os.path.dirname(sys.argv[0]) + "/" + "table.csv"
    with open(file=SRC, mode="r", encoding="utf-8") as rf:
        f = csv.reader(rf, lineterminator="\r\n")
        for i in f:
            data.append(i)
    return data

table.csv

Aa0,Bb1,,,
Cc2,Dd3,
,Ee4
Ff5,,,Gg6
Hh7,Ii8,Jj9,Ll10,Mm11
,,Nn12,,Oo13

Answer 1

機能自体は正常に動いてるように見えます。

enumerate が解らないという事でしょうか？
for in のループ変数に関しては、Extended Iterable Unpacking
という機能で、for文に限らず通常の変数の代入でも使えます。
(但し、関数の引数では使えなくなりました。2.xの頃は出来た)

a, (b, c) = 1, ["a", "b"]

print(a, b, c) # => 1 a b

data = {"A": 1, "B": 2, "C": 3}

for key, value in data.items(): # items() => [("A", 1), ("B", 2), ("C", 3)]
    print(key, value)

# => "A", 1
# => "B", 2
# => "C", 3

# ループなしで単体で見た場合
# key, value = ("A", 1)

enumerate により、リストの各要素に index が付いてるため

data = {"A": 1, "B": 2, "C": 3}

for index, (key, value) in enumerate(data.items()):
    print(index, key, value)

# => 0, "A", 1
# => 0, "B", 2
# => 0, "C", 3

---

子要素なしの場合ですが、順序を元に戻すには iid 順序にすると
insert された順序に戻ります。(ひとつ前に戻すは別途、ソート状態を保存しておく必要があります）

def reset():
    for index, iid in enumerate(sorted(tree.get_children(""))):
        tree.move(iid, "", index)
ttk.Button(win, text="reset", command=reset).grid()

---

実行時に躓いた点

実行環境によっては SRC は /table.csv になってしまいます。間に os.path.abspath が必要ですが、
pyinstaller が絡むと少し複雑になるので、参考リンク。pathlib を使った解決策が良さそうです。

https://stackoverflow.com/questions/404744/determining-application-path-in-a-python-exe-generated-by-pyinstaller

Comments

[person]

回答ありがとうございます。
質問文に書いた「forの前にあるやつ」というのは
l = [(tree.set(k, select_column_int), k) for k in tree.get_children("")]
というやつです。

[teamikl]

文法的には、「リスト内包表記」というものです。
for 文に書き直すと以下の様になります（※ コメント内の為、全角インデント）

l = []
for k in tree.get_children(""): # アイテムID (IID) のリスト I001, I002 ～
　value = tree.set(k, select_column_int) # ソート対象のカラムの値
　l.append((value, k))

(ソート対象の値, IID) のリストをソートすることにより、
目的の IID の順序を得ています。

----
tree.set の挙動は引数によって変わります。

> https://docs.python.org/ja/3/library/tkinter.ttk.html#tkinter.ttk.Treeview
set(item, column=None, value=None)
1 引数で呼び出された場合、指定された item のカラムと値のペアからなる辞書を返します。
2 引数で呼び出された場合、指定された column の現在の値を返します。
3 引数で呼び出された場合、与えられた item の column を指定された値 value に設定します。

[person]

ありがとうございます。