1

#!/usr/bin/env python

     2

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

     3

     4

import sys

     5

from twisted.python import log

     6

     7

from twisted.internet import reactor, ssl

     8

from autobahn.twisted import wamp

     9

from autobahn.twisted.websocket import WampWebSocketClientFactory, connectWS

    10

from autobahn.wamp import types

    11

import json

    12

    13

_WampSession = None

    14

_PySrv = None

    15

    16

class WampSession(wamp.ApplicationSession):

    17

    def onJoin(self, details):

    18

        global _WampSession

    19

        _WampSession = self

    20

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

    21

    22

    def onLeave(self, details):

    23

        global _WampSession

    24

        _WampSession = None

    25

        print 'WAMP session left'

    26

    27

    28

def RegisterWampClient(wampconf):

    29

    30

    WSClientConf = json.load(open(wampconf))

    31

    32

    ## TODO log to PLC console instead

    33

    ## 0) start logging to console

    34

    log.startLogging(sys.stdout)

    35

    36

    ## 1) create a WAMP application session factory

    37

    component_config = types.ComponentConfig(

    38

        realm = WSClientConf["realm"],

    39

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

    40

    session_factory = wamp.ApplicationSessionFactory(

    41

        config = component_config)

    42

    session_factory.session = WampSession

    43

    44

    ## TODO select optimum serializer for passing session lists

    45

    ## optional: use specific set of serializers

    46

    #from autobahn.wamp.serializer import *

    47

    #serializers = []

    48

    ##serializers.append(JsonSerializer(batched = True))

    49

    ##serializers.append(MsgPackSerializer(batched = True))

    50

    #serializers.append(JsonSerializer())

    51

    ##serializers.append(MsgPackSerializer())

    52

    serializers = None

    53

    54

    ## 2) create a WAMP-over-WebSocket transport client factory

    55

    transport_factory = WampWebSocketClientFactory(

    56

        session_factory,

    57

        url = WSClientConf["url"],

    58

        serializers = serializers,

    59

        debug = False,

    60

        debug_wamp = False)

    61

    62

    ## 3) start the client from a Twisted endpoint

    63

    conn = connectWS(transport_factory)

    64

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

    65

    return conn

    66

    67

def GetSession():

    68

    global _WampSession

    69

    return _WampSession

    70

    71

def SetServer(pysrv):

    72

    global _PySrv

    73

    _PySrv = pysrv

    74