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

    55

    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

   103

    def ThreadProc():

   104

        global _WampConnection

   105

        _WampConnection = RegisterWampClient()

   106

        AddToDoBeforeQuit(reactor.stop)

   107

        reactor.run(installSignalHandlers=False)

   108

   109

    def WampSessionProcMapper(funcname):

   110

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

   111

   112

        def catcher_func(*args, **kwargs):

   113

            global _WampSession

   114

            if _WampSession is not None:

   115

                try:

   116

                    return threads.blockingCallFromThread(

   117

                        reactor, _WampSession.call, wampfuncname,

   118

                        *args, **kwargs)

   119

                except TransportLost, e:

   120

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

   121

                    confnodesroot._SetConnector(None)

   122

                except Exception, e:

   123

                    errmess = traceback.format_exc()

   124

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

   125

                    print errmess

   126

                    #confnodesroot._SetConnector(None)

   127

            return PLCObjDefaults.get(funcname)

   128

        return catcher_func

   129

   130

    class WampPLCObjectProxy(object):

   131

        def __init__(self):

   132

            global _WampSessionEvent, _WampConnection

   133

            if not reactor.running:

   134

                Thread(target=ThreadProc).start()

   135

            else:

   136

                _WampConnection = threads.blockingCallFromThread(

   137

                    reactor, RegisterWampClient)

   138

            if not _WampSessionEvent.wait(5):

   139

                _WampConnection = stopConnecting()

   140

                raise Exception(_("WAMP connection timeout"))

   141

   142

        def __del__(self):

   143

            global _WampConnection

   144

            _WampConnection.disconnect()

   145

            #

   146

            # reactor.stop()

   147

   148

        def __getattr__(self, attrName):

   149

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

   150

            if member is None:

   151

                member = WampSessionProcMapper(attrName)

   152

                self.__dict__[attrName] = member

   153

            return member

   154

   155

    # Try to get the proxy object

   156

    try:

   157

        return WampPLCObjectProxy()

   158

    except Exception, msg:

   159

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

   160

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

   161

        return None