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

    28

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

    29

# supported by this connector plugin.

    30

#

    31

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

    32

# or set to an empty list.

    33

supported_dnssd_services = ["_PYRO._tcp.local."]

    34

    35

def PYRO_connector_factory(uri, pluginsroot):

    36

    """

    37

    This returns the connector to Pyro style PLCobject

    38

    """

    39

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

    40

    41

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

    42

    43

    # Try to get the proxy object

    44

    try :

    45

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

    46

    except Exception, msg:

    47

        pluginsroot.logger.write_error(_("Wrong URI, please check it !\n"))

    48

        pluginsroot.logger.write_error(traceback.format_exc())

    49

        return None

    50

    51

    def PyroCatcher(func, default=None):

    52

        """

    53

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

    54

        and return defaul value when it happen

    55

        """

    56

        def catcher_func(*args,**kwargs):

    57

            try:

    58

                return func(*args,**kwargs)

    59

            except Pyro.errors.ProtocolError, e:

    60

                pass

    61

            except Pyro.errors.ConnectionClosedError, e:

    62

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

    63

                pluginsroot._connector = None

    64

            except Exception,e:

    65

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

    66

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

    67

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

    68

                print errmess

    69

                pluginsroot._connector = None

    70

                return default

    71

        return catcher_func

    72

    73

    # Check connection is effective. 

    74

    # lambda is for getattr of GetPLCstatus to happen inside catcher

    75

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

    76

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

    77

        return None

    78

    79

    80

    class PyroProxyProxy:

    81

        """

    82

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

    83

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

    84

        """

    85

        def __init__(self):

    86

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

    87

            self.RemotePLCObjectProxyCopy = None

    88

    89

        def GetPyroProxy(self):

    90

            """

    91

            This func returns the real Pyro Proxy.

    92

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

    93

            """

    94

            return RemotePLCObjectProxy

    95

    96

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

    97

            """

    98

            pluginsroot._connector.GetPyroProxy() is used 

    99

            rather than RemotePLCObjectProxy because

   100

            object is recreated meanwhile, 

   101

            so we must not keep ref to it here

   102

            """

   103

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

   104

            if current_status == "Dirty":

   105

                """

   106

                Some bad libs with static symbols may polute PLC

   107

                ask runtime to suicide and come back again

   108

                """

   109

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

   110

                pluginsroot._connector.GetPyroProxy().ForceReload()

   111

                pluginsroot._Disconnect()

   112

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

   113

                sleep(0.5)

   114

                pluginsroot._Connect()

   115

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

   116

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

   117

        StartPLC = PyroCatcher(_PyroStartPLC, False)

   118

   119

   120

        def _PyroGetTraceVariables(self):

   121

            """

   122

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

   123

            """

   124

            if self.RemotePLCObjectProxyCopy is None:

   125

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

   126

            return self.RemotePLCObjectProxyCopy.GetTraceVariables()

   127

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

   128

   129

        def _PyroGetPLCstatus(self):

   130

            return RemotePLCObjectProxy.GetPLCstatus()

   131

        GetPLCstatus = PyroCatcher(_PyroGetPLCstatus, "Broken")

   132

   133

        def __getattr__(self, attrName):

   134

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

   135

            if member is None:

   136

                def my_local_func(*args,**kwargs):

   137

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

   138

                member = PyroCatcher(my_local_func, None)

   139

                self.__dict__[attrName] = member

   140

            return member

   141

   142

    return PyroProxyProxy()

   143

   144