source: trunk/textNewCnx.py @ 2290

# -*- coding: utf-8 -*-
# vim: ts=4:sw=4
#    This file is part of LOME.
#
#    LOME is free software: you can redistribute it and/or modify
#    it under the terms of the GNU General Public License as published by
#    the Free Software Foundation, either version 3 of the License, or
#    (at your option) any later version.
#
#    LOME is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU General Public License for more details.
#
#    You should have received a copy of the GNU General Public License
#    along with LOME.  If not, see <http://www.gnu.org/licenses/>.
# 

import os
import sys
import socket
import SocketServer
import threading
import struct

from time            import *
from PyQt4           import uic
from PyQt4.QtGui     import *
from PyQt4.QtCore    import *
from PyQt4.QtNetwork import QTcpSocket
from os.path         import join, dirname


sys.path.append( os.path.join( sys.path[0], 'common'))
from cnxObj          import *


#==============================================================================
## Class which implements the request handler used by plugRouter
class ServerHandler( SocketServer.BaseRequestHandler):
# ----------------------------------------------------------------------------
  ## Constructor, invoked when the plug connect to the router
  # @param self              object instance
  # @param request           instance of the request
  # @param clientAddress     IP Address of the client (plug)
  # @param server            instance of the caller
  def __init__( self, request, clientAddress, server):
    print "[ServerHandler] connection established with client"
    
    self.cnxObj = CnxObj( request)
    
    # Invoke __init__ method of the parent class
    SocketServer.BaseRequestHandler.__init__( self, request, clientAddress, server)

# ----------------------------------------------------------------------------
  ## Setup method. It is time to init some internal variables
  # The router is informed about this plug only after the plug name is known
  # @param self                object instance
  def setup( self):
    print "[ServerHandler] setup done"    
    return SocketServer.BaseRequestHandler.setup( self)

# ----------------------------------------------------------------------------
  ## Method to handle data available on socket
  # @param self                object instance
  def handle( self):
    print "[ServerHandler] handle invoked"    
    
    # Loop on receive
    while( True):
      # Get one message
      try:
        frames = self.cnxObj.receiveFrame()
      except:
        # Connection probably reset by peer
        return
      
      if len( frames[0]):
        for frame in frames:
          print "Frame received: %s" % frame
      else: 
        return
      # End of: if len(message):
    # End of: while(True):

# ----------------------------------------------------------------------------
  ## Call back invoked when the connection is closed
  # @param self                object instance
  def finish( self):
    print "[ServerHandler] Socket closed. Bye ..."    
    return SocketServer.BaseRequestHandler.finish( self)


#==============================================================================
class Server( SocketServer.ThreadingMixIn, SocketServer.TCPServer):
  allow_reuse_address = True
  daemon_threads      = True

# --------------------------------------------------------------------------------------------------------------------------------
  # Constructor
  # @param self                     object instance
  # @param serverAddress            Address of the server
  def __init__( self, serverAddress):
    # Start routerSocket manager
    SocketServer.TCPServer.__init__( self, serverAddress, ServerHandler)

g_window=None
cnxObj = None
def connected():
  print "connected"
  global cnxObj
  cnxObj.sendFrame( "This is a buffer tansmitted in the clear mode")
  cnxObj.sendFrame( "This is a buffer tansmitted in the zip mode", forceZip=True)
  cnxObj.sendFrame( "This is a buffer tansmitted in the zip mode\n" * 10)
  cnxObj.flush()
  
  frame = ""
  for index in range(0,5):
    frame += cnxObj.buildFrame( "This is the frame: %02d in a multiframe" % index)
  cnxObj.sendFrame( frame, multiFrame = True)
  
  
  cnxObj.disconnectFromHost()
  
   
   
def launch():
  print "launched"
  global cnxObj
  # QT test
  cnxObj = QTCnxObj()
  
  g_window.connect( cnxObj, SIGNAL("connected()"), connected )
  cnxObj.connectToHost( 'localhost', 10312)
   

def TestTransmition():
  global g_window
  g_pathname = os.path.abspath(dirname( sys.argv[0] ))

  app = QApplication(sys.argv)

  g_window = QMainWindow()
  button = QPushButton("launch")
  g_window.layout().addWidget( button )
  g_window.connect( button, SIGNAL("clicked()"), launch )


  # Start graphical interface
  g_window.show()
  status = app.exec_()

  # Create and connect socket
  sock = socket.socket( socket.AF_INET, socket.SOCK_STREAM)
  print "socket created"
  try:
    sock.connect( ( 'localhost', 10312))
    print "socket connected"
  except socket.error, msg:
    print msg
    return
  cnxObj = CnxObj( sock)
  cnxObj.sendFrame( "This is a buffer tansmitted in the clear mode")
  cnxObj.sendFrame( "This is a buffer tansmitted in the zip mode" ,  forceZip=True)    
  cnxObj.sendFrame( "This is a buffer tansmitted in the zip mode\n"*10)    
  
  sleep(1)
  sock.shutdown(2)

# =================================================
if __name__ == "__main__":
  server = Server(('localhost', 10312))
  serverThread = threading.Thread( target = server.serve_forever)
  serverThread.setDaemon( True) 
  serverThread.start()
  
  sleep(1)
  
  TestTransmition()