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

            if i is None : raise Exception, "'%s' not found"%location

    49

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

    50

            port = str(i.getPort())

    51

            newlocation = ip+':'+port

    52

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

    53

            location = newlocation

    54

            r.close()

    55

        except Exception, msg:

    56

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

    57

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

    58

            return None

    59

    60

    # Try to get the proxy object

    61

    try :

    62

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

    63

    except Exception, msg:

    64

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

    65

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

    66

        return None

    67

    68

    def PyroCatcher(func, default=None):

    69

        """

    70

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

    71

        and return defaul value when it happen

    72

        """

    73

        def catcher_func(*args,**kwargs):

    74

            try:

    75

                return func(*args,**kwargs)

    76

            except Pyro.errors.ConnectionClosedError, e:

    77

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

    78

                confnodesroot._SetConnector(None)

    79

            except Pyro.errors.ProtocolError, e:

    80

                confnodesroot.logger.write_error("Pyro exception: "+str(e)+"\n")

    81

            except Exception,e:

    82

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

    83

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

    84

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

    85

                print errmess

    86

                confnodesroot._SetConnector(None)

    87

            return default

    88

        return catcher_func

    89

    90

    # Check connection is effective.

    91

    # lambda is for getattr of GetPLCstatus to happen inside catcher

    92

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

    93

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

    94

        return None

    95

    96

    97

    class PyroProxyProxy:

    98

        """

    99

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

   100

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

   101

        """

   102

        def __init__(self):

   103

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

   104

            self.RemotePLCObjectProxyCopy = None

   105

   106

        def GetPyroProxy(self):

   107

            """

   108

            This func returns the real Pyro Proxy.

   109

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

   110

            """

   111

            return RemotePLCObjectProxy

   112

   113

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

   114

            """

   115

            confnodesroot._connector.GetPyroProxy() is used

   116

            rather than RemotePLCObjectProxy because

   117

            object is recreated meanwhile,

   118

            so we must not keep ref to it here

   119

            """

   120

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

   121

            if current_status == "Dirty":

   122

                """

   123

                Some bad libs with static symbols may polute PLC

   124

                ask runtime to suicide and come back again

   125

                """

   126

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

   127

                confnodesroot._connector.GetPyroProxy().ForceReload()

   128

                confnodesroot._Disconnect()

   129

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

   130

                sleep(0.5)

   131

                confnodesroot._Connect()

   132

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

   133

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

   134

        StartPLC = PyroCatcher(_PyroStartPLC, False)

   135

   136

   137

        def _PyroGetTraceVariables(self):

   138

            """

   139

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

   140

            """

   141

            if self.RemotePLCObjectProxyCopy is None:

   142

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

   143

            return self.RemotePLCObjectProxyCopy.GetTraceVariables()

   144

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

   145

   146

        def _PyroGetPLCstatus(self):

   147

            return RemotePLCObjectProxy.GetPLCstatus()

   148

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

   149

   150

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

   151

            return RemotePLCObjectProxy.RemoteExec(script, **kwargs)

   152

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

   153

   154

        def __getattr__(self, attrName):

   155

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

   156

            if member is None:

   157

                def my_local_func(*args,**kwargs):

   158

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

   159

                member = PyroCatcher(my_local_func, None)

   160

                self.__dict__[attrName] = member

   161

            return member

   162

   163

    return PyroProxyProxy()

   164

   165