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

    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",

    28

                ]

    29

    30

def MakeCallee(name):

    31

    global _PySrv

    32

    def Callee(*args,**kwargs):

    33

        return getattr(_PySrv.plcobj, name)(*args,**kwargs)

    34

    return Callee

    35

    36

    37

class WampSession(wamp.ApplicationSession):

    38

    39

    @inlineCallbacks

    40

    def onJoin(self, details):

    41

        global _WampSession

    42

        _WampSession = self

    43

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

    44

        for name in ExposedCalls:

    45

            reg = yield self.register(MakeCallee(name), name)

    46

    47

    def onLeave(self, details):

    48

        global _WampSession

    49

        _WampSession = None

    50

        print 'WAMP session left'

    51

    60

def RegisterWampClient(wampconf):

    61

    62

    WSClientConf = json.load(open(wampconf))

    63

    64

    ## start logging to console

    65

    # log.startLogging(sys.stdout)

    66

    67

    # create a WAMP application session factory

    68

    component_config = types.ComponentConfig(

    69

        realm = WSClientConf["realm"],

    70

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

    71

    session_factory = wamp.ApplicationSessionFactory(

    72

        config = component_config)

    73

    session_factory.session = WampSession

    74

    75

    # create a WAMP-over-WebSocket transport client factory

    77

        session_factory,

    78

        url = WSClientConf["url"],

    79

        serializers = [MsgPackSerializer()],

    80

        debug = False,

    81

        debug_wamp = False)

    82

    83

    # start the client from a Twisted endpoint

    84

    conn = connectWS(transport_factory)

    85

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

    86

    return conn

    87

    88

def GetSession():

    89

    global _WampSession

    90

    return _WampSession

    91

    92

def SetServer(pysrv):

    93

    global _PySrv

    94

    _PySrv = pysrv

    95