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

PLCObjDefaults = { "StartPLC":          False,

    56

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

    57

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

    58

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

    59

    60

    61

def WAMP_connector_factory(uri, confnodesroot):

    62

    """

    63

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

    64

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

    65

    """

    66

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

    67

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

    68

    urlprefix = {"WAMP":  "ws",

    69

                 "WAMPS": "wss"}[servicetype]

    70

    url = urlprefix+"://"+urlpath

    71

    72

    def RegisterWampClient():

    73

    74

        ## start logging to console

    75

        # log.startLogging(sys.stdout)

    76

    77

        # create a WAMP application session factory

    78

        component_config = types.ComponentConfig(

    79

            realm = realm,

    80

            extra = {"ID": ID})

    81

        session_factory = wamp.ApplicationSessionFactory(

    82

            config = component_config)

    83

        session_factory.session = WampSession

    84

    85

        # create a WAMP-over-WebSocket transport client factory

    86

        transport_factory = WampWebSocketClientFactory(

    87

            session_factory,

    88

            url = url,

    89

            serializers = [MsgPackSerializer()],

    90

            debug = False,

    91

            debug_wamp = False)

    92

    93

        # start the client from a Twisted endpoint

    94

        conn = connectWS(transport_factory)

    95

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

    96

        return conn

    97

    98

    AddToDoBeforeQuit = confnodesroot.AppFrame.AddToDoBeforeQuit

    99

    def ThreadProc():

   100

        global _WampConnection

   101

        _WampConnection = RegisterWampClient()

   102

        AddToDoBeforeQuit(reactor.stop)

   103

        reactor.run(installSignalHandlers=False)

   104

   105

    def WampSessionProcMapper(funcname):

   106

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

   107

        def catcher_func(*args, **kwargs):

   108

            global _WampSession

   109

            if _WampSession is not None:

   110

                try:

   111

                    return threads.blockingCallFromThread(

   112

                        reactor, _WampSession.call, wampfuncname,

   113

                        *args, **kwargs)

   114

                except TransportLost, e:

   115

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

   116

                    confnodesroot._SetConnector(None)

   117

                except Exception, e:

   118

                    errmess = traceback.format_exc()

   119

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

   120

                    print errmess

   121

                    #confnodesroot._SetConnector(None)

   122

            return PLCObjDefaults.get(funcname)

   123

        return catcher_func

   124

   125

    class WampPLCObjectProxy(object):

   126

        def __init__(self):

   127

            global _WampSessionEvent, _WampConnection

   128

            if not reactor.running:

   129

                Thread(target=ThreadProc).start()

   130

            else:

   131

                _WampConnection = threads.blockingCallFromThread(

   132

                    reactor, RegisterWampClient)

   133

            if not _WampSessionEvent.wait(5):

   134

                _WampConnection = stopConnecting()

   135

                raise Exception, _("WAMP connection timeout")

   136

   137

        def __del__(self):

   138

            global _WampConnection

   139

            _WampConnection.disconnect()

   140

            #

   141

            # reactor.stop()

   142

   143

        def __getattr__(self, attrName):

   144

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

   145

            if member is None:

   146

                member = WampSessionProcMapper(attrName)

   147

                self.__dict__[attrName] = member

   148

            return member

   149

   150

    # Try to get the proxy object

   151

    try:

   152

        return WampPLCObjectProxy()

   153

    except Exception, msg:

   154

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

   155

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

   156

        return None

   157

   158

   159

   160