1

#!/usr/bin/env python

     2

# -*- coding: utf-8 -*-

     3

     4

# See COPYING file for copyrights details.

     5

     6

from __future__ import absolute_import

     7

import os

     8

import wx

     9

import wx.dataview as dv

    12

    13

class IDBrowserModel(dv.PyDataViewIndexListModel):

    14

    def __init__(self, psk_path, columncount):

    15

        self.psk_path = psk_path

    16

        self.columncount = columncount

    18

        dv.PyDataViewIndexListModel.__init__(self, len(self.data))

    19

    20

    def _saveData(self):

    22

    23

    def GetColumnType(self, col):

    24

        return "string"

    25

    26

    def GetValueByRow(self, row, col):

    27

        return self.data[row][col]

    28

    29

    def SetValueByRow(self, value, row, col):

    30

        self.data[row][col] = value

    31

        self._saveData()

    32

    33

    def GetColumnCount(self):

    34

        return len(self.data[0]) if self.data else self.columncount

    35

    36

    def GetCount(self):

    37

        return len(self.data)

    38

    39

    def GetAttrByRow(self, row, col, attr):

    40

        if col == 3:

    41

            attr.SetColour('blue')

    42

            attr.SetBold(True)

    43

            return True

    44

        return False

    45

    46

    47

    def Compare(self, item1, item2, col, ascending):

    48

        if not ascending: # swap sort order?

    49

            item2, item1 = item1, item2

    50

        row1 = self.GetRow(item1)

    51

        row2 = self.GetRow(item2)

    52

        if col == 0:

    53

            return cmp(int(self.data[row1][col]), int(self.data[row2][col]))

    54

        else:

    55

            return cmp(self.data[row1][col], self.data[row2][col])

    56

    57

    def DeleteRows(self, rows):

    58

        rows = list(rows)

    59

        rows.sort(reverse=True)

    60

    61

        for row in rows:

    63

            del self.data[row]

    64

            self.RowDeleted(row)

    65

        self._saveData()

    66

    67

    def AddRow(self, value):

    68

        self.data.append(value)

    69

        self.RowAppended()

    70

        self._saveData()

    71

    72

colflags = dv.DATAVIEW_COL_RESIZABLE|dv.DATAVIEW_COL_SORTABLE

    73

    74

class IDBrowser(wx.Panel):

    75

    def __init__(self, parent, ctr, SelectURICallBack=None, SelectIDCallBack=None, **kwargs):

    76

        wx.Panel.__init__(self, parent, -1, size=(400,200))

    77

    78

        self.isManager = SelectURICallBack is None and SelectIDCallBack is None 

    79

        self.SelectURICallBack = SelectURICallBack

    80

        self.SelectIDCallBack = SelectIDCallBack

    81

    82

        dvStyle = wx.BORDER_THEME | dv.DV_ROW_LINES

    83

        if self.isManager :

    84

            # no multiple selection in selector mode

    85

            dvStyle |= dv.DV_MULTIPLE

    86

        self.dvc = dv.DataViewCtrl(self, style = dvStyle)

    87

    88

        args = lambda *a,**k:(a,k)

    89

    90

        ColumnsDesc = [

    91

            args(_("ID"), COL_ID, width = 100),

    92

            args(_("Last URI"), COL_URI, width = 80),

    93

            args(_("Description"), COL_DESC, width = 200, 

    94

                mode = dv.DATAVIEW_CELL_EDITABLE 

    95

                       if self.isManager 

    96

                       else dv.DATAVIEW_CELL_INERT),

    97

            args(_("Last connection"),  COL_LAST, width = 100),

    98

        ]

    99

   101

        self.dvc.AssociateModel(self.model)

   102

   103

        for a,k in ColumnsDesc:

   104

            self.dvc.AppendTextColumn(*a,**dict(k, flags = colflags))

   105

   106

        # TODO : when select,

   107

        #  - update ID field of scheme editor

   108

        #  - enable use URI button

   109

   110

        self.Sizer = wx.BoxSizer(wx.VERTICAL) 

   111

        self.Sizer.Add(self.dvc, 1, wx.EXPAND)

   112

   113

        btnbox = wx.BoxSizer(wx.HORIZONTAL)

   114

        if self.isManager :

   115

   116

            # deletion of secret and metadata

   117

            deleteButton = wx.Button(self, label=_("Delete ID"))

   118

            self.Bind(wx.EVT_BUTTON, self.OnDeleteButton, deleteButton)

   119

            btnbox.Add(deleteButton, 0, wx.LEFT|wx.RIGHT, 5)

   120

   121

            # export all

   122

            exportButton = wx.Button(self, label=_("Export all"))

   123

            self.Bind(wx.EVT_BUTTON, self.OnExportButton, exportButton)

   124

            btnbox.Add(exportButton, 0, wx.LEFT|wx.RIGHT, 5)

   125

   126

            # import with a merge -> duplicates are asked for

   127

            importButton = wx.Button(self, label=_("Import"))

   128

            self.Bind(wx.EVT_BUTTON, self.OnImportButton, importButton)

   129

            btnbox.Add(importButton, 0, wx.LEFT|wx.RIGHT, 5)

   130

   131

        else :

   132

            # selector mode

   133

            # use last known URI button

   134

            # TODO : disable use URI button until something selected

   135

            self.useURIButton = wx.Button(self, label=_("Use last URI"))

   136

            self.Bind(wx.EVT_BUTTON, self.OnUseURIButton, self.useURIButton)

   137

            self.useURIButton.Disable()

   138

            btnbox.Add(self.useURIButton, 0, wx.LEFT|wx.RIGHT, 5)

   139

   140

        self.Sizer.Add(btnbox, 0, wx.TOP|wx.BOTTOM, 5)

   141

        self.Bind(dv.EVT_DATAVIEW_SELECTION_CHANGED, self.OnSelectionChanged, self.dvc)

   142

   143

   144

    def OnDeleteButton(self, evt):

   145

        items = self.dvc.GetSelections()

   146

        rows = [self.model.GetRow(item) for item in items]

   147

   148

        # Ask if user really wants to delete

   149

        if wx.MessageBox(_('Are you sure to delete selected IDs?'),

   150

                         _('Delete IDs'),

   151

                             wx.YES_NO | wx.CENTRE | wx.NO_DEFAULT) != wx.YES:

   152

            return

   153

   154

        self.model.DeleteRows(rows)

   155

   156

    def OnSelectionChanged(self, evt):

   157

        if not self.isManager :

   158

            items = self.dvc.GetSelections()

   159

            somethingSelected = len(items) > 0

   160

            self.useURIButton.Enable(somethingSelected)

   161

            if somethingSelected:

   162

                row = self.model.GetRow(items[0])

   163

                ID = self.model.GetValueByRow(row, COL_ID)

   164

                self.SelectIDCallBack(ID)

   165

   166

   167

    def OnUseURIButton(self, evt):

   168

        row = self.model.GetRow(self.dvc.GetSelections()[0])

   169

        URI = self.model.GetValueByRow(row, COL_URI)

   170

        if URI:

   171

            self.SelectURICallBack(URI)

   172

   173

    def OnExportButton(self, evt):

   174

        # TODO 

   175

        wx.MessageBox(_('?'),

   176

                      _('Mhe'),

   177

                      wx.YES_NO | wx.CENTRE | wx.NO_DEFAULT)

   178

   179

    def OnImportButton(self, evt):

   180

        # TODO 

   181

        wx.MessageBox(_('?'),

   182

                      _('Mhe'),

   183

                      wx.YES_NO | wx.CENTRE | wx.NO_DEFAULT)

   184