1

#!/usr/bin/env python

     2

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

     3

     4

import sys

     5

#from twisted.python import log

     6

from autobahn.twisted import wamp

     7

from autobahn.twisted.websocket import WampWebSocketClientFactory, connectWS

     8

from twisted.internet.defer import inlineCallbacks

     9

from autobahn.wamp import types

    10

from autobahn.wamp.serializer import MsgPackSerializer

    11

from twisted.internet.protocol import ReconnectingClientFactory

    12

import json

    13

    14

_WampSession = None

    15

_PySrv = None

    16

    17

ExposedCalls = ["StartPLC",

    18

                "StopPLC",

    19

                "ForceReload",

    20

                "GetPLCstatus",

    21

                "NewPLC",

    22

                "MatchMD5",

    23

                "SetTraceVariablesList",

    24

                "GetTraceVariables",

    25

                "RemoteExec",

    26

                "GetLogMessage",

    27

                "ResetLogCount",

    28

                ]

    29

    30

SubscribedEvents = []

    31

    32

DoOnJoin = []

    33

    34

def GetCallee(name):

    35

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

    36

    global _PySrv

    37

    names = name.split('.')

    38

    obj = _PySrv.plcobj

    39

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

    40

    return obj

    41

    42

class WampSession(wamp.ApplicationSession):

    43

    44

    @inlineCallbacks

    45

    def onJoin(self, details):

    46

        global _WampSession

    47

        _WampSession = self

    48

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

    49

        print 'WAMP session joined by :', ID

    50

        for name in ExposedCalls:

    51

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

    52

    53

        for name in SubscribedEvents:

    54

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

    55

    56

        for func in DoOnJoin:

    57

            yield func(self)

    58

    59

    def onLeave(self, details):

    60

        global _WampSession

    61

        _WampSession = None

    62

        print 'WAMP session left'

    63

    64

class ReconnectingWampWebSocketClientFactory(WampWebSocketClientFactory, ReconnectingClientFactory):

    65

    def clientConnectionFailed(self, connector, reason):

    66

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

    67

        self.retry(connector)

    68

    def clientConnectionLost(self, connector, reason):

    69

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

    70

        self.retry(connector)

    71

    72

def LoadWampClientConf(wampconf):

    73

    74

    WSClientConf = json.load(open(wampconf))

    75

    return WSClientConf

    76

    77

def RegisterWampClient(wampconf):

    78

    79

    WSClientConf = LoadWampClientConf(wampconf)

    80

    81

    ## start logging to console

    82

    # log.startLogging(sys.stdout)

    83

    84

    # create a WAMP application session factory

    85

    component_config = types.ComponentConfig(

    86

        realm = WSClientConf["realm"],

    87

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

    88

    session_factory = wamp.ApplicationSessionFactory(

    89

        config = component_config)

    90

    session_factory.session = WampSession

    91

    92

    # create a WAMP-over-WebSocket transport client factory

    93

    transport_factory = ReconnectingWampWebSocketClientFactory(

    94

        session_factory,

    95

        url = WSClientConf["url"],

    96

        serializers = [MsgPackSerializer()],

    97

        debug = False,

    98

        debug_wamp = False)

    99

   100

    # start the client from a Twisted endpoint

   101

    conn = connectWS(transport_factory)

   102

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

   103

    return conn

   104

   105

def GetSession():

   106

    global _WampSession

   107

    return _WampSession

   108

   109

def SetServer(pysrv):

   110

    global _PySrv

   111

    _PySrv = pysrv

   112