PyGTKでツリービューにリストのデータ(ListStore)を表示(データを変更可能にする・コード例)

「PyGTKでツリービューにリストのデータ(ListStore)を表示(データを変更可能にする・メモ)」の続き。そのメモを踏まえた上でのコード例となる。
なお、今回の例では画像を描画するため、そのデータを先に貼り付ける。これは今後貼り付ける別のコードでも使用するため、再利用できるように別ファイルに保存してからこれをimportするという形をとる。import文で使用したいファイルの配置場所については「Pythonのimport文について」を参照(メインのスクリプトと同じディレクトリならOK)。
画像の中身は男女それぞれを示すマーク(「♂」と「♀」)。
ファイル名: sexmark.py

male_icon_xpm = \
[
"16 16 65 1",
" 	c None",
".	c #9999FF",
"+	c #6E6EE2",
"@	c #0E0EA2",
"#	c #4B4BCB",
"$	c #9191F9",
"%	c #7C7CEC",
"&	c #7676E7",
"*	c #1616A8",
"=	c #000099",
"-	c #4343C5",
";	c #8E8EF8",
">	c #2121AF",
",	c #1C1CAC",
"'	c #3535BD",
")	c #8989F4",
"!	c #1919AA",
"~	c #2323B0",
"{	c #8F8FF9",
"]	c #8383F0",
"^	c #1313A6",
"/	c #2A2AB5",
"(	c #9393FB",
"_	c #8282F0",
":	c #5C5CD6",
"<	c #6161DA",
"[	c #0D0DA2",
"}	c #3232BA",
"|	c #9595FD",
"1	c #5858D4",
"2	c #03039B",
"3	c #1111A5",
"4	c #02029B",
"5	c #05059D",
"6	c #3A3AC0",
"7	c #9797FE",
"8	c #8383F1",
"9	c #7A7AEB",
"0	c #02029A",
"a	c #9696FD",
"b	c #5E5ED8",
"c	c #7777E9",
"d	c #4F4FCD",
"e	c #9898FF",
"f	c #1E1EAD",
"g	c #4444C6",
"h	c #5555D2",
"i	c #0C0CA1",
"j	c #9494FC",
"k	c #9797FD",
"l	c #1111A4",
"m	c #4444C7",
"n	c #8686F2",
"o	c #09099F",
"p	c #3838BF",
"q	c #7575E7",
"r	c #7A7AEA",
"s	c #3C3CC1",
"t	c #05059C",
"u	c #7D7DEC",
"v	c #7979E9",
"w	c #2222B0",
"x	c #01019A",
"y	c #2020AE",
"z	c #7373E6",
"................",
"................",
".......+@@@@@@#.",
".......$%%%&*=-.",
"..........;>,'-.",
".........)!~{--.",
"........]^/(.--.",
"..._:<;%[}|..--.",
"..12[34567...88.",
".901a7b=c.......",
".d*e...fg.......",
".hij..klm.......",
".nopqrstu.......",
"..vw0xyz........",
"....aj..........",
"................",
]
female_icon_xpm = \
[
"16 16 52 1",
" 	c None",
".	c #FF9999",
"+	c #FA9191",
"@	c #E87676",
"#	c #E67474",
"$	c #F98F8F",
"%	c #DB6262",
"&	c #9F0A0A",
"*	c #A30E0E",
"=	c #A51313",
"-	c #9D0707",
";	c #D35757",
">	c #F08282",
",	c #9B0404",
"'	c #D96060",
")	c #FE9797",
"!	c #FF9898",
"~	c #DF6A6A",
"{	c #990101",
"]	c #E67373",
"^	c #D55959",
"/	c #AC1C1C",
"(	c #B22626",
"_	c #C84646",
":	c #D65B5B",
"<	c #A71515",
"[	c #FE9898",
"}	c #C94848",
"|	c #F28686",
"1	c #9D0606",
"2	c #CD4E4E",
"3	c #F88F8F",
"4	c #D05353",
"5	c #9A0101",
"6	c #E97878",
"7	c #B32626",
"8	c #AD1E1E",
"9	c #C23D3D",
"0	c #B52A2A",
"a	c #FD9595",
"b	c #E97777",
"c	c #BC3434",
"d	c #B22525",
"e	c #FC9595",
"f	c #F68C8C",
"g	c #B52B2B",
"h	c #A20D0D",
"i	c #F68B8B",
"j	c #C64343",
"k	c #B92F2F",
"l	c #DD6666",
"m	c #D55B5B",
"................",
"......+@#$......",
".....%&*=-;.....",
"....>,')!~{]....",
"....^/....(_....",
"....:<[.../}....",
"....|123+456....",
".....#7,18~.....",
".......90.......",
"..abbbbcdbbbbe..",
"..fgggg=hggggi..",
".......jk.......",
".......jk.......",
".......jk.......",
".......lm.......",
"................",
]

