# -*- coding: utf-8 -*-
##
# ftp_client
# Created on 2016 12 15
# @author:huangpingyi
##
import socket
import osclass FTP(object):def __init__(self, host, port):self.host = hostself.port = portdef put(self,msg):  #put local_fileprint '--put:', msgif len(msg) > 0:remote_filename = msg[0]#判断文件是否存在if os.path.isfile(remote_filename):file_size = os.path.getsize(remote_filename)#发送文件名和大小给服务cmd_msg = "file_transfer|put|%s|%s" % (remote_filename,file_size)self.sock.send(cmd_msg)#服务器返回应答feedback = self.sock.recv(1024)print '--feedback:',feedback#如果服务器准备好，就开始传输if feedback.startswith("file_transfer|put|ready"):f = file( remote_filename,'rb')send_size =0#文件很大，传送不完，每次以1024传输while not file_size == send_size:if file_size-send_size>1024:data = f.read(1024)send_size += 1024#最后一轮else:data = f.read(file_size-send_size)send_size+=file_size-send_sizeself.sock.send(data)else:print "--send file %s " % remote_filenamef.close()else:print feedbackdef get(self,msg):  #get remote_fileprint '--get:', msgif len(msg) > 0:remote_filename = msg[0]#发送get消息给服务器cmd_msg = "file_transfer|get|%s" % remote_filenameself.sock.send(cmd_msg)#服务器查看文件是否存在并响应feedback = self.sock.recv(1024)print '--feedback:',feedback#如果文件存在，服务器发送文件大小过来if feedback.startswith("file_transfer|get|ready"):#文件大小file_size = int(feedback.split("|")[-1])#客户端发送准备接收ack_msg = "file_transfer|get|recv|ready"self.sock.send(ack_msg)#建立文件f = file("recv\%s"% remote_filename,'wb')#客户端接受#收到的数据和总大小比较recv_size =0#文件很大，接受不完，开始计数，每次接收1024while not file_size == recv_size:if  file_size-recv_size>1024:data = self.sock.recv(1024)recv_size += len(data)#收到的小于1024，实时性不一定是1024#最后一轮else:data = self.sock.recv(file_size-recv_size)recv_size+=file_size-recv_sizef.write(data)else:print "--recv file %s " % remote_filenamef.close()else:print feedbackdef interactive(self):#客户端交互while True:user_input = raw_input("ftp_client::").strip()if len(user_input) == 0:continuecmd = user_input.split()if  hasattr(self, cmd[0]):#用反射判断用户是要get还是put#判断对象是否有cmd[0]方法func = getattr(self, cmd[0])#有则将cmd[0]方法赋值func(cmd[1:])#把后面的参数传给cmd[0]方法,并执行cmd[0]函数else:#如果是其他命令，就打印错误print "\033[31;1mWrong cmd usage!\033[0m"def connect(self):#连接self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)self.sock.connect((self.host, self.port))self.interactive()if __name__ == '__main__':ftp = FTP('localhost',7126 )ftp.connect()

服务器端代码：

# -*- coding: utf-8 -*-
##
# socket_server
# Created on 2016 12 15
# @author:huangpingyi
##import SocketServer
import osclass MyTCPHandler(SocketServer.BaseRequestHandler):def handle(self):while True:data = self.request.recv(1024).strip()data = data.split("|")if hasattr(self, data[0]):func = getattr(self, data[0])func(data[1:])#将后面参数传入file_transferdef file_transfer(self,msg):if msg[0] == "get":print "----going to send file:%s to client" % msg[1]filename = msg[1]if os.path.isfile(filename):#判断文件存在#如果文件存在，就判断文件大小file_size = os.path.getsize(filename)#将文件大小发送給客户端ack_msg = "file_transfer|get|ready|%s" % file_sizeself.request.send(ack_msg)client_ack = self.request.recv(1024)print '-->client ack:', client_ack#如果客户端准备好就开始传输文件if client_ack.startswith("file_transfer|get|recv|ready"):f = file(filename, "rb")send_size =0Flag = Truewhile Flag:if send_size+1024 > file_size:data = f.read(file_size-send_size)Flag = Falseelse:data = f.read(1024)send_size+=1024self.request.send(data)print '--file %s send done---' % filenamef.close()else:ack_msg = "file_transfer|get|doesn't exist"self.request.send(ack_msg)if msg[0] == "put":print "----going to put file:%s" % msg[1]#获取文件名字和大小file_name = msg[1]file_size = msg[2]base_path = "D:\untitled4"#发送准备好接收的请求ack_msg = "file_transfer|put|ready"self.request.send(ack_msg)recv_size = 0#上传文件路径拼接file_dir = os.path.join(base_path,file_name)f = file(file_dir,'wb')Flag = Truewhile Flag:#未上传完毕，if int(file_size)>recv_size:#最多接收1024，可能接收的小于1024data =self.request.recv(1024)recv_size+=len(data)#上传完毕，则退出循环else:recv_size = 0Flag = Falsecontinue#写入文件f.write(data)print 'upload successed.'f.close()if __name__ == '__main__':HOST,PORT = 'localhost',7126server = SocketServer.ThreadingTCPServer((HOST,PORT),MyTCPHandler)server.serve_forever()