1

#!/usr/bin/env python

     2

# -*- coding: utf-8 -*-

     3

     4

# This file is part of Beremiz runtime.

     5

#

     6

# Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD

     7

#

     8

# See COPYING.Runtime file for copyrights details.

     9

#

    10

# This library is free software; you can redistribute it and/or

    11

# modify it under the terms of the GNU Lesser General Public

    12

# License as published by the Free Software Foundation; either

    13

# version 2.1 of the License, or (at your option) any later version.

    14

    15

# This library is distributed in the hope that it will be useful,

    16

# but WITHOUT ANY WARRANTY; without even the implied warranty of

    17

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

    18

# Lesser General Public License for more details.

    19

    20

# You should have received a copy of the GNU Lesser General Public

    21

# License along with this library; if not, write to the Free Software

    22

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA

    23

    24

    25

from __future__ import print_function

    26

import json

    27

    28

from autobahn.twisted import wamp

    29

from autobahn.twisted.websocket import WampWebSocketClientFactory, connectWS

    30

from autobahn.wamp import types

    31

from autobahn.wamp.serializer import MsgPackSerializer

    32

from twisted.internet.defer import inlineCallbacks

    33

from twisted.internet.protocol import ReconnectingClientFactory

    34

    35

    36

_WampSession = None

    37

_PySrv = None

    38

    52

    53

SubscribedEvents = []

    54

    55

DoOnJoin = []

    56

    57

    58

def GetCallee(name):

    59

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

    60

    names = name.split('.')

    61

    obj = _PySrv.plcobj

    62

    while names:

    63

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

    64

    return obj

    65

    66

    67

class WampSession(wamp.ApplicationSession):

    68

    69

    @inlineCallbacks

    70

    def onJoin(self, details):

    71

        global _WampSession

    72

        _WampSession = self

    73

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

    74

        print('WAMP session joined by :', ID)

    75

        for name in ExposedCalls:

    76

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

    77

    78

        for name in SubscribedEvents:

    79

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

    80

    81

        for func in DoOnJoin:

    82

            yield func(self)

    83

    84

    def onLeave(self, details):

    85

        global _WampSession

    86

        _WampSession = None

    87

        print('WAMP session left')

    88

    89

    90

class ReconnectingWampWebSocketClientFactory(WampWebSocketClientFactory, ReconnectingClientFactory):

    91

    def clientConnectionFailed(self, connector, reason):

    92

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

    93

        self.retry(connector)

    94

    95

    def clientConnectionLost(self, connector, reason):

    96

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

    97

        self.retry(connector)

    98

    99

   100

def LoadWampClientConf(wampconf):

   101

   102

    WSClientConf = json.load(open(wampconf))

   103

    return WSClientConf

   104

   105

   106

def RegisterWampClient(wampconf):

   107

   108

    WSClientConf = LoadWampClientConf(wampconf)

   109

   110

    # start logging to console

   111

    # log.startLogging(sys.stdout)

   112

   113

    # create a WAMP application session factory

   114

    component_config = types.ComponentConfig(

   115

        realm=WSClientConf["realm"],

   116

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

   117

    session_factory = wamp.ApplicationSessionFactory(

   118

        config=component_config)

   119

    session_factory.session = WampSession

   120

   121

    # create a WAMP-over-WebSocket transport client factory

   122

    transport_factory = ReconnectingWampWebSocketClientFactory(

   123

        session_factory,

   124

        url=WSClientConf["url"],

   125

        serializers=[MsgPackSerializer()],

   126

        debug=False,

   127

        debug_wamp=False)

   128

   129

    # start the client from a Twisted endpoint

   130

    conn = connectWS(transport_factory)

   131

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

   132

    return conn

   133

   134

   135

def GetSession():

   136

    return _WampSession

   137

   138

   139

def SetServer(pysrv):

   140

    global _PySrv

   141

    _PySrv = pysrv