下がメインのコードとなる。前回と同様この中で使用した名前は架空のものであり、実在の個人名や団体名などと一致するものがあったとしても関係はない。
[任意]ファイル名: treeviewliststoretest2.py

#! /usr/bin/python
# -*- encoding: utf-8 -*-

import sys
try:
  import pygtk
  pygtk.require('2.0')
except:
  pass
try:
  import gtk
except:
  print >> sys.stderr, 'Error: PyGTK is not installed'
  sys.exit(1)
try:
  import sexmark
except:
  print >> sys.stderr, 'Error: sexmark.py required'
  sys.exit(1)


class TreeViewWithColumn(gtk.TreeView):
  """
  コラムを含んだツリービュー
  """
  (
    COLUMN_NUM,
    COLUMN_SEX,
    COLUMN_FAMILY,
    COLUMN_GIVEN,
  ) = range(4)
  def __init__(self, *args, **kwargs):
    gtk.TreeView.__init__(self, *args, **kwargs)
    # テキスト用セルレンダラでは編集可能にできる
    # セル編集後に「edited」シグナルのハンドラが呼ばれるので
    # その中でListStore/TreeStoreの該当データを新しいデータで上書きする流れ
    # ここでは番号用/family name用/given name用の3つの各セルレンダラを設定
    self.renderer_num = gtk.CellRendererText()
    self.renderer_num.connect('edited', self.on_text_edited, self.COLUMN_NUM)  # コラム番号をユーザデータとして追加で渡している
    self.renderer_num.set_property('editable', True)
    self.renderer_family = gtk.CellRendererText()
    self.renderer_family.connect('edited', self.on_text_edited, self.COLUMN_FAMILY)
    self.renderer_family.set_property('editable', True)
    self.renderer_given = gtk.CellRendererText()
    self.renderer_given.connect('edited', self.on_text_edited, self.COLUMN_GIVEN)
    self.renderer_given.set_property('editable', True)
    # コラムの設定
    self.col_num = gtk.TreeViewColumn('No.',
                                      self.renderer_num,
                                      text=self.COLUMN_NUM)
    self.col_sex = gtk.TreeViewColumn('Sex',
                                      gtk.CellRendererPixbuf(),
                                      pixbuf=self.COLUMN_SEX)
    self.col_family = gtk.TreeViewColumn('Family name',
                                         self.renderer_family,
                                         text=self.COLUMN_FAMILY)
    self.col_given = gtk.TreeViewColumn('Given name',
                                        self.renderer_given,
                                        text=self.COLUMN_GIVEN)
    # コラムを追加
    self.append_column(self.col_num)
    self.append_column(self.col_sex)
    self.append_column(self.col_family)
    self.append_column(self.col_given)
    # コンテキストメニュー
    self.item_changesex = gtk.MenuItem('_Change Sex')
    self.menu_popup = gtk.Menu()
    self.menu_popup.add(self.item_changesex)
    self.menu_popup.show_all()
    self.item_changesex.connect('activate', self.on_changesex_activated)
    self.connect('button_press_event', self.on_button_press_event)
  def on_text_edited(self, widget, path, new_text, col):
    """
    テキスト用セルが編集されたときの処理
    col(ユーザデータ)はコラム番号
    """
    # get_model()でツリービューに関連付けられたデータを取り出して
    # 引数の情報とあわせてデータの値を変更する
    model = self.get_model()
    if col == self.COLUMN_NUM:
      try:
        model.set_value(model.get_iter(path), col, int(new_text))
      except ValueError:  # 整数以外が指定された
        pass
    else:
      model.set_value(model.get_iter(path), col, new_text)
  def on_button_press_event(self, widget, event):
    """
    ツリービュー上でマウスボタンが押されたときの処理
    """
    if event.button == 3:
      # データがない部分でコンテキストメニューが開けないようにする
      # (本スクリプトの場合、データの長さを数人分にしないと効果は確認できない)
      # get_path_at_pos()はTreeView内の座標から項目のパスを取得
      path_at_pos = self.get_path_at_pos(int(event.x), int(event.y))
      # クリックした場所に項目があるのが確認できた場合のみポップアップする
      if path_at_pos:  # Noneの場合がある
        self.menu_popup.popup(None, None, None, event.button, event.time)
  def on_changesex_activated(self, widget):
    """
    ツリービュー上のコンテキストメニュー項目を選択したときの処理
    性別のアイコンを変更
    """
    # 現在選択されている項目を取り出す
    (model, iter) = self.get_selection().get_selected()
    # 現在値がmaleであればfemaleに、femaleならmaleに変更
    if model.get_value(iter, self.COLUMN_SEX) == SexIcon.male:
      model.set_value(iter, self.COLUMN_SEX, SexIcon.female)
    else:
      model.set_value(iter, self.COLUMN_SEX, SexIcon.male)

