1

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

     2

     3

#This file is part of Beremiz, a Integrated Development Environment for

     4

#programming IEC 61131-3 automates supporting plcopen standard and CanFestival. 

     5

#

     6

#Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD

     7

#

     8

#See COPYING file for copyrights details.

     9

#

    10

#This library is free software; you can redistribute it and/or

    11

#modify it under the terms of the GNU General Public

    12

#License as published by the Free Software Foundation; either

    13

#version 2.1 of the License, or (at your option) any later version.

    14

#

    15

#This library is distributed in the hope that it will be useful,

    16

#but WITHOUT ANY WARRANTY; without even the implied warranty of

    17

#MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

    18

#General Public License for more details.

    19

#

    20

#You should have received a copy of the GNU General Public

    21

#License along with this library; if not, write to the Free Software

    22

#Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

    23

    24

import wx

    25

from Zeroconf import *

    26

import socket

    27

import  wx.lib.mixins.listctrl  as  listmix

    28

    29

class AutoWidthListCtrl(wx.ListCtrl, listmix.ListCtrlAutoWidthMixin):

    30

    def __init__(self, parent, ID, pos=wx.DefaultPosition,

    31

                 size=wx.DefaultSize, style=0):

    32

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

    33

        listmix.ListCtrlAutoWidthMixin.__init__(self)

    34

    35

