wxPython - Как получить данные, проверенные после сортировки (listmix.ColumnSorterMixin)

У меня есть данные в словаре, которые пойдут в ListCtrl. У меня есть флажки listmix.CheckListCtrlMixin для выбора данных и listmix.ColumnSorterMixin для сортировки.

Проблема в том, что после того, как я отсортировал данные по любому столбцу, первоначальный порядок данных изменился, и после проверки элементов я не могу найти данные для дальнейших действий.

Можно ли получить новый "после сортировки" дикт или что-то в этом роде?

Вот демонстрационный пример.

import sys
import wx
import  wx.lib.mixins.listctrl  as  listmix

musicdata = {
1  : ( 'Bad English', 'The Price Of Love', 'Rock' ),
2  : ( 'Michael Bolton', 'How Am I Supposed To Live Without You', 'Blues' ),
3  : ( 'Paul Young', 'Oh Girl', 'Rock' ),
4  : ( 'Paula Abdul', 'Opposites Attract', 'Rock' ),
5  : ( 'Richard Marx', 'Should\'ve Known Better', 'Rock' ),
6 : ( 'Rod Stewart', 'Forever Young', 'Rock' ),
7 : ( 'Roxette', 'Dangerous', 'Rock' ),
}

class TestListCtrl( wx.ListCtrl, listmix.ListCtrlAutoWidthMixin, listmix.CheckListCtrlMixin ) :

    def __init__( self, parent, ID, pos=wx.DefaultPosition,
              size=wx.DefaultSize, style=0 ) :

        wx.ListCtrl.__init__( self, parent, ID, pos, size, style )

        listmix.ListCtrlAutoWidthMixin.__init__( self )
        listmix.CheckListCtrlMixin.__init__(self)
        self.Bind(wx.EVT_LIST_COL_CLICK, self.OnCheckItem)


    def OnCheckItem(self, data, flag):
        print(data, flag)

class TabPanel( wx.Panel, listmix.ColumnSorterMixin ) :

    def __init__( self, parent  ) :

        wx.Panel.__init__( self, parent=parent, id=wx.ID_ANY )

        self.createAndLayout()

    def createAndLayout( self ) :

        sizer = wx.BoxSizer( wx.VERTICAL )
        self.list = TestListCtrl( self, wx.ID_ANY, style=wx.LC_REPORT
                             | wx.BORDER_NONE
                             | wx.LC_EDIT_LABELS
                             | wx.LC_SORT_ASCENDING )

        sizer.Add( self.list, 1, wx.EXPAND )
        self.populateList()

        self.itemDataMap = musicdata
        listmix.ColumnSorterMixin.__init__( self, 3 )
        self.SetSizer( sizer )
        self.SetAutoLayout( True )

    def populateList( self ) :

        self.list.InsertColumn( 0, 'Artist' )
        self.list.InsertColumn( 1, 'Title', wx.LIST_FORMAT_RIGHT )
        self.list.InsertColumn( 2, 'Genre' )
        items = musicdata.items()

        for key, data in items :
            index = self.list.InsertStringItem( sys.maxint, data[ 0 ] )
            self.list.SetStringItem( index, 1, data[ 1 ] )
            self.list.SetStringItem( index, 2, data[ 2 ] )
            self.list.SetItemData( index, key )

        self.list.SetColumnWidth( 0, wx.LIST_AUTOSIZE )
        self.list.SetColumnWidth( 1, wx.LIST_AUTOSIZE )
        self.list.SetColumnWidth( 2, 100 )

        self.list.SetItemState( 5, wx.LIST_STATE_SELECTED, wx.LIST_STATE_SELECTED )

        self.currentItem = 0

    def GetListCtrl( self ) :
        return self.list

class DemoFrame( wx.Frame ) :

     def __init__( self ) :

        wx.Frame.__init__( self, None, wx.ID_ANY, title="Panel Tutorial",
                           size=(600, 300) )

        panel = TabPanel( self )
        self.Show()

app = wx.App()
frame = DemoFrame()
app.MainLoop()

1 ответ

Решение

Как показывает ответ на этот вопрос: Сортировка словаря Python по значению, сортировку dict невозможно, только чтобы получить представление сортировки dict.

Вы можете получить список элементов в отсортированном порядке после нажатия на один из заголовков столбцов, определив функцию в class TestListCtrl как это:

class TestListCtrl( wx.ListCtrl, listmix.ListCtrlAutoWidthMixin, listmix.CheckListCtrlMixin ) :

    def __init__( self, parent, ID, pos=wx.DefaultPosition,
              size=wx.DefaultSize, style=0 ) :

        wx.ListCtrl.__init__( self, parent, ID, pos, size, style )

        listmix.ListCtrlAutoWidthMixin.__init__( self )
        listmix.CheckListCtrlMixin.__init__(self)
        self.Bind(wx.EVT_LIST_COL_CLICK, self.OnCheckItem)
        self.Binf(wx.EVT_LIST_COL_CLICK, self.GetSortedOrder)


    def OnCheckItem(self, data, flag):
        print(data, flag)

    def GetSortedOrder(self, event):
        count = self.GetItemCount()
        self.sorted_artists = [self.GetItem(itemId=row, col=0).GetText() for row in xrange(count)]
        print self.sorted_artists

Метод GetSortedOrder() создаст sorted_artists атрибут, представляющий собой список исполнителей в отсортированном порядке, обновляемый каждый раз, когда вы щелкаете по любому из заголовков столбцов. Вы можете получить другие значения, такие как Название или Жанр, изменив имя столбца. Используйте это, чтобы сравнить с исходным диктом и создать новый, если хотите

Другие вопросы по тегам