class SexIcon:
  """
  性別のアイコンデータのPixbufを保持
  各データは「[本クラス名].[メンバ変数]」で取り出す
  """
  male = gtk.gdk.pixbuf_new_from_xpm_data(sexmark.male_icon_xpm)
  female = gtk.gdk.pixbuf_new_from_xpm_data(sexmark.female_icon_xpm)

class MainWindow(gtk.Window):
  """
  メインウィンドウ
  """
  # 直接ウィンドウとは関係ないが、データは便宜上ここに用意しておくことにする
  data = \
  [
    (1, SexIcon.male, 'Tanaka', 'Ichiro'),
    (2, SexIcon.female, 'Yamana', 'Hanako'),
    (3, SexIcon.male, 'Urashima', 'Saburo'),
    (4, SexIcon.male, 'Kurusu', 'Santa'),
    (5, SexIcon.male, 'Handa', 'Fuuta'),
    (6, SexIcon.female, 'Umeno', 'Tsubomi'),
    (7, SexIcon.male, 'Yoshi', 'Yaruzo'),
    (8, SexIcon.female, 'Kawai', 'Nuko'),
    (9, SexIcon.male, 'Hoshi', 'Kintaro'),
    (10, SexIcon.female, 'Shirayuki', 'Himeko'),
    (11, SexIcon.female, 'Ashigaka', 'Yui'),
    (12, SexIcon.female, 'Ageyanagi', 'Masako'),
    (13, SexIcon.male, 'Torino', 'Kenta'),
    (14, SexIcon.male, 'Kubota', 'Mochio'),
    (15, SexIcon.female, 'Kuroi', 'Sora'),
    (16, SexIcon.male, 'Hirai', 'Shin'),
    (17, SexIcon.female, 'Akai', 'Midori'),
    (18, SexIcon.female, 'Nakano', 'Anko'),
    (19, SexIcon.male, 'Imai', 'Takeo'),
    (20, SexIcon.male, 'Kouno', 'Torio'),
    (21, SexIcon.male, 'Yoshino', 'Yasu'),
    (22, SexIcon.male, 'Komatsu', 'Taro'),
    (23, SexIcon.male, 'Kondo', 'Musashi'),
    (24, SexIcon.male, 'Ono', 'Ken'),
    (25, SexIcon.male, 'Mochida', 'Usuichi'),
    (26, SexIcon.female, 'Mochida', 'Kineko'),
    (27, SexIcon.female, 'Honma', 'Kayo'),
    (28, SexIcon.male, 'Matsuno', 'Sarunosuke'),
    (29, SexIcon.female, 'Nishi', 'Minami'),
    (30, SexIcon.female, 'Usui', 'Hikaru'),
    (31, SexIcon.male, 'Sato', 'Toshio'),
    (32, SexIcon.male, 'Doi', 'Tsubasa'),
    (33, SexIcon.female, 'Ishimaru', 'Denko'),
    (34, SexIcon.female, 'Usami', 'Mimi'),
    (35, SexIcon.male, 'Hattori', 'Shinobu'),
    (36, SexIcon.female, 'Kago', 'Yuri'),
    (37, SexIcon.male, 'Takeda', 'Ingen'),
    (38, SexIcon.male, 'Kai', 'Dankichi'),
    (39, SexIcon.male, 'Okusa', 'Ben'),
    (40, SexIcon.male, 'Hara', 'Tatsuo'),
    (41, SexIcon.female, 'Mizuno', 'Shizuku'),
    (42, SexIcon.female, 'Baba', 'Nana'),
  ]
  def __init__(self, *args, **kwargs):
    gtk.Window.__init__(self, *args, **kwargs)
    # ショートカットキー(アクセラレータ)
    self.accelgroup = gtk.AccelGroup()
    self.add_accel_group(self.accelgroup)
    # メニュー項目
    self.item_quit = gtk.ImageMenuItem(gtk.STOCK_QUIT, self.accelgroup)
    self.menu_file = gtk.Menu()
    self.menu_file.add(self.item_quit)
    self.item_file = gtk.MenuItem('_File')
    self.item_file.set_submenu(self.menu_file)
    self.menubar = gtk.MenuBar()
    self.menubar.append(self.item_file)
    # ツリービュー
    self.treeview = TreeViewWithColumn(model=gtk.ListStore(int, gtk.gdk.Pixbuf, str, str))
    self.treeview.set_rules_hint(True)
    # ツリービュー向けスクロールウィンドウ
    self.sw = gtk.ScrolledWindow()
    self.sw.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
    self.sw.add(self.treeview)
    # レイアウト用コンテナ
    self.vbox = gtk.VBox()
    self.vbox.pack_start(self.menubar, expand=False, fill=False)
    self.vbox.pack_start(self.sw)
    # シグナル
    self.connect('delete_event', gtk.main_quit)
    self.item_quit.connect('activate', gtk.main_quit)
    # データ追加
    for rec in self.data:
      self.treeview.get_model().append(rec)
    # ウィンドウ
    self.add(self.vbox)
    self.set_size_request(350, 300)

