socketClient.py

#!/usr/bin/env python
# encoding: UTF-8

import socket
import sys,getopt
import xmlReader
import CondorTools

chirp = CondorTools.CondorChirp()
class Client:

	def __init__(self,config):
		self.config = config
		print ("client init")

	def get_constant(self, prefix):
		"""Create a dictionary mapping socket module constants to their names."""
		return dict( (getattr(socket, n), n)
						for n in dir(socket) if n.startswith(prefix)
					)

	def get_constants(self, sock):
		families = self.get_constant('AF_')
		types = self.get_constant('SOCK_')
		protocols = self.get_constant('IPPROTO_')		
		
		print >>sys.stderr, 'Family  :', families[sock.family]
		print >>sys.stderr, 'Type    :', types[sock.type]
		print >>sys.stderr, 'Protocol:', protocols[sock.proto]
		print >>sys.stderr

	def demand(self):
		
		print ("client demand")
		try:
			xmlHandler = xmlReader.XmlHandler(self.config)
		except:
			print "xml read error"
			sys.exit()
		host, port, path, timestamp, offset = xmlHandler.read()
		print "five vals:"
		print host, port, path, timestamp, offset
			
		interval = 5
		maxtries = 12*3
		serverInfo = chirp.getJobAttrWait("SocketServer",None,interval, maxtries)	
		print "serverInfo is:" + serverInfo
		#hostFromCondor,portFromCondor = serverInfo.strip("'").split()
		#print hostFromCondor, portFromCondor
		# Create a TCP/IP socket
		sock = socket.create_connection((host, int(port)))
		self.get_constants(sock)
		
		print host, port, path, timestamp, offset

		try:
			# Send data
			message = timestamp + ',' + offset
			sock.sendall(message)
			amount_received = 0			
			amount = int(sock.recv(8))
			#print amount
			sock.sendall("kunBegin")
    			
			dataComp = ""
			while amount_received < int(amount):
				print (amount_received, amount)
				data = sock.recv(min(4096, int(amount) - amount_received))
				dataComp += data
				amount_received += len(data)
			
			strAdded, timestamp, offset = dataComp.split("KUNSIGN")
			
			print strAdded
			if not amount_received < amount:
				with open(path, "a") as output:
					output.write(strAdded)
				print "time is " + timestamp
				if timestamp and offset:
					xmlHandler.write(timestamp, offset, self.config)
		except:
			sock.sendall("kunStop")
			print "amount value error"
		
		finally:
			print >>sys.stderr, 'closing socket'
			sock.close()
			print("dadada")

if __name__ == '__main__':
	if len(sys.argv) < 2:
		print ("client val num error!")
	client = Client(sys.argv[1])
	client.demand()