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, traceback, atexit

    26

#from twisted.python import log

    27

from twisted.internet import reactor, threads

    28

from autobahn.twisted import wamp

    29

from autobahn.twisted.websocket import WampWebSocketClientFactory, connectWS

    30

from autobahn.wamp import types

    31

from autobahn.wamp.exception import TransportLost

    32

from autobahn.wamp.serializer import MsgPackSerializer

    33

from threading import Thread, Event

    34

    35

_WampSession = None

    36

_WampConnection = None

    37

_WampSessionEvent = Event()

    38

    39

class WampSession(wamp.ApplicationSession):

    40

    def onJoin(self, details):

    41

        global _WampSession, _WampSessionEvent

    42

        _WampSession = self

    43

        _WampSessionEvent.set()

    44

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

    45

    46

    def onLeave(self, details):

    47

        global _WampSession, _WampSessionEvent

    48

        _WampSessionEvent.clear()

    49

        _WampSession = None

    50

        print 'WAMP session left'

    51

    52

PLCObjDefaults = { "StartPLC": False,

    53

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

    54

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

    55

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

    56

    57

def WAMP_connector_factory(uri, confnodesroot):

    58

    """

    59

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

    60

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

    61

    """

    62

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

    63

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

    64

    urlprefix = {"WAMP":"ws",

    65

                 "WAMPS":"wss"}[servicetype]

    66

    url = urlprefix+"://"+urlpath

    67

    68

    def RegisterWampClient():

    69

    70

        ## start logging to console

    71

        # log.startLogging(sys.stdout)

    72

    73

        # create a WAMP application session factory

    74

        component_config = types.ComponentConfig(

    75

            realm = realm,

    76

            extra = {"ID":ID})

    77

        session_factory = wamp.ApplicationSessionFactory(

    78

            config = component_config)

    79

        session_factory.session = WampSession

    80

    81

        # create a WAMP-over-WebSocket transport client factory

    82

        transport_factory = WampWebSocketClientFactory(

    83

            session_factory,

    84

            url = url,

    85

            serializers = [MsgPackSerializer()],

    86

            debug = False,

    87

            debug_wamp = False)

    88

    89

        # start the client from a Twisted endpoint

    90

        conn = connectWS(transport_factory)

    91

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

    92

        return conn

    93

    94

    AddToDoBeforeQuit = confnodesroot.AppFrame.AddToDoBeforeQuit

    95

    def ThreadProc():

    96

        global _WampConnection

    97

        _WampConnection = RegisterWampClient()

    98

        AddToDoBeforeQuit(reactor.stop)

    99

        reactor.run(installSignalHandlers=False)

   100

   101

    def WampSessionProcMapper(funcname):

   102

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

   103

        def catcher_func(*args,**kwargs):

   104

            global _WampSession

   105

            if _WampSession is not None :

   106

                try:

   107

                    return threads.blockingCallFromThread(

   108

                        reactor, _WampSession.call, wampfuncname,

   109

                        *args,**kwargs)

   110

                except TransportLost, e:

   111

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

   112

                    confnodesroot._SetConnector(None)

   113

                except Exception,e:

   114

                    errmess = traceback.format_exc()

   115

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

   116

                    print errmess

   117

                    #confnodesroot._SetConnector(None)

   118

            return PLCObjDefaults.get(funcname)

   119

        return catcher_func

   120

   121

    class WampPLCObjectProxy(object):

   122

        def __init__(self):

   123

            global _WampSessionEvent, _WampConnection

   124

            if not reactor.running:

   125

                Thread(target=ThreadProc).start()

   126

            else:

   127

                _WampConnection = threads.blockingCallFromThread(

   128

                    reactor, RegisterWampClient)

   129

            if not _WampSessionEvent.wait(5):

   130

                _WampConnection = stopConnecting()

   131

                raise Exception, _("WAMP connection timeout")

   132

   133

        def __del__(self):

   134

            global _WampConnection

   135

            _WampConnection.disconnect()

   136

            #

   137

            # reactor.stop()

   138

   139

        def __getattr__(self, attrName):

   140

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

   141

            if member is None:

   142

                member = WampSessionProcMapper(attrName)

   143

                self.__dict__[attrName] = member

   144

            return member

   145

   146

    # Try to get the proxy object

   147

    try :

   148

        return WampPLCObjectProxy()

   149

    except Exception, msg:

   150

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

   151

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

   152

        return None

   153

   154

   155

   156