1

#!/usr/bin/env python

     2

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

     3

#

     4

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

     5

#

     6

#See COPYING file for copyrights details.

     7

#

     8

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

     9

#modify it under the terms of the GNU General Public

    10

#License as published by the Free Software Foundation; either

    11

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

    12

#

    13

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

    14

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

    15

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

    16

#General Public License for more details.

    17

#

    18

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

    19

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

    20

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

    21

import Pyro.core as pyro

    22

from Pyro.errors import PyroError

    23

import Pyro.util

    24

import traceback

    25

from time import sleep

    26

import copy

    27

import socket

    28

service_type = '_PYRO._tcp.local.'

    29

    30

# this module attribute contains a list of DNS-SD (Zeroconf) service types

    31

# supported by this connector confnode.

    32

#

    33

# for connectors that do not support DNS-SD, this attribute can be omitted

    34

# or set to an empty list.

    35

    36

def PYRO_connector_factory(uri, confnodesroot):

    37

    """

    38

    This returns the connector to Pyro style PLCobject

    39

    """

    40

    confnodesroot.logger.write(_("Connecting to URI : %s\n")%uri)

    41

    42

    servicetype, location = uri.split("://")

    43

    if location.find(service_type) != -1:

    44

        try :

    45

            from util.Zeroconf import Zeroconf

    46

            r = Zeroconf()

    47

            i=r.getServiceInfo(service_type, location)

    48

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

    49

            port = str(i.getPort())

    50

            newlocation = ip+':'+port

    51

            confnodesroot.logger.write(_("'%s' is located at %s\n")%(location, newlocation))

    52

            location = newlocation

    53

            r.close()

    54

        except Exception, msg:

    55

            confnodesroot.logger.write_error(_("MDNS resolution failure for '%s'\n")%location)

    56

            confnodesroot.logger.write_error(traceback.format_exc())

    57

            return None

    58

    59

    # Try to get the proxy object

    60

    try :

    61

        RemotePLCObjectProxy = pyro.getAttrProxyForURI("PYROLOC://"+location+"/PLCObject")

    62

    except Exception, msg:

    63

        confnodesroot.logger.write_error(_("Connection to '%s' failed.\n")%location)

    64

        confnodesroot.logger.write_error(traceback.format_exc())

    65

        return None

    66

    67

    def PyroCatcher(func, default=None):

    68

        """

    69

        A function that catch a pyro exceptions, write error to logger

    70

        and return defaul value when it happen

    71

        """

    72

        def catcher_func(*args,**kwargs):

    73

            try:

    74

                return func(*args,**kwargs)

    75

            except Pyro.errors.ProtocolError, e:

    76

                pass

    77

            except Pyro.errors.ConnectionClosedError, e:

    78

                confnodesroot.logger.write_error("Connection lost!\n")

    79

                confnodesroot._connector = None

    80

            except Exception,e:

    81

                #confnodesroot.logger.write_error(traceback.format_exc())

    82

                errmess = ''.join(Pyro.util.getPyroTraceback(e))

    83

                confnodesroot.logger.write_error(errmess+"\n")

    84

                print errmess

    85

                confnodesroot._connector = None

    86

                return default

    87

        return catcher_func

    88

    89

    # Check connection is effective. 

    90

    # lambda is for getattr of GetPLCstatus to happen inside catcher

    91

    if PyroCatcher(lambda:RemotePLCObjectProxy.GetPLCstatus())() is None:

    92

        confnodesroot.logger.write_error(_("Cannot get PLC status - connection failed.\n"))

    93

        return None

    94

    95

    96

    class PyroProxyProxy:

    97

        """

    98

        A proxy proxy class to handle Beremiz Pyro interface specific behavior.

    99

        And to put pyro exception catcher in between caller and pyro proxy

   100

        """

   101

        def __init__(self):

   102

            # for safe use in from debug thread, must create a copy

   103

            self.RemotePLCObjectProxyCopy = None

   104

   105

        def GetPyroProxy(self):

   106

            """

   107

            This func returns the real Pyro Proxy.

   108

            Use this if you musn't keep reference to it.

   109

            """

   110

            return RemotePLCObjectProxy

   111

   112

        def _PyroStartPLC(self, *args, **kwargs):

   113

            """

   114

            confnodesroot._connector.GetPyroProxy() is used 

   115

            rather than RemotePLCObjectProxy because

   116

            object is recreated meanwhile, 

   117

            so we must not keep ref to it here

   118

            """

   119

            current_status = confnodesroot._connector.GetPyroProxy().GetPLCstatus()

   120

            if current_status == "Dirty":

   121

                """

   122

                Some bad libs with static symbols may polute PLC

   123

                ask runtime to suicide and come back again

   124

                """

   125

                confnodesroot.logger.write(_("Force runtime reload\n"))

   126

                confnodesroot._connector.GetPyroProxy().ForceReload()

   127

                confnodesroot._Disconnect()

   128

                # let remote PLC time to resurect.(freeze app)

   129

                sleep(0.5)

   130

                confnodesroot._Connect()

   131

            self.RemotePLCObjectProxyCopy = copy.copy(confnodesroot._connector.GetPyroProxy())

   132

            return confnodesroot._connector.GetPyroProxy().StartPLC(*args, **kwargs)

   133

        StartPLC = PyroCatcher(_PyroStartPLC, False)

   134

   135

   136

        def _PyroGetTraceVariables(self):

   137

            """

   138

            for safe use in from debug thread, must use the copy

   139

            """

   140

            if self.RemotePLCObjectProxyCopy is None:

   141

                self.RemotePLCObjectProxyCopy = copy.copy(confnodesroot._connector.GetPyroProxy())

   142

            return self.RemotePLCObjectProxyCopy.GetTraceVariables()

   143

        GetTraceVariables = PyroCatcher(_PyroGetTraceVariables,("Broken",None,None))

   144

   145

        def _PyroGetPLCstatus(self):

   146

            return RemotePLCObjectProxy.GetPLCstatus()

   147

        GetPLCstatus = PyroCatcher(_PyroGetPLCstatus, ("Broken",None))

   148

   149

        def _PyroRemoteExec(self, script, **kwargs):

   150

            return RemotePLCObjectProxy.RemoteExec(script, **kwargs)

   151

        RemoteExec = PyroCatcher(_PyroRemoteExec, (-1, "RemoteExec script failed!"))

   152

   153

        def __getattr__(self, attrName):

   154

            member = self.__dict__.get(attrName, None)

   155

            if member is None:

   156

                def my_local_func(*args,**kwargs):

   157

                    return RemotePLCObjectProxy.__getattr__(attrName)(*args,**kwargs)

   158

                member = PyroCatcher(my_local_func, None)

   159

                self.__dict__[attrName] = member

   160

            return member

   161

   162

    return PyroProxyProxy()

   163

   164