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

class WampSession(wamp.ApplicationSession):

    42

    def onJoin(self, details):

    43

        global _WampSession, _WampSessionEvent

    44

        _WampSession = self

    45

        _WampSessionEvent.set()

    46

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

    47

    48

    def onLeave(self, details):

    49

        global _WampSession, _WampSessionEvent

    50

        _WampSessionEvent.clear()

    51

        _WampSession = None

    52

        print 'WAMP session left'

    53

    54

PLCObjDefaults = { "StartPLC": False,

    55

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

    56

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

    57

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

    58

    59

def WAMP_connector_factory(uri, confnodesroot):

    60

    """

    61

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

    62

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

    63

    """

    64

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

    65

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

    66

    urlprefix = {"WAMP":"ws",

    67

                 "WAMPS":"wss"}[servicetype]

    68

    url = urlprefix+"://"+urlpath

    69

    70

    def RegisterWampClient():

    71

    72

        ## start logging to console

    73

        # log.startLogging(sys.stdout)

    74

    75

        # create a WAMP application session factory

    76

        component_config = types.ComponentConfig(

    77

            realm = realm,

    78

            extra = {"ID":ID})

    79

        session_factory = wamp.ApplicationSessionFactory(

    80

            config = component_config)

    81

        session_factory.session = WampSession

    82

    83

        # create a WAMP-over-WebSocket transport client factory

    84

        transport_factory = WampWebSocketClientFactory(

    85

            session_factory,

    86

            url = url,

    87

            serializers = [MsgPackSerializer()],

    88

            debug = False,

    89

            debug_wamp = False)

    90

    91

        # start the client from a Twisted endpoint

    92

        conn = connectWS(transport_factory)

    94

        return conn

    95

    96

    AddToDoBeforeQuit = confnodesroot.AppFrame.AddToDoBeforeQuit

    97

    def ThreadProc():

    98

        global _WampConnection

    99

        _WampConnection = RegisterWampClient()

   100

        AddToDoBeforeQuit(reactor.stop)

   101

        reactor.run(installSignalHandlers=False)

   102

   103

    def WampSessionProcMapper(funcname):

   104

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

   105

        def catcher_func(*args,**kwargs):

   106

            global _WampSession

   107

            if _WampSession is not None :

   108

                try:

   109

                    return threads.blockingCallFromThread(

   110

                        reactor, _WampSession.call, wampfuncname,

   111

                        *args,**kwargs)

   112

                except TransportLost, e:

   113

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

   114

                    confnodesroot._SetConnector(None)

   115

                except Exception,e:

   116

                    errmess = traceback.format_exc()

   117

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

   118

                    print errmess

   119

                    #confnodesroot._SetConnector(None)

   120

            return PLCObjDefaults.get(funcname)

   121

        return catcher_func

   122

   123

    class WampPLCObjectProxy(object):

   124

        def __init__(self):

   125

            global _WampSessionEvent, _WampConnection

   126

            if not reactor.running:

   127

                Thread(target=ThreadProc).start()

   128

            else:

   129

                _WampConnection = threads.blockingCallFromThread(

   130

                    reactor, RegisterWampClient)

   131

            if not _WampSessionEvent.wait(5):

   132

                _WampConnection = stopConnecting()

   133

                raise Exception, _("WAMP connection timeout")

   134

   135

        def __del__(self):

   136

            global _WampConnection

   137

            _WampConnection.disconnect()

   138

            #

   139

            # reactor.stop()

   140

   141

        def __getattr__(self, attrName):

   142

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

   143

            if member is None:

   144

                member = WampSessionProcMapper(attrName)

   145

                self.__dict__[attrName] = member

   146

            return member

   147

   148

    # Try to get the proxy object

   149

    try :

   150

        return WampPLCObjectProxy()

   151

    except Exception, msg:

   153

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

   154

        return None

   155

   156

   157

   158