import ftplib
import string, socket
import time, ntptime
import config
import readsetting


readsetting = readsetting.READSETTING()

CONFIGFILE = "config.cfg"


class LFTP:

	def __init__(self):
		self.file_size = 0
		self.status = 0
		self.ftp = ftplib.FTP()
		self.servername = config.ftpserver
		self.userid = readsetting.get_user_id("/usr/local/lgvp/vdci.cfg")
		self.userpw = ""
		self.mode = 0 # 0:text mode, 1:binary mode
		self.totalreceive = 0
		self.conntime = ntptime.ntime()
		self.filelist = []

		config.firmware_file_response_code = []
		config.firmware_update_response_phrase = []
		config.firmware_file_response_crc = []

#by elpis, add return value

	def connection(self, ServerName=None, UserID=None, UserPW=None):
		while 1:
			try:
				if ServerName == None:
					break
				if config.update_debug:print "Connecting..."
				self.ftp.connect(ServerName)
				if config.update_debug:print "Connected to %s" % ServerName

			except socket.gaierror:
				if config.update_debug:print "No address associated with Server Name"
				ServerName = None
				return 0
			else:
				break

		if UserID == None:
			UserID = "ftp"
			UserPW = "temp@temp.com"


		while 1:
			try:
				test = self.ftp.login(UserID, UserPW)
				if config.update_debug:print "User login Success"
			except ftplib.error_perm:
				self.ftp.close()
				if config.update_debug:print "User login Fail"
				return 0
				break

			else:
				break

			if config.update_debug:print "User Login Success!!!"


		self.servername = ServerName
		self.userid = UserID
		self.userpw = UserPW
		return 1


	def getfilelist(self):
		try :

			conn = self.ftp.transfercmd('LIST')
                        fd = conn.makefile('rb')
                        while 1:
                                line = fd.readline()

				if config.update_debug:print line
				if len(line) == 0:
                                        if config.update_debug:print "+ No such file : %s" % (filename)
                                        break
                                if line[-2:] == "\r\n":
                                        tmp = line[:-2]
                                elif line[-1:] == "\n":
                                        tmp = line[:-1]
				tmp2 = string.split(tmp)
                                self.filelist.append(tmp2)

                        conn.close()
                        fd.close()
                except:
                        pass



	def getfilesize(self, filename):

		self.file_size = 0
		for fname in self.filelist:
			if config.update_debug:print fname
                        if fname[3] == filename:
                            	self.file_size = int(fname[2]) #서버에 따라서 틀림
                               	if config.update_debug:print 'self filesize :', self.file_size
                                break
        		if self.file_size != 0: break



	def setremotepassive(self):

		resp = self.ftp.sendcmd("PASV")

		if config.update_debug:print resp

	def setlocalpassive(self, mode): # need not to use, default is passive
		if self.localpassive:
			if mode:
				if config.update_debug:print "Already local passive mode enabled!!!"
			else:
				self.ftp.set_pasv(mode)
				self.localpassive = mode

		else:

			if mode:
				self.ftp.set_pasv(mode)
				self.localpassive = mode

			else:
				if config.update_debug:print "Already local passive mode disabled!!!"

	def settransfermode(self, mode):


		resp = self.ftp.sendcmd("TYPE I")

		self.mode = 1

		if config.update_debug:print resp


#changed
	def parse_response_message(self, response):

 		config.info_response_code = int(response[:3])

#		config.info_response_phrase = "\"" + response[4:] + "\"" //changed by elpis 20061115
		config.info_response_phrase = "\'" + response[4:] + "\'"
		
		config.firmware_update_response_code = config.info_response_code

		config.config_response_code = config.info_response_code

		config.config_response_phrase = config.info_response_phrase

		config.firmware_file_response_code.append(str(config.info_response_code))

		config.firmware_update_response_phrase.append(str(config.info_response_phrase))

		if config.update_debug:print config.firmware_file_response_code
		if config.update_debug:print config.firmware_update_response_phrase



	def getrequest(self, downfile, writefile):

		#insert try and except, and return value to process ftp error message by elpis

		try:
			#download  =  'RETR ' + config.hardware_version + '/' + config.firmware_version + '/' + writefile

			if self.mode:


				if config.update_debug:print "downloading %s .... " % writefile
				flen = 0
				f = file(writefile, 'wb')
				conn = self.ftp.transfercmd('RETR ' + downfile)

				while 1:
					block = conn.recv(8012)
					if len(block) == 0:
						break
					flen = flen + len(block)
					self.totalreceive += len(block)
					f.write(block)
				f.close()
				conn.close()
				return self.ftp.voidresp()


			else:
				self.getfilesize(filename)
				if config.update_debug:print "get text mode"
				self.ftp.retrlines('RETR ' + downfile, open(writefile, 'wb').write)
				return self.ftp.voidresp()
		except:
			import sys
			type, value, tb = sys.exc_info()
			return str(value) 
			#return self.ftp.response
			

	def getfile(self, downfile, writefile, comm):

		#insert try and except, and return value to process ftp error message by elpis
	
		try:
			#download  =  'RETR ' + config.hardware_version + '/' + config.firmware_version + '/' + writefile

			if self.mode:

				if config.update_debug:print "downloading %s .... " % writefile
				flen = 0
				f = file(writefile, 'wb')
				conn = self.ftp.transfercmd('RETR ' + downfile)

				while 1:
					block = conn.recv(8012)
					if len(block) == 0:
						break
					flen = flen + len(block)
					self.totalreceive += len(block)
					f.write(block)

					comm.send_packet("total_received_files_size")

					comm.send_packet(str(self.totalreceive))

					time.sleep(0)
				f.close()
				conn.close()
				return self.ftp.voidresp()


			else:
				self.getfilesize(filename)
				if config.update_debug:print "get text mode"
				self.ftp.retrlines('RETR ' + downfile, open(writefile, 'wb').write)
				return self.ftp.voidresp()
		except:		
			import sys
			type, value, tb = sys.exc_info()
			return str(value)
			#return self.ftp.response
			

        def reconnect(self,tar = None):
                try:
                        self.ftp.quit()
                except:
                        pass
                if config.update_debug:print "+ starting reconnection."
                self.connection(self.servername ,self.userid , self.userpw)



	def putfile(self, upfile):


		try:

			file = "STOR " + upfile

			if config.update_debug:print  file

			resp = self.ftp.storbinary(file, open(upfile, 'r'))

			if config.update_debug:print resp

		except:

			if config.update_debug:print "Can't upload : %s" % upfile

			pass