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 socket

    25

import wx

    26

import  wx.lib.mixins.listctrl  as  listmix

    27

from Zeroconf import *

    28

    29

import connectors

    30

    31

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

    32

    def __init__(self, parent, id, name, pos=wx.DefaultPosition,

    33

                 size=wx.DefaultSize, style=0):

    34

        wx.ListCtrl.__init__(self, parent, id, pos, size, style, name=name)

    35

        listmix.ListCtrlAutoWidthMixin.__init__(self)

    36

    37

[ID_DISCOVERYDIALOG, ID_DISCOVERYDIALOGSTATICTEXT1, 

    38

 ID_DISCOVERYDIALOGSERVICESLIST, ID_DISCOVERYDIALOGREFRESHBUTTON, 

    39

 ID_DISCOVERYDIALOGLOCALBUTTON, 

    40

] = [wx.NewId() for _init_ctrls in range(5)]

    41

    42

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

    43

    44

    def _init_coll_MainSizer_Items(self, parent):

    45

        parent.AddWindow(self.staticText1, 0, border=20, flag=wx.TOP|wx.LEFT|wx.RIGHT|wx.GROW)

    46

        parent.AddWindow(self.ServicesList, 0, border=20, flag=wx.LEFT|wx.RIGHT|wx.GROW)

    47

        parent.AddSizer(self.ButtonGridSizer, 0, border=20, flag=wx.LEFT|wx.RIGHT|wx.BOTTOM|wx.GROW)

    48

    49

    def _init_coll_MainSizer_Growables(self, parent):

    50

        parent.AddGrowableCol(0)

    51

        parent.AddGrowableRow(1)

    52

    53

    def _init_coll_ButtonGridSizer_Items(self, parent):

    54

        parent.AddWindow(self.RefreshButton, 0, border=0, flag=0)

    55

        parent.AddWindow(self.LocalButton, 0, border=0, flag=0)

    56

        parent.AddSizer(self.ButtonSizer, 0, border=0, flag=0)

    57

    58

    def _init_coll_ButtonGridSizer_Growables(self, parent):

    59

        parent.AddGrowableCol(0)

    60

        parent.AddGrowableCol(1)

    61

        parent.AddGrowableRow(1)

    62

    63

    def _init_sizers(self):

    64

        self.MainSizer = wx.FlexGridSizer(cols=1, hgap=0, rows=3, vgap=10)

    65

        self.ButtonGridSizer = wx.FlexGridSizer(cols=3, hgap=5, rows=1, vgap=0)

    66

    67

        self._init_coll_MainSizer_Items(self.MainSizer)

    68

        self._init_coll_MainSizer_Growables(self.MainSizer)

    69

        self._init_coll_ButtonGridSizer_Items(self.ButtonGridSizer)

    70

        self._init_coll_ButtonGridSizer_Growables(self.ButtonGridSizer)

    71

    72

        self.SetSizer(self.MainSizer)

    73

    74

    def _init_ctrls(self, prnt):

    75

        wx.Dialog.__init__(self, id=ID_DISCOVERYDIALOG, 

    76

              name='DiscoveryDialog', parent=prnt,  

    77

              size=wx.Size(600, 600), style=wx.DEFAULT_DIALOG_STYLE,

    78

              title='Service Discovery')

    79

    80

        self.staticText1 = wx.StaticText(id=ID_DISCOVERYDIALOGSTATICTEXT1,

    81

              label=_('Services available:'), name='staticText1', parent=self,

    82

              pos=wx.Point(0, 0), size=wx.DefaultSize, style=0)

    83

    84

        # Set up list control

    85

        self.ServicesList = AutoWidthListCtrl(id=ID_DISCOVERYDIALOGSERVICESLIST,

    86

              name='ServicesList', parent=self, pos=wx.Point(0, 0), size=wx.Size(0, 0), 

    87

              style=wx.LC_REPORT|wx.LC_EDIT_LABELS|wx.LC_SORT_ASCENDING|wx.LC_SINGLE_SEL)

    88

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

    89

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

    90

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

    91

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

    92

        self.ServicesList.SetColumnWidth(0, 150)

    93

        self.ServicesList.SetColumnWidth(1, 150)

    94

        self.ServicesList.SetColumnWidth(2, 150)

    95

        self.ServicesList.SetColumnWidth(3, 150)

    96

        self.Bind(wx.EVT_LIST_ITEM_SELECTED, self.OnItemSelected, id=ID_DISCOVERYDIALOGSERVICESLIST)

    97

        self.Bind(wx.EVT_LIST_ITEM_ACTIVATED, self.OnItemActivated, id=ID_DISCOVERYDIALOGSERVICESLIST)

    98

    99

        listmix.ColumnSorterMixin.__init__(self, 4)

   100

   101

        self.RefreshButton = wx.Button(id=ID_DISCOVERYDIALOGREFRESHBUTTON,

   102

              label=_('Refresh'), name='RefreshButton', parent=self,

   103

              pos=wx.Point(0, 0), size=wx.DefaultSize, style=0)

   104

        self.Bind(wx.EVT_BUTTON, self.OnRefreshButton, id=ID_DISCOVERYDIALOGREFRESHBUTTON)

   105

   106

        self.LocalButton = wx.Button(id=ID_DISCOVERYDIALOGLOCALBUTTON,

   107

              label=_('Local'), name='LocalButton', parent=self,

   108

              pos=wx.Point(0, 0), size=wx.DefaultSize, style=0)

   109

        self.Bind(wx.EVT_BUTTON, self.OnLocalButton, id=ID_DISCOVERYDIALOGLOCALBUTTON)

   110

   111

        self.ButtonSizer = self.CreateButtonSizer(wx.OK|wx.CANCEL|wx.CENTER)

   112

   113

        self._init_sizers()

   114

   115

    def __init__(self, parent):

   116

        self._init_ctrls(parent)

   117

   118

        self.itemDataMap = {}

   119

        self.nextItemId = 0

   120

   121

        self.URI = None

   122

        self.Browsers = []

   123

   124

        self.ZeroConfInstance = Zeroconf()

   125

        self.RefreshList()

   126

   127

    def __del__(self):

   128

        for browser in self.Browsers:

   129

            browser.cancel()

   130

        self.ZeroConfInstance.close()

   131

   132

    def RefreshList(self):

   133

        for browser in self.Browsers:

   134

            browser.cancel()

   135

   136

        self.Browsers = []

   137

        for t in connectors.dnssd_connectors.keys():

   138

            self.Browsers.append(ServiceBrowser(self.ZeroConfInstance, t, self))

   139

   140

    def OnRefreshButton(self, event):

   141

        self.ServicesList.DeleteAllItems()

   142

        self.RefreshList()

   143

   144

    def OnLocalButton(self, event):

   145

        self.URI = "LOCAL://"

   146

        self.EndModal(wx.ID_OK)

   147

        event.Skip()

   148

   149

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

   150

    def GetListCtrl(self):

   151

        return self.ServicesList

   152

   153

    def getColumnText(self, index, col):

   154

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

   155

        return item.GetText()

   156

   157

    def OnItemSelected(self, event):

   158

        self.SetURI(event.m_itemIndex)

   159

        event.Skip()

   160

   161

    def OnItemActivated(self, event):

   162

        self.SetURI(event.m_itemIndex)

   163

        self.EndModal(wx.ID_OK)

   164

        event.Skip()

   165

   166

    def SetURI(self, idx):

   167

        connect_type = self.getColumnText(idx, 1)

   168

        connect_address = self.getColumnText(idx, 2)

   169

        connect_port = self.getColumnText(idx, 3)

   170

   171

        self.URI = "%s://%s:%s"%(connect_type, connect_address, connect_port)

   172

   173

    def GetURI(self):

   174

        return self.URI

   175

   176

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

   177

        '''

   178

        called when a service with the desired type goes offline.

   179

        '''

   180

   181

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

   182

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

   183

            # this is the unique identifier assigned to the item

   184

            item_id = self.ServicesList.GetItemData(idx)

   185

   186

            # this is the full typename that was received by addService

   187

            item_name = self.itemDataMap[item_id][4]

   188

   189

            if item_name == name:

   190

                self.ServicesList.DeleteItem(idx)

   191

                break

   192

   193

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

   194

        '''

   195

        called when a service with the desired type is discovered.

   196

        '''

   197

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

   198

   199

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

   200

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

   201

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

   202

        port     = info.getPort()

   203

   204

        num_items = self.ServicesList.GetItemCount()

   205

   206

        # display the new data in the list

   207

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

   208

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

   209

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

   210

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

   211

   212

        # record the new data for the ColumnSorterMixin

   213

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

   214

        # are added or removed)

   215

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

   216

   217

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

   218

        # so that ColumnSorterMixin knows how to sort the column.

   219

   220

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

   221

        # can access it.

   222

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

   223

   224

        self.nextItemId += 1

   225