1

#!/usr/bin/env python

     2

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

     3

#

     5

#

     7

#

    12

#

    17

#

    24

from Pyro.errors import PyroError

    25

import traceback

    26

from time import sleep

    27

import copy

    28

import socket

    29

service_type = '_PYRO._tcp.local.'

    31

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

    32

# supported by this connector confnode.

    33

#

    34

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

    35

# or set to an empty list.

    36

    37

def PYRO_connector_factory(uri, confnodesroot):

    38

    """

    39

    This returns the connector to Pyro style PLCobject

    40

    """

    42

    43

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

    76

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

    78

            from util.Zeroconf import Zeroconf

    79

            r = Zeroconf()

    83

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

    84

            port = str(i.getPort())

    87

            location = newlocation

    88

            r.close()

    89

        except Exception, msg:

    91

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

    92

            return None

    93

    94

    # Try to get the proxy object

    97

    except Exception, msg:

    99

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

   100

        return None

   101

   102

    def PyroCatcher(func, default=None):

   103

        """

   106

        """

   108

            try:

   110

            except Pyro.errors.ConnectionClosedError, e:

   111

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

   112

                confnodesroot._SetConnector(None)

   113

            except Pyro.errors.ProtocolError, e:

   117

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

   119

                print errmess

   120

                confnodesroot._SetConnector(None)

   121

            return default

   122

        return catcher_func

   123

   124

    # Check connection is effective.

   125

    # lambda is for getattr of GetPLCstatus to happen inside catcher

   127

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

   128

        return None

   129

   130

    class PyroProxyProxy(object):

   131

        """

   132

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

   134

        """

   135

        def __init__(self):

   136

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

   137

            self.RemotePLCObjectProxyCopy = None

   138

   139

        def GetPyroProxy(self):

   140

            """

   141

            This func returns the real Pyro Proxy.

   142

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

   143

            """

   144

            return RemotePLCObjectProxy

   145

   146

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

   147

            """

   148

            confnodesroot._connector.GetPyroProxy() is used

   149

            rather than RemotePLCObjectProxy because

   150

            object is recreated meanwhile,

   151

            so we must not keep ref to it here

   152

            """

   153

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

   154

            if current_status == "Dirty":

   155

                """

   156

                Some bad libs with static symbols may polute PLC

   157

                ask runtime to suicide and come back again

   158

                """

   159

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

   160

                confnodesroot._connector.GetPyroProxy().ForceReload()

   161

                confnodesroot._Disconnect()

   162

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

   163

                sleep(0.5)

   164

                confnodesroot._Connect()

   165

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

   166

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

   167

        StartPLC = PyroCatcher(_PyroStartPLC, False)

   168

   169

        def _PyroGetTraceVariables(self):

   170

            """

   171

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

   172

            """

   173

            if self.RemotePLCObjectProxyCopy is None:

   174

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

   175

            return self.RemotePLCObjectProxyCopy.GetTraceVariables()

   177

   178

        def _PyroGetPLCstatus(self):

   179

            return RemotePLCObjectProxy.GetPLCstatus()

   181

   182

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

   183

            return RemotePLCObjectProxy.RemoteExec(script, **kwargs)

   184

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

   185

   186

        def __getattr__(self, attrName):

   187

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

   188

            if member is None:

   191

                member = PyroCatcher(my_local_func, None)

   192

                self.__dict__[attrName] = member

   193

            return member

   194

   195

    return PyroProxyProxy()