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

#from twisted.python import log

    27

from autobahn.twisted import wamp

    28

from autobahn.twisted.websocket import WampWebSocketClientFactory, connectWS

    29

from twisted.internet.defer import inlineCallbacks

    30

from autobahn.wamp import types

    31

from autobahn.wamp.serializer import MsgPackSerializer

    32

from twisted.internet.protocol import ReconnectingClientFactory

    33

import json

    34

    35

_WampSession = None

    36

_PySrv = None

    37

    38

ExposedCalls = ["StartPLC",

    39

                "StopPLC",

    40

                "ForceReload",

    41

                "GetPLCstatus",

    42

                "NewPLC",

    43

                "MatchMD5",

    44

                "SetTraceVariablesList",

    45

                "GetTraceVariables",

    46

                "RemoteExec",

    47

                "GetLogMessage",

    48

                "ResetLogCount",

    49

                ]

    50

    51

SubscribedEvents = []

    52

    53

DoOnJoin = []

    54

    55

def GetCallee(name):

    56

    """ Get Callee or Subscriber corresponding to '.' spearated object path """

    57

    global _PySrv

    58

    names = name.split('.')

    59

    obj = _PySrv.plcobj

    60

    while names: obj = getattr(obj, names.pop(0))

    61

    return obj

    62

    63

class WampSession(wamp.ApplicationSession):

    64

    65

    @inlineCallbacks

    66

    def onJoin(self, details):

    67

        global _WampSession

    68

        _WampSession = self

    69

        ID = self.config.extra["ID"]

    70

        print 'WAMP session joined by :', ID

    71

        for name in ExposedCalls:

    72

            reg = yield self.register(GetCallee(name), '.'.join((ID,name)))

    73

    74

        for name in SubscribedEvents:

    75

            reg = yield self.subscribe(GetCallee(name), name)

    76

    77

        for func in DoOnJoin:

    78

            yield func(self)

    79

    80

    def onLeave(self, details):

    81

        global _WampSession

    82

        _WampSession = None

    83

        print 'WAMP session left'

    84

    85

class ReconnectingWampWebSocketClientFactory(WampWebSocketClientFactory, ReconnectingClientFactory):

    86

    def clientConnectionFailed(self, connector, reason):

    87

        print("WAMP Client connection failed .. retrying ..")

    88

        self.retry(connector)

    89

    def clientConnectionLost(self, connector, reason):

    90

        print("WAMP Client connection lost .. retrying ..")

    91

        self.retry(connector)

    92

    93

def LoadWampClientConf(wampconf):

    94

    95

    WSClientConf = json.load(open(wampconf))

    96

    return WSClientConf

    97

    98

def RegisterWampClient(wampconf):

    99

   100

    WSClientConf = LoadWampClientConf(wampconf)

   101

   102

    ## start logging to console

   103

    # log.startLogging(sys.stdout)

   104

   105

    # create a WAMP application session factory

   106

    component_config = types.ComponentConfig(

   107

        realm = WSClientConf["realm"],

   108

        extra = {"ID":WSClientConf["ID"]})

   109

    session_factory = wamp.ApplicationSessionFactory(

   110

        config = component_config)

   111

    session_factory.session = WampSession

   112

   113

    # create a WAMP-over-WebSocket transport client factory

   114

    transport_factory = ReconnectingWampWebSocketClientFactory(

   115

        session_factory,

   116

        url = WSClientConf["url"],

   117

        serializers = [MsgPackSerializer()],

   118

        debug = False,

   119

        debug_wamp = False)

   120

   121

    # start the client from a Twisted endpoint

   122

    conn = connectWS(transport_factory)

   123

    print "WAMP client connecting to :",WSClientConf["url"]

   124

    return conn

   125

   126

def GetSession():

   127

    global _WampSession

   128

    return _WampSession

   129

   130

def SetServer(pysrv):

   131

    global _PySrv

   132

    _PySrv = pysrv

   133