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.internet import reactor, threads

    29

from autobahn.twisted import wamp

    30

from autobahn.twisted.websocket import WampWebSocketClientFactory, connectWS

    31

from autobahn.wamp import types

    32

from autobahn.wamp.exception import TransportLost

    33

from autobahn.wamp.serializer import MsgPackSerializer

    34

from threading import Thread, Event

    35

    36

_WampSession = None

    37

_WampConnection = None

    38

_WampSessionEvent = Event()

    39

    40

    41

class WampSession(wamp.ApplicationSession):

    42

    def onJoin(self, details):

    43

        global _WampSession, _WampSessionEvent

    44

        _WampSession = self

    45

        _WampSessionEvent.set()

    46

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

    47

    48

    def onLeave(self, details):

    49

        global _WampSession, _WampSessionEvent

    50

        _WampSessionEvent.clear()

    51

        _WampSession = None

    52

        print 'WAMP session left'

    53

    54

    55

PLCObjDefaults = {

    56

    "StartPLC":          False,

    57

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

    58

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

    59

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

    60

}

    61

    62

    63

def WAMP_connector_factory(uri, confnodesroot):

    64

    """

    65

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

    66

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

    67

    """

    68

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

    69

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

    70

    urlprefix = {"WAMP":  "ws",

    71

                 "WAMPS": "wss"}[servicetype]

    72

    url = urlprefix+"://"+urlpath

    73

    74

    def RegisterWampClient():

    75

    76

        # start logging to console

    77

        # log.startLogging(sys.stdout)

    78

    79

        # create a WAMP application session factory

    80

        component_config = types.ComponentConfig(

    81

            realm=realm,

    82

            extra={"ID": ID})

    83

        session_factory = wamp.ApplicationSessionFactory(

    84

            config=component_config)

    85

        session_factory.session = WampSession

    86

    87

        # create a WAMP-over-WebSocket transport client factory

    88

        transport_factory = WampWebSocketClientFactory(

    89

            session_factory,

    90

            url=url,

    91

            serializers=[MsgPackSerializer()],

    92

            debug=False,

    93

            debug_wamp=False)

    94

    95

        # start the client from a Twisted endpoint

    96

        conn = connectWS(transport_factory)

    97

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

    98

        return conn

    99

   100

    AddToDoBeforeQuit = confnodesroot.AppFrame.AddToDoBeforeQuit

   101

   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

   111

        def catcher_func(*args, **kwargs):

   112

            global _WampSession

   113

            if _WampSession is not None:

   114

                try:

   115

                    return threads.blockingCallFromThread(

   116

                        reactor, _WampSession.call, wampfuncname,

   117

                        *args, **kwargs)

   118

                except TransportLost, e:

   119

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

   120

                    confnodesroot._SetConnector(None)

   121

                except Exception, e:

   122

                    errmess = traceback.format_exc()

   123

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

   124

                    print errmess

   125

                    # confnodesroot._SetConnector(None)

   126

            return PLCObjDefaults.get(funcname)

   127

        return catcher_func

   128

   129

    class WampPLCObjectProxy(object):

   130

        def __init__(self):

   131

            global _WampSessionEvent, _WampConnection

   132

            if not reactor.running:

   133

                Thread(target=ThreadProc).start()

   134

            else:

   135

                _WampConnection = threads.blockingCallFromThread(

   136

                    reactor, RegisterWampClient)

   137

            if not _WampSessionEvent.wait(5):

   138

                _WampConnection = stopConnecting()

   139

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

   140

   141

        def __del__(self):

   142

            global _WampConnection

   143

            _WampConnection.disconnect()

   144

            #

   145

            # reactor.stop()

   146

   147

        def __getattr__(self, attrName):

   148

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

   149

            if member is None:

   150

                member = WampSessionProcMapper(attrName)

   151

                self.__dict__[attrName] = member

   152

            return member

   153

   154

    # Try to get the proxy object

   155

    try:

   156

        return WampPLCObjectProxy()

   157

    except Exception, msg:

   158

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

   159

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

   160

        return None