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

from __future__ import print_function

    27

import sys

    28

import traceback

    29

import atexit

    30

from threading import Thread, Event

    31

    32

from twisted.internet import reactor, threads

    33

from autobahn.twisted import wamp

    34

from autobahn.twisted.websocket import WampWebSocketClientFactory, connectWS

    35

from autobahn.wamp import types

    36

from autobahn.wamp.exception import TransportLost

    37

from autobahn.wamp.serializer import MsgPackSerializer

    38

    39

    40

_WampSession = None

    41

_WampConnection = None

    42

_WampSessionEvent = Event()

    43

    44

    45

class WampSession(wamp.ApplicationSession):

    46

    def onJoin(self, details):

    47

        global _WampSession

    48

        _WampSession = self

    49

        _WampSessionEvent.set()

    50

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

    51

    52

    def onLeave(self, details):

    53

        global _WampSession

    54

        _WampSessionEvent.clear()

    55

        _WampSession = None

    56

        print('WAMP session left')

    57

    58

    59

PLCObjDefaults = {

    60

    "StartPLC":          False,

    61

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

    62

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

    63

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

    64

}

    65

    66

    67

def WAMP_connector_factory(uri, confnodesroot):

    68

    """

    69

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

    70

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

    71

    """

    72

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

    73

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

    74

    urlprefix = {"WAMP":  "ws",

    75

                 "WAMPS": "wss"}[servicetype]

    76

    url = urlprefix+"://"+urlpath

    77

    78

    def RegisterWampClient():

    79

    80

        # start logging to console

    81

        # log.startLogging(sys.stdout)

    82

    83

        # create a WAMP application session factory

    84

        component_config = types.ComponentConfig(

    85

            realm=realm,

    86

            extra={"ID": ID})

    87

        session_factory = wamp.ApplicationSessionFactory(

    88

            config=component_config)

    89

        session_factory.session = WampSession

    90

    91

        # create a WAMP-over-WebSocket transport client factory

    92

        transport_factory = WampWebSocketClientFactory(

    93

            session_factory,

    94

            url=url,

    95

            serializers=[MsgPackSerializer()],

    96

            debug=False,

    97

            debug_wamp=False)

    98

    99

        # start the client from a Twisted endpoint

   100

        conn = connectWS(transport_factory)

   101

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

   102

        return conn

   103

   104

    AddToDoBeforeQuit = confnodesroot.AppFrame.AddToDoBeforeQuit

   105

   106

    def ThreadProc():

   107

        global _WampConnection

   108

        _WampConnection = RegisterWampClient()

   109

        AddToDoBeforeQuit(reactor.stop)

   110

        reactor.run(installSignalHandlers=False)

   111

   112

    def WampSessionProcMapper(funcname):

   113

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

   114

   115

        def catcher_func(*args, **kwargs):

   116

            if _WampSession is not None:

   117

                try:

   118

                    return threads.blockingCallFromThread(

   119

                        reactor, _WampSession.call, wampfuncname,

   120

                        *args, **kwargs)

   121

                except TransportLost:

   122

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

   123

                    confnodesroot._SetConnector(None)

   124

                except Exception:

   125

                    errmess = traceback.format_exc()

   126

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

   127

                    print(errmess)

   128

                    # confnodesroot._SetConnector(None)

   129

            return PLCObjDefaults.get(funcname)

   130

        return catcher_func

   131

   132

    class WampPLCObjectProxy(object):

   133

        def __init__(self):

   134

            global _WampConnection

   135

            if not reactor.running:

   136

                Thread(target=ThreadProc).start()

   137

            else:

   138

                _WampConnection = threads.blockingCallFromThread(

   139

                    reactor, RegisterWampClient)

   140

            if not _WampSessionEvent.wait(5):

   142

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

   143

   144

        def __del__(self):

   145

            _WampConnection.disconnect()

   146

            #

   147

            # reactor.stop()

   148

   149

        def __getattr__(self, attrName):

   150

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

   151

            if member is None:

   152

                member = WampSessionProcMapper(attrName)

   153

                self.__dict__[attrName] = member

   154

            return member

   155

   156

    # Try to get the proxy object

   157

    try:

   158

        return WampPLCObjectProxy()

   159

    except Exception:

   160

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

   161

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

   162

        return None