Edouard@1440: #!/usr/bin/env python Edouard@1440: # -*- coding: utf-8 -*- andrej@1571: andrej@1571: # This file is part of Beremiz, a Integrated Development Environment for andrej@1571: # programming IEC 61131-3 automates supporting plcopen standard and CanFestival. Edouard@1440: # andrej@1571: # Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD Edouard@1440: # andrej@1571: # See COPYING file for copyrights details. Edouard@1440: # andrej@1571: # This program is free software; you can redistribute it and/or andrej@1571: # modify it under the terms of the GNU General Public License andrej@1571: # as published by the Free Software Foundation; either version 2 andrej@1571: # of the License, or (at your option) any later version. Edouard@1440: # andrej@1571: # This program is distributed in the hope that it will be useful, andrej@1571: # but WITHOUT ANY WARRANTY; without even the implied warranty of andrej@1571: # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the andrej@1571: # GNU General Public License for more details. Edouard@1440: # andrej@1571: # You should have received a copy of the GNU General Public License andrej@1571: # along with this program; if not, write to the Free Software andrej@1571: # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. Edouard@1440: andrej@1732: import sys andrej@1732: import traceback andrej@1732: import atexit Edouard@1440: from twisted.internet import reactor, threads Edouard@1440: from autobahn.twisted import wamp Edouard@1440: from autobahn.twisted.websocket import WampWebSocketClientFactory, connectWS Edouard@1440: from autobahn.wamp import types Edouard@1440: from autobahn.wamp.exception import TransportLost Edouard@1440: from autobahn.wamp.serializer import MsgPackSerializer Edouard@1440: from threading import Thread, Event Edouard@1440: Edouard@1440: _WampSession = None Edouard@1441: _WampConnection = None Edouard@1440: _WampSessionEvent = Event() Edouard@1440: andrej@1736: Edouard@1440: class WampSession(wamp.ApplicationSession): Edouard@1440: def onJoin(self, details): Edouard@1441: global _WampSession, _WampSessionEvent Edouard@1440: _WampSession = self Edouard@1440: _WampSessionEvent.set() Edouard@1440: print 'WAMP session joined for :', self.config.extra["ID"] Edouard@1440: Edouard@1440: def onLeave(self, details): Edouard@1441: global _WampSession, _WampSessionEvent Edouard@1440: _WampSessionEvent.clear() Edouard@1440: _WampSession = None Edouard@1440: print 'WAMP session left' Edouard@1440: andrej@1749: andrej@1747: PLCObjDefaults = { andrej@1747: "StartPLC": False, andrej@1747: "GetTraceVariables": ("Broken", None), andrej@1747: "GetPLCstatus": ("Broken", None), andrej@1747: "RemoteExec": (-1, "RemoteExec script failed!") andrej@1747: } Edouard@1440: andrej@1736: Edouard@1440: def WAMP_connector_factory(uri, confnodesroot): Edouard@1440: """ Edouard@1440: WAMP://127.0.0.1:12345/path#realm#ID Edouard@1440: WAMPS://127.0.0.1:12345/path#realm#ID Edouard@1440: """ Edouard@1440: servicetype, location = uri.split("://") Edouard@1440: urlpath, realm, ID = location.split('#') andrej@1740: urlprefix = {"WAMP": "ws", andrej@1740: "WAMPS": "wss"}[servicetype] Edouard@1440: url = urlprefix+"://"+urlpath Edouard@1440: Edouard@1440: def RegisterWampClient(): Edouard@1440: andrej@1753: # start logging to console Edouard@1440: # log.startLogging(sys.stdout) Edouard@1440: Edouard@1440: # create a WAMP application session factory Edouard@1440: component_config = types.ComponentConfig( andrej@1744: realm=realm, andrej@1744: extra={"ID": ID}) Edouard@1440: session_factory = wamp.ApplicationSessionFactory( andrej@1744: config=component_config) Edouard@1440: session_factory.session = WampSession Edouard@1440: Edouard@1440: # create a WAMP-over-WebSocket transport client factory Edouard@1440: transport_factory = WampWebSocketClientFactory( Edouard@1440: session_factory, andrej@1744: url=url, andrej@1744: serializers=[MsgPackSerializer()], andrej@1744: debug=False, andrej@1744: debug_wamp=False) Edouard@1440: Edouard@1440: # start the client from a Twisted endpoint Edouard@1440: conn = connectWS(transport_factory) andrej@1734: confnodesroot.logger.write(_("WAMP connecting to URL : %s\n") % url) Edouard@1440: return conn Edouard@1440: Edouard@1441: AddToDoBeforeQuit = confnodesroot.AppFrame.AddToDoBeforeQuit andrej@1750: Edouard@1441: def ThreadProc(): Edouard@1441: global _WampConnection Edouard@1441: _WampConnection = RegisterWampClient() Edouard@1441: AddToDoBeforeQuit(reactor.stop) Edouard@1441: reactor.run(installSignalHandlers=False) Edouard@1441: Edouard@1440: def WampSessionProcMapper(funcname): andrej@1740: wampfuncname = '.'.join((ID, funcname)) andrej@1750: andrej@1740: def catcher_func(*args, **kwargs): Edouard@1441: global _WampSession andrej@1739: if _WampSession is not None: Edouard@1440: try: Edouard@1440: return threads.blockingCallFromThread( Edouard@1443: reactor, _WampSession.call, wampfuncname, andrej@1740: *args, **kwargs) Edouard@1440: except TransportLost, e: andrej@1595: confnodesroot.logger.write_error(_("Connection lost!\n")) Edouard@1440: confnodesroot._SetConnector(None) andrej@1740: except Exception, e: Edouard@1440: errmess = traceback.format_exc() Edouard@1440: confnodesroot.logger.write_error(errmess+"\n") Edouard@1440: print errmess andrej@1782: # confnodesroot._SetConnector(None) Edouard@1440: return PLCObjDefaults.get(funcname) Edouard@1440: return catcher_func Edouard@1440: Edouard@1440: class WampPLCObjectProxy(object): Edouard@1440: def __init__(self): Edouard@1441: global _WampSessionEvent, _WampConnection Edouard@1440: if not reactor.running: Edouard@1440: Thread(target=ThreadProc).start() Edouard@1440: else: Edouard@1441: _WampConnection = threads.blockingCallFromThread( Edouard@1440: reactor, RegisterWampClient) Edouard@1440: if not _WampSessionEvent.wait(5): Edouard@1441: _WampConnection = stopConnecting() andrej@1765: raise Exception(_("WAMP connection timeout")) Edouard@1440: Edouard@1440: def __del__(self): Edouard@1441: global _WampConnection Edouard@1441: _WampConnection.disconnect() Edouard@1441: # Edouard@1441: # reactor.stop() Edouard@1440: Edouard@1440: def __getattr__(self, attrName): Edouard@1440: member = self.__dict__.get(attrName, None) Edouard@1440: if member is None: Edouard@1440: member = WampSessionProcMapper(attrName) Edouard@1440: self.__dict__[attrName] = member Edouard@1440: return member Edouard@1440: Edouard@1440: # Try to get the proxy object andrej@1739: try: Edouard@1440: return WampPLCObjectProxy() Edouard@1440: except Exception, msg: andrej@1734: confnodesroot.logger.write_error(_("WAMP connection to '%s' failed.\n") % location) Edouard@1440: confnodesroot.logger.write_error(traceback.format_exc()) Edouard@1440: return None