class DiscoveryDialog(wx.Dialog, listmix.ColumnSorterMixin):

    36

    def __init__(self, parent, id=-1, title=_('Service Discovery')):

    37

        self.my_result=None

    38

        wx.Dialog.__init__(self, parent, id, title, size=(600,600), style=wx.DEFAULT_DIALOG_STYLE)

    39

    40

        # set up dialog sizer

    41

    42

        sizer = wx.FlexGridSizer(2, 1, 2, 2)  # rows, cols, vgap, hgap

    43

        sizer.AddGrowableRow(0)

    44

        sizer.AddGrowableCol(0)

    45

    46

        # set up list control

    47

    48

        self.list = AutoWidthListCtrl(self, -1,

    49

                                      #pos=(50,20), 

    50

                                      #size=(500,300),

    51

                                      style=wx.LC_REPORT 

    52

                                     | wx.LC_EDIT_LABELS

    53

                                     | wx.LC_SORT_ASCENDING

    54

                                     | wx.LC_SINGLE_SEL 

    55

                                     )

    56

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

    57

    58

        btsizer = wx.FlexGridSizer(1, 6, 2, 2)  # rows, cols, vgap, hgap

    59

    60

        sizer.Add(btsizer, 1, wx.EXPAND)

    61

    62

        self.PopulateList()

    63

    64

        self.Bind(wx.EVT_LIST_ITEM_SELECTED, self.OnItemSelected, self.list)

    65

        self.Bind(wx.EVT_LIST_ITEM_ACTIVATED, self.OnItemActivated, self.list)

    66

    67

        # set up buttons

    68

    69

        local_id = wx.NewId()

    70

        b = wx.Button(self, local_id, _("Refresh"))

    71

        self.Bind(wx.EVT_BUTTON, self.OnRefreshButton, b)

    72

        btsizer.Add(b)

    73

    74

        btsizer.AddSpacer(0)

    75

        btsizer.AddGrowableCol(1)

    76

    77

        local_id = wx.NewId()

    78

        b = wx.Button(self, local_id, _("Local"))

    79

        self.Bind(wx.EVT_BUTTON, self.ChooseLocalID, b)

    80

        btsizer.Add(b)

    81

    82

        btsizer.AddSpacer(0)

    83

        btsizer.AddGrowableCol(3)

    84

    85

        b = wx.Button(self, wx.ID_CANCEL, _("Cancel"))

    86

        self.Bind(wx.EVT_BUTTON, self.OnCancel, b)

    87

        btsizer.Add(b)

    88

    89

        b = wx.Button(self, wx.ID_OK, _("OK"))

    90

        self.Bind(wx.EVT_BUTTON, self.OnOk, b)

    91

        b.SetDefault()

    92

        btsizer.Add(b)

    93

    94

        self.SetSizer(sizer)

    95

    96

        listmix.ColumnSorterMixin.__init__(self, 4)

    97

    98

        # Used by the ColumnSorterMixin, see wx/lib/mixins/listctrl.py

    99

        self.itemDataMap = {}

   100

   101

        # a counter used to assign a unique id to each listctrl item

   102

        self.nextItemId = 0

   103

   104

        self.browser = None

   105

        self.zConfInstance = Zeroconf()

   106

        self.RefreshList()

   107

   108

        self.Bind(wx.EVT_CLOSE, self.OnClose)

   109

   110

    def RefreshList(self):

   111

        type = "_PYRO._tcp.local."

   112

        self.browser = ServiceBrowser(self.zConfInstance, type, self)        

   113

   114

    def OnRefreshButton(self, event):

   115

        self.list.DeleteAllItems()

   116

        self.browser.cancel()

   117

        self.RefreshList()

   118

   119

    def OnClose(self, event):

   120

        self.zConfInstance.close()

   121

        event.Skip()

   122

   123

    def OnCancel(self, event):

   124

        self.zConfInstance.close()

   125

        event.Skip()

   126

   127

    def OnOk(self, event):

   128

        self.zConfInstance.close()

   129

        event.Skip()

   130

   131

    # Used by the ColumnSorterMixin, see wx/lib/mixins/listctrl.py

   132

    def GetListCtrl(self):

   133

        return self.list

   134

   135

    def PopulateList(self):

   136

        self.list.InsertColumn(0, 'NAME')

   137

        self.list.InsertColumn(1, 'TYPE')

   138

        self.list.InsertColumn(2, 'IP')

   139

        self.list.InsertColumn(3, 'PORT')

   140

        self.list.SetColumnWidth(0, 150)

   141

        self.list.SetColumnWidth(1, 150)

   142

        self.list.SetColumnWidth(2, 150)

   143

        self.list.SetColumnWidth(3, 150)

   144

   145

    def getColumnText(self, index, col):

   146

        item = self.list.GetItem(index, col)

   147

        return item.GetText()

   148

   149

    def OnItemSelected(self, event):

   150

        self.currentItem = event.m_itemIndex

   151

        self.setresult()

   152

        event.Skip()

   153

   154

    def OnItemActivated(self, event):

   155

        self.currentItem = event.m_itemIndex

   156

        self.setresult()

   157

        self.Close()

   158

        event.Skip()

   159

   160

    def setresult(self):

   161

        connect_type = self.getColumnText(self.currentItem, 1)

   162

        connect_address = self.getColumnText(self.currentItem, 2)

   163

        connect_port = self.getColumnText(self.currentItem, 3)

   164

   165

        uri = self.CreateURI(connect_type, connect_address, connect_port)

   166

        self.my_result=uri

   167

   168

    def GetResult(self):

   169

        return self.my_result

   170

   171

    def removeService(self, zeroconf, type, name):

   172

        '''

   173

        called when a service with the desired type goes offline.

   174

        '''

   175

   176

        # loop through the list items looking for the service that went offline

   177

        for idx in xrange(self.list.GetItemCount()):

   178

            # this is the unique identifier assigned to the item

   179

            item_id = self.list.GetItemData(idx)

   180

   181

            # this is the full typename that was received by addService

   182

            item_name = self.itemDataMap[item_id][4]

   183

   184

            if item_name == name:

   185

                self.list.DeleteItem(idx)

   186

                break

   187

   188

    def addService(self, zeroconf, type, name):

   189

        '''

   190

        called when a service with the desired type is discovered.

   191

        '''

   192

   193

        info = self.zConfInstance.getServiceInfo(type, name)

   194

   195

        svcname  = name.split(".")[0]

   196

        typename = type.split(".")[0][1:]

   197

        ip       = str(socket.inet_ntoa(info.getAddress()))

   198

        port     = info.getPort()

   199

   200

        num_items = self.list.GetItemCount()

   201

   202

        # display the new data in the list

   203

        new_item = self.list.InsertStringItem(num_items, svcname)

   204

        self.list.SetStringItem(new_item, 1, "%s" % typename)

   205

        self.list.SetStringItem(new_item, 2, "%s" % ip)

   206

        self.list.SetStringItem(new_item, 3, "%s" % port)

   207

   208

        # record the new data for the ColumnSorterMixin

   209

        # we assign every list item a unique id (that won't change when items

   210

        # are added or removed)

   211

        self.list.SetItemData(new_item, self.nextItemId)

   212

   213

        # the value of each column has to be stored in the itemDataMap

   214

        # so that ColumnSorterMixin knows how to sort the column.

   215

   216

        # "name" is included at the end so that self.removeService

   217

        # can access it.

   218

        self.itemDataMap[self.nextItemId] = [ svcname, typename, ip, port, name ]

   219

   220

        self.nextItemId += 1

   221

   222

    def CreateURI(self, connect_type, connect_address, connect_port):

   223

        uri = "%s://%s:%s"%(connect_type, connect_address, connect_port)

   224

        return uri

   225

   226

    def ChooseLocalID(self, event):

   227

        self.my_result = "LOCAL://"

   228

        self.Close()