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

    31

    global _PySrv

    36

    37

class WampSession(wamp.ApplicationSession):

    38

    39

    @inlineCallbacks

    40

    def onJoin(self, details):

    41

        global _WampSession

    42

        _WampSession = self

    43

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

    44

        print 'WAMP session joined by :', ID

    45

        for name in ExposedCalls:

    47

    48

    def onLeave(self, details):

    49

        global _WampSession

    50

        _WampSession = None

    51

        print 'WAMP session left'

    52

    53

class ReconnectingWampWebSocketClientFactory(WampWebSocketClientFactory, ReconnectingClientFactory):

    54

    def clientConnectionFailed(self, connector, reason):

    55

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

    56

        self.retry(connector)

    57

    def clientConnectionLost(self, connector, reason):

    58

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

    59

        self.retry(connector)

    60

    61

def RegisterWampClient(wampconf):

    62

    63

    WSClientConf = json.load(open(wampconf))

    64

    65

    ## start logging to console

    66

    # log.startLogging(sys.stdout)

    67

    68

    # create a WAMP application session factory

    69

    component_config = types.ComponentConfig(

    70

        realm = WSClientConf["realm"],

    71

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

    72

    session_factory = wamp.ApplicationSessionFactory(

    73

        config = component_config)

    74

    session_factory.session = WampSession

    75

    76

    # create a WAMP-over-WebSocket transport client factory

    77

    transport_factory = ReconnectingWampWebSocketClientFactory(

    78

        session_factory,

    79

        url = WSClientConf["url"],

    80

        serializers = [MsgPackSerializer()],

    81

        debug = False,

    82

        debug_wamp = False)

    83

    84

    # start the client from a Twisted endpoint

    85

    conn = connectWS(transport_factory)

    86

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

    87

    return conn

    88

    89

def GetSession():

    90

    global _WampSession

    91

    return _WampSession

    92

    93

def SetServer(pysrv):

    94

    global _PySrv

    95

    _PySrv = pysrv

    96