class PyGTKTreeViewListStoreTest2:
  """
  リストを用いたツリービューのテスト2
  """
  def main(self):
    """
    アプリケーションのメイン処理
    """
    win = MainWindow()
    win.show_all()
    gtk.main()


if __name__ == '__main__':
  app = PyGTKTreeViewListStoreTest2()
  app.main()

コンテキストメニューから性別を変更

性別変更後更に編集可能セルの文字列を編集しているところ

関連記事:

• Pythonのimport文について
• PyGTKでツリービューにリストのデータ(ListStore)を表示(概要とメモ)
• PyGTKでツリービューにリストのデータ(ListStore)を表示(簡単な例)
• Pythonにおけるクラスの静的メンバについて
• PyGTKでツリービューにリストのデータ(ListStore)を表示(データを変更可能にする・メモ)
• PyGTKでツリービューに階層を持ったデータ(TreeStore)を表示
• PyGTKでツリービューにリストのデータ(ListStore)を表示(項目をGUI上で追加/削除、データの並べ替えや入れ替えなど)
• PyGTKでツリービューにリストのデータ(ListStore)を表示(モデルの内容を取り出す)
• PyGTKでツリービューにリストのデータ(ListStore)を表示(ボタンからデータの順番を入れ替える・メモ)
• PyGTKでツリービューにリストのデータ(ListStore)を表示(ボタンからデータの順番を入れ替える・コード例)