1

#!/usr/bin/env python

     2

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

     3

     4

# This file is part of Beremiz, a Integrated Development Environment for

     5

# programming IEC 61131-3 automates supporting plcopen standard and CanFestival.

     6

#

     7

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

     8

#

     9

# See COPYING file for copyrights details.

    10

#

    11

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

    12

# modify it under the terms of the GNU General Public License

    13

# as published by the Free Software Foundation; either version 2

    14

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

    15

#

    16

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

    17

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

    18

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

    19

# GNU General Public License for more details.

    20

#

    21

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

    22

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

    23

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

    24

    25

import sys

    26

import traceback

    27

import atexit

    28

#from twisted.python import log

    29

from twisted.internet import reactor, threads

    30

from autobahn.twisted import wamp

    31

from autobahn.twisted.websocket import WampWebSocketClientFactory, connectWS

    32

from autobahn.wamp import types

    33

from autobahn.wamp.exception import TransportLost

    34

from autobahn.wamp.serializer import MsgPackSerializer

    35

from threading import Thread, Event

    36

    37

_WampSession = None

    38

_WampConnection = None

    39

_WampSessionEvent = Event()

    40

    41

    42

class WampSession(wamp.ApplicationSession):

    43

    def onJoin(self, details):

    44

        global _WampSession, _WampSessionEvent

    45

        _WampSession = self

    46

        _WampSessionEvent.set()

    47

        print 'WAMP session joined for :', self.config.extra["ID"]

    48

    49

    def onLeave(self, details):

    50

        global _WampSession, _WampSessionEvent

    51

        _WampSessionEvent.clear()

    52

        _WampSession = None

    53

        print 'WAMP session left'

    54

    56

PLCObjDefaults = {

    57

    "StartPLC":          False,

    58

    "GetTraceVariables": ("Broken", None),

    59

    "GetPLCstatus":      ("Broken", None),

    60

    "RemoteExec":        (-1, "RemoteExec script failed!")

    61

}

    62

    63

    64

def WAMP_connector_factory(uri, confnodesroot):

    65

    """

    66

    WAMP://127.0.0.1:12345/path#realm#ID

    67

    WAMPS://127.0.0.1:12345/path#realm#ID

    68

    """

    69

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

    70

    urlpath, realm, ID = location.split('#')

    71

    urlprefix = {"WAMP":  "ws",

    72

                 "WAMPS": "wss"}[servicetype]

    73

    url = urlprefix+"://"+urlpath

    74

    75

    def RegisterWampClient():

    76

    77

        ## start logging to console

    78

        # log.startLogging(sys.stdout)

    79

    80

        # create a WAMP application session factory

    81

        component_config = types.ComponentConfig(

    82

            realm=realm,

    83

            extra={"ID": ID})

    84

        session_factory = wamp.ApplicationSessionFactory(

    85

            config=component_config)

    86

        session_factory.session = WampSession

    87

    88

        # create a WAMP-over-WebSocket transport client factory

    89

        transport_factory = WampWebSocketClientFactory(

    90

            session_factory,

    91

            url=url,

    92

            serializers=[MsgPackSerializer()],

    93

            debug=False,

    94

            debug_wamp=False)

    95

    96

        # start the client from a Twisted endpoint

    97

        conn = connectWS(transport_factory)

    98

        confnodesroot.logger.write(_("WAMP connecting to URL : %s\n") % url)

    99

        return conn

   100

   101

    AddToDoBeforeQuit = confnodesroot.AppFrame.AddToDoBeforeQuit

   102

    def ThreadProc():

   103

        global _WampConnection

   104

        _WampConnection = RegisterWampClient()

   105

        AddToDoBeforeQuit(reactor.stop)

   106

        reactor.run(installSignalHandlers=False)

   107

   108

    def WampSessionProcMapper(funcname):

   109

        wampfuncname = '.'.join((ID, funcname))

   110

        def catcher_func(*args, **kwargs):

   111

            global _WampSession

   112

            if _WampSession is not None:

   113

                try:

   114

                    return threads.blockingCallFromThread(

   115

                        reactor, _WampSession.call, wampfuncname,

   116

                        *args, **kwargs)

   117

                except TransportLost, e:

   118

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

   119

                    confnodesroot._SetConnector(None)

   120

                except Exception, e:

   121

                    errmess = traceback.format_exc()

   122

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

   123

                    print errmess

   124

                    #confnodesroot._SetConnector(None)

   125

            return PLCObjDefaults.get(funcname)

   126

        return catcher_func

   127

   128

    class WampPLCObjectProxy(object):

   129

        def __init__(self):

   130

            global _WampSessionEvent, _WampConnection

   131

            if not reactor.running:

   132

                Thread(target=ThreadProc).start()

   133

            else:

   134

                _WampConnection = threads.blockingCallFromThread(

   135

                    reactor, RegisterWampClient)

   136

            if not _WampSessionEvent.wait(5):

   137

                _WampConnection = stopConnecting()

   138

                raise Exception, _("WAMP connection timeout")

   139

   140

        def __del__(self):

   141

            global _WampConnection

   142

            _WampConnection.disconnect()

   143

            #

   144

            # reactor.stop()

   145

   146

        def __getattr__(self, attrName):

   147

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

   148

            if member is None:

   149

                member = WampSessionProcMapper(attrName)

   150

                self.__dict__[attrName] = member

   151

            return member

   152

   153

    # Try to get the proxy object

   154

    try:

   155

        return WampPLCObjectProxy()

   156

    except Exception, msg:

   157

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

   158

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

   159

        return None