ModClientTCP.py

Go to the documentation of this file.

#!/usr/bin/python

## @package serverportfolio.ModClientTCP
# @brief Make a class for sending new data to the server of ROOT_application.
#
# A simple TCP client using module socket
#
# Last Changed $Id: ModClientTCP.py 13 2015-04-12 19:45:14Z michael $

import socket
import logging, sys

from GlobalDicts import HOST,PORT_GUI

## @class TCPClient
# @brief Make a connection to the root application server to send directly new data.
# do not accept answer from the server.\n
# Test is listening (correct connect function) to do
class TCPClient:
    
    def __init__(self):
        self.logger = logging.getLogger( "SP.TCPClient")
        self.logger.debug("Constructor TCPClient PORT to send %d" % PORT_GUI)
        
        # can set only once ? seems to need every time it is closed ?
        # Address Family : AF_INET (this is IP version 4 or IPv4)
        # Type : SOCK_STREAM (this means connection oriented TCP protocol)
        #self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        
        ## default true at creation, SendData will send back false if the connection is impossible.
        # now tested by Connect every time
        self.active = False #True #False
        
    def GetActivity(self):
        return self.active
    
    ## Try to connect to the server
    #
    # connection is opened at the end ??
    #
    # @param only_check: if True try to connect and close the socket and return the status
    def Connect(self, only_check=False):
        
        # Create a socket (SOCK_STREAM means a TCP socket)
        # seems to reassign after each call to close() ??
        try :
            self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        # all function in socket throw socket.error
        except socket.error, msg:
            #print 'Failed to create socket. Error code: ' + str(msg[0]) + ' , Error message : ' + msg[1]
            self.logger.info('Failed to create socket. Error code: %d Error message : %d ' % (msg[0], msg[1]))
    
        # Connect to server before sending data
        try:
            self.sock.connect((HOST, PORT_GUI))
            self.logger.debug("OK for connection to %d" % PORT_GUI)
            # here root server crash, if necessary to close, should use finally
            if only_check:
                self.logger.debug("ONLY_CHECK, close the socket")
                self.sock.close()

            return True 

        # got exception if connection impossible (not running)
        except socket.error, ex:
            self.sock.close()
            #print "Get exception socket.error in connect(), no connection possible"
            self.logger.info('Failed in socket.connect. Error code: %d, Error message : %s' % (ex[0], ex[1]) )
            # could raise exception if caught by caller    
            return False

        # should not be necessary, keep for testing
        except ex:
            self.sock.close()
            self.logger.error("Get a base exception in connect(), SHOULD NOT HAPPEN. no connection possible")
            return False
    
    ## Connect to the server
    #  send the data
    #  no answer at this point
    #  Close the socket
    def SendData ( self,data ):
        self.logger.debug("Entry SendData")
        self.logger.debug("Data to send %s" % data)

        status = self.Connect()
        self.logger.debug("status %d" % status)
        if status == False :
            self.logger.info("GUI Server NOT RUNNING")
            return False;

        # alternative sendall() return None
        # append " END" for marking the end of the message, expected by TCPServer
        try :
            nb_bytes_send = self.sock.send(data + " END")
            self.logger.debug("nb_bytes_send %d" % nb_bytes_send)
        except socket.error, ex:
            self.logger.error("Get exception socket.error in connect(), no connection possible")
            self.logger.error('Failed in socket.coonect. Error code: ' + str(ex[0]) + ' , Error message : ' + ex[1])
            return False

        # not used for the moment
        # Receive data from the server and shut down
        #received = sock.recv(1024)
        self.sock.close()
        return True
        # something send back from server gui ? 
        #return received

####################    main function ##########
if __name__ == "__main__":

    # should make some test ??
    #data = " ".join(sys.argv[1:])
    
    #  END  added in SendData
    data="DJ,2014-09-22 22:08:09,17172.680000,74298798.000000"
    #data="1 END" #crash but send the message, certainly in creation DataStock
    #print "data ", data
    tcp = TCPClient()
    
    # test connection manually, could be in __init__
    # need to close
    #tcp.active = tcp.Connect( only_check=True )
    #tcp.sock.close()
    
    #print "after tcp.Connect, only_check=True"
    print "GetActivity ",  tcp.GetActivity()

    #tcp.SendData( data )
    
    #tcp.SendData( data )
    ## Receive data from the server and shut down
    #received = sock.recv(1024)
    #sock.close()
    received = tcp.SendData( data )

    print "Sent:     %s" % data
    print "Received: %s" % received
    # version for web application
    #print "%s" % received