Python tkinterでボタンを押したときにラジオボタンで選択されている内容を取得したい

Question

前提・実現したいこと

プログラムの条件分岐の中で、
変数hogeが空白のときに tkinter のラジオボタンを用いてユーザー入力をする
という処理をさせたいと考えています。

該当のソースコード

#!python3.7
# -*- coding: utf-8 -*-
import tkinter as tk

def hoge_input():
    root = tk.Tk()
    root.option_add('*font', ('FixedSys', 14))
    root.title('えらぶ')
    selection = ['選択肢A','選択肢B','選択肢C']
    rdo_var = tk.IntVar()

    for i in range(len(selection)):
        tk.Radiobutton(root, value=i, variable=rdo_var, text=selection[i]).pack(anchor = tk.CENTER,padx=30,pady=3)
    
    def btn_click():
        num = rdo_var.get()
        hoge = selection[num]
        root.destroy()
        print(hoge)
        return hoge

    btn = tk.Button(root, text='OK', command=btn_click)
    btn.pack(fill ='x',padx=40,pady=3)
    root.mainloop()


if __name__ == "__main__":
    # hogeを格納する関数hoge_search()→hogeに該当する値が無かった場合にユーザー選択
    # hoge_search()

    if hoge == '':
        hoge_input()

    # ここでユーザーが選択した選択肢A/B/Cを表示させたい
    print(hoge)
    # None

わかっていること

btn_clickのイベントのタイミングで変数を格納したいのですが、
returnしてもtk.Button(root, text='OK', command=btn_click)ここに値が戻ってしまう形になってしまい
main関数のhogeへの値の渡し方がわかりません。

    def btn_click():
        num = rdo_var.get()
        hoge = selection[num]
        root.destroy()
        print(hoge)　　#この時点ではhogeが格納されている
        return hoge　　#returnされない

Answer 1

基本的には shiracamusさんが回答されているように hoge_input関数の戻り値として選択されている値を渡すと良いかと思います が、気になる点を２点ほど

(1)
最終的に得たい値が "選択肢のIndex" ではなく "選択肢の内容" なのでしたら、tk.IntVar() ではなくて tk.StringVar() を使うほうが適当ではないでしょうか。
こうすることで Radiobutton生成部のループも

    for s in selection:
        tk.Radiobutton(root, value=s, variable=rdo_var, text=s).pack(anchor = tk.CENTER,padx=30,pady=3)

と少しスッキリ書けます

(2)
ラジオボックスが選択されている状態でウインドウ上部の "閉じる(X)ボタン"を押されたときの挙動ですが、

• 選択されていないと見なす
• 現在選択されている値を返す

によって記述が異なります。後者の場合はshiracamusさんが書かれているようにhoge_input関数の最後に

def hoge_input():
    # （略）
    return rdo_var.get()

とすると良いのですが、前者の場合は質問のコードのようにbtn_click関数を作成してその中で一度**"hoge_input関数のローカ ル変数"**に格納した後に、その変数を戻り値として返すようにするかと良いと思います。

以上を修正すると以下のようになります。

import tkinter as tk

def hoge_input():
    root = tk.Tk()
    root.option_add('*font', ('FixedSys', 14))
    root.title('えらぶ')
    selection = ['選択肢A','選択肢B','選択肢C']
    result = 'NotSelected' # 選択されていないときに返す値を入れておく
    rdo_var = tk.StringVar(value=result)

    for s in selection:
        rb = tk.Radiobutton(root, value=s, variable=rdo_var, text=s)
        rb.pack(anchor = tk.CENTER,padx=30,pady=3)

    def btn_click():
        nonlocal result # 親関数の変数に書き込むのでこれが必要
        result = rdo_var.get()
        root.destroy()

    btn = tk.Button(root, text='OK', command=btn_click)
    btn.pack(fill ='x',padx=40,pady=3)
    root.mainloop()
    return result # 最終的にこの変数に書き込まれている内容を返す


if __name__ == "__main__":
    hoge = '' # とりあえずコードを動作させるために追記

    if hoge == '':
        hoge = hoge_input()

    # ここでユーザーが選択した選択肢A/B/Cを表示させたい
    print(hoge)

Comments

[shiracamus]

選択されていない値には、Noneを使うといいですね。
ただ、値が無いときにこの処理を呼び出しているので、再びこの処理を呼び出すことになりそうな予感。

[magichan]

それは少し考えたのですが、元々 hoge には'空文字'が入っていますので、それを None に戻すのは逆に？？？な気がします。（受け側に処理を追加するとよいだけなのですが）
まあこの後の処理次第なのですが、Str型を前提としている処理が書かれているかもしれないので空文字のままのほうが適当かもしれないとも思いましたが、とりあえずサンプルとして違う文字列に変えております。
この辺は明確な仕様が提示されている訳ではないので、質問者さんが決定して必要に応じて変更するとよいのではないでしょうか。

[shiracamus]

'NotSelected'やNoneや空文字が返ってきたら、結局再びこの関数を呼び出すことになるのではないかと思った次第です。
それならば、選択項目の先頭をデフォルト値にしておけば、X でウィンドウを閉じてもデフォルト値が帰るので問題ないかと考えました。IntVar はデフォルトで 0 になりますので。
Xでウィンドウを閉じたときにどうしたいのか、質問者に確認しない限り、どうするのが最善かわからないですね。

[per_]

> (1)tk.IntVar() ではなくて tk.StringVar() を使うほうが適当
まさしくこの通りで、この後の処理では文字列そのものを処理するため、たいへん助かりました。

> (2)"閉じる(X)ボタン"を押されたときの挙動
ここの部分を質問に記載しておらず恐縮ですが、今回は”選択されていないと見なす”こととしたかったため、 magichanさんのこの回答をベストアンサーとさせていただきます。

`nonlocal result` 目から鱗でした。もっと勉強します。
ご指摘いただいた箇所を書き直したところ、要求していた動作が実現できました。たいへんありがとうございました。

Answer 2

これでいかがですか？
なお、Python3では coding:utf-8 指定不要です。

#!python3.7
import tkinter as tk


def hoge_input():
    root = tk.Tk()
    root.option_add('*font', ('FixedSys', 14))
    root.title('えらぶ')
    selections = '選択肢A', '選択肢B', '選択肢C'
    rdo_var = tk.IntVar()

    for i, selection in enumerate(selections):
        tk.Radiobutton(root, value=i, variable=rdo_var, text=selection).pack(anchor=tk.CENTER, padx=30, pady=3)

    btn = tk.Button(root, text='OK', command=root.destroy)
    btn.pack(fill='x', padx=40, pady=3)
    root.mainloop()
    return selections[rdo_var.get()]


if __name__ == "__main__":
    # hogeを格納する関数hoge_search()→hogeに該当する値が無かった場合にユーザー選
択
    hoge = ''
    # hoge_search()

    if hoge == '':
        hoge = hoge_input()

    # ここでユーザーが選択した選択肢A/B/Cを表示させたい
    print(hoge)

Comments

[per_]

shiracamusさんご回答ありがとうございました！ばっちり動作なうえ、とてもすっきりとしたコードを記載いただきました。
特に `for i, selection in enumerate(selections):` ここと同じようなことを別の箇所で（ぐちゃぐちゃなコードで）書いていたためとても改善できました。
×ボタン挙動のことがあり別の回答をベストアンサーとさせていただきましたが、とても勉強になりました。

> なお、Python3では coding:utf-8 指定不要です。
Python2系での記載事項だったことを知りました…ありがとうございます。