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

    27

from __future__ import print_function

    28

import sys

    29

import traceback

    30

import atexit

    31

from threading import Thread, Event

    32

    33

from twisted.internet import reactor, threads

    34

from autobahn.twisted import wamp

    35

from autobahn.twisted.websocket import WampWebSocketClientFactory, connectWS

    36

from autobahn.wamp import types

    37

from autobahn.wamp.exception import TransportLost

    38

from autobahn.wamp.serializer import MsgPackSerializer

    39

    40

    41

_WampSession = None

    42

_WampConnection = None

    43

_WampSessionEvent = Event()

    44

    45

    46

class WampSession(wamp.ApplicationSession):

    47

    def onJoin(self, details):

    48

        global _WampSession

    49

        _WampSession = self

    50

        _WampSessionEvent.set()

    51

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

    52

    53

    def onLeave(self, details):

    54

        global _WampSession

    55

        _WampSessionEvent.clear()

    56

        _WampSession = None

    57

        print('WAMP session left')

    58

    59

    60

PLCObjDefaults = {

    61

    "StartPLC":          False,

    62

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

    63

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

    64

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

    65

}

    66

    67

    68

def WAMP_connector_factory(uri, confnodesroot):

    69

    """

    70

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

    71

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

    72

    """

    73

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

    74

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

    75

    urlprefix = {"WAMP":  "ws",

    76

                 "WAMPS": "wss"}[servicetype]

    77

    url = urlprefix+"://"+urlpath

    78

    79

    def RegisterWampClient():

    80

    81

        # start logging to console

    82

        # log.startLogging(sys.stdout)

    83

    84

        # create a WAMP application session factory

    85

        component_config = types.ComponentConfig(

    86

            realm=realm,

    87

            extra={"ID": ID})

    88

        session_factory = wamp.ApplicationSessionFactory(

    89

            config=component_config)

    90

        session_factory.session = WampSession

    91

    92

        # create a WAMP-over-WebSocket transport client factory

    93

        transport_factory = WampWebSocketClientFactory(

    94

            session_factory,

    95

            url=url,

    96

            serializers=[MsgPackSerializer()],

    97

            debug=False,

    98

            debug_wamp=False)

    99

   100

        # start the client from a Twisted endpoint

   101

        conn = connectWS(transport_factory)

   102

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

   103

        return conn

   104

   105

    AddToDoBeforeQuit = confnodesroot.AppFrame.AddToDoBeforeQuit

   106

   107

    def ThreadProc():

   108

        global _WampConnection

   109

        _WampConnection = RegisterWampClient()

   110

        AddToDoBeforeQuit(reactor.stop)

   111

        reactor.run(installSignalHandlers=False)

   112

   113

    def WampSessionProcMapper(funcname):

   114

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

   115

   116

        def catcher_func(*args, **kwargs):

   117

            if _WampSession is not None:

   118

                try:

   119

                    return threads.blockingCallFromThread(

   120

                        reactor, _WampSession.call, wampfuncname,

   121

                        *args, **kwargs)

   122

                except TransportLost:

   123

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

   124

                    confnodesroot._SetConnector(None)

   125

                except Exception:

   126

                    errmess = traceback.format_exc()

   127

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

   128

                    print(errmess)

   129

                    # confnodesroot._SetConnector(None)

   130

            return PLCObjDefaults.get(funcname)

   131

        return catcher_func

   132

   133

    class WampPLCObjectProxy(object):

   134

        def __init__(self):

   135

            global _WampConnection

   136

            if not reactor.running:

   137

                Thread(target=ThreadProc).start()

   138

            else:

   139

                _WampConnection = threads.blockingCallFromThread(

   140

                    reactor, RegisterWampClient)

   141

            if not _WampSessionEvent.wait(5):

   142

                _WampConnection.stopConnecting()

   143

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

   144

   145

        def __del__(self):

   146

            _WampConnection.disconnect()

   147

            #

   148

            # reactor.stop()

   149

   150

        def __getattr__(self, attrName):

   151

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

   152

            if member is None:

   153

                member = WampSessionProcMapper(attrName)

   154

                self.__dict__[attrName] = member

   155

            return member

   156

   157

    # Try to get the proxy object

   158

    try:

   159

        return WampPLCObjectProxy()

   160

    except Exception:

   161

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

   162

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

   163

        return None