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

    26

import sys

    27

import traceback

    28

from functools import partial

    29

from threading import Thread, Event

    30

    31

from twisted.internet import reactor, threads

    32

from autobahn.twisted import wamp

    33

from autobahn.twisted.websocket import WampWebSocketClientFactory, connectWS

    34

from autobahn.wamp import types

    35

from autobahn.wamp.exception import TransportLost

    36

from autobahn.wamp.serializer import MsgPackSerializer

    37

    38

    39

_WampSession = None

    40

_WampConnection = None

    41

_WampSessionEvent = Event()

    42

    43

    44

class WampSession(wamp.ApplicationSession):

    45

    def onJoin(self, details):

    46

        global _WampSession

    47

        _WampSession = self

    48

        _WampSessionEvent.set()

    49

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

    50

    51

    def onLeave(self, details):

    52

        global _WampSession

    53

        _WampSessionEvent.clear()

    54

        _WampSession = None

    55

        print('WAMP session left')

    56

    57

    58

def _WAMP_connector_factory(cls, uri, confnodesroot):

    59

    """

    60

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

    61

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

    62

    """

    63

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

    64

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

    65

    urlprefix = {"WAMP":  "ws",

    66

                 "WAMPS": "wss"}[scheme]

    67

    url = urlprefix+"://"+urlpath

    68

    69

    def RegisterWampClient():

    70

    71

        # start logging to console

    72

        # log.startLogging(sys.stdout)

    73

    74

        # create a WAMP application session factory

    75

        component_config = types.ComponentConfig(

    76

            realm=str(realm),

    77

            extra={"ID": ID})

    78

        session_factory = wamp.ApplicationSessionFactory(

    79

            config=component_config)

    80

        session_factory.session = cls

    81

    82

        # create a WAMP-over-WebSocket transport client factory

    83

        transport_factory = WampWebSocketClientFactory(

    84

            session_factory,

    85

            url=url,

    86

            serializers=[MsgPackSerializer()])

    87

    88

        # start the client from a Twisted endpoint

    89

        conn = connectWS(transport_factory)

    90

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

    91

        return conn

    92

    93

    AddToDoBeforeQuit = confnodesroot.AppFrame.AddToDoBeforeQuit

    94

    95

    def ThreadProc():

    96

        global _WampConnection

    97

        _WampConnection = RegisterWampClient()

    98

        AddToDoBeforeQuit(reactor.stop)

    99

        reactor.run(installSignalHandlers=False)

   100

   101

    class WampPLCObjectProxy(object):

   102

        def __init__(self):

   103

            global _WampConnection

   104

            if not reactor.running:

   105

                Thread(target=ThreadProc).start()

   106

            else:

   107

                _WampConnection = threads.blockingCallFromThread(

   108

                    reactor, RegisterWampClient)

   109

            if not _WampSessionEvent.wait(5):

   110

                _WampConnection.stopConnecting()

   111

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

   112

   113

        def __del__(self):

   114

            _WampConnection.disconnect()

   115

            #

   116

            # reactor.stop()

   117

   118

        def WampSessionProcMapper(self, funcname):

   119

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

   120

   121

            def catcher_func(*args, **kwargs):

   122

                if _WampSession is not None:

   123

                    try:

   124

                        return threads.blockingCallFromThread(

   125

                            reactor, _WampSession.call, wampfuncname,

   126

                            *args, **kwargs)

   127

                    except TransportLost:

   128

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

   129

                        confnodesroot._SetConnector(None)

   130

                    except Exception:

   131

                        errmess = traceback.format_exc()

   132

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

   133

                        print(errmess)

   134

                        # confnodesroot._SetConnector(None)

   135

                return self.PLCObjDefaults.get(funcname)

   136

            return catcher_func

   137

   138

        def __getattr__(self, attrName):

   139

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

   140

            if member is None:

   141

                member = self.WampSessionProcMapper(attrName)

   142

                self.__dict__[attrName] = member

   143

            return member

   144

   145

    # TODO : GetPLCID()

   146

    # TODO : PSK.UpdateID()

   147

   148

    return WampPLCObjectProxy()

   149

   150

   151

WAMP_connector_factory = partial(_WAMP_connector_factory, WampSession)