socket 網絡編程【實現FTP文件上傳和下載】

有四個基本的具體服務器類：

class socketserver.TCPServer(server_address, RequestHandlerClass, bind_and_activate=True)
    這使用Internet TCP協議，它在客戶端和服務器之間提供連續的數據流。如果bind_and_activate為true，構造函數將自動嘗試調用server_bind()和server_activate()。其他參數傳遞到BaseServer基類。

class socketserver.UDPServer(server_address, RequestHandlerClass, bind_and_activate=True)
　 這使用數據報，其是可能在運輸中不按順序到達或丟失的信息的離散分組。參數與TCPServer相同。

class socketserver.UnixStreamServer(server_address, RequestHandlerClass, bind_and_activate=True)
class socketserver.UnixDatagramServer(server_address, RequestHandlerClass, bind_and_activate=True)
    這些更常用的類與TCP和UDP類類似，但使用Unix域套接字；它們在非Unix平台上不可用。參數與TCPServer相同。

示例：使用SocketServer網絡服務框架實現FTP文件上傳和下載功能

服務端：

# !/usr/bin/python
# -*- coding:utf-8 -*-
'''
_author_=Captain
'''
import SocketServer
import os

class Myserver(SocketServer.BaseRequestHandler):
    def handle(self):
        base_path="D:\\temp"    #文件保存路徑
        conn=self.request
        client_address=self.client_address
        print "客戶端："+str(client_address)+"connected..."
        conn.send("服務端已准備接收...")
        while True:
            pre_data1,pre_data2=conn.recv(1024).split('|')
            if pre_data1=='put':  #如果客戶端是要上傳文件
                file_name,file_size=pre_data2.split('-')  #對客戶端傳遞過來的文件名和文件大小兩個值分割開
                recv_size=0
                file_dir=os.path.join(base_path,file_name)   #生成文件保存的路徑
                with open(file_dir,'wb') as f:
                    Flag = True
                    while Flag:
                        if int(file_size)>recv_size:
                            data = conn.recv(1024)
                            recv_size += len(data)
                            f.write(data)
                        else:
                            recv_size = 0
                            Flag = False
                            conn.send("upload successed")

            elif pre_data1 == 'get':   #如果客戶端是要下載文件
                file_size = os.stat(pre_data2).st_size
                file_path,file_name = os.path.split(pre_data2)
                conn.send(file_name + '|' + str(file_size))  #將文件名和計算出的文件大小發給客戶端
                send_size = 0
                with open(pre_data2, 'rb') as f:
                    Flag = True
                    while Flag:
                        if send_size + 1024 > file_size:   #如果文件大小不是1024的倍數，即最后的數據
                            data = f.read(file_size - send_size)
                            Flag = False
                        else:
                            data = f.read(1024)
                            send_size += 1024
                        conn.send(data)


if __name__ == '__main__':
    instance = SocketServer.ThreadingTCPServer(('127.0.0.1',8866), Myserver)  #使用異步多進程
    instance.serve_forever()

客戶端：

# !/usr/bin/python
# -*- coding:utf-8 -*-
'''
_author_=Captain
'''
import socket
import sys
import os

ip_port=('127.0.0.1',8866)
sk=socket.socket()
sk.connect(ip_port)
print sk.recv(1024)   #接收服務端提示建立連接的消息
while True:
    cmd,file_dir = raw_input('path:').split()  #對輸入的命令進行分割，如：put c:\xx\11.txt
    if cmd=='put':
        file_path, file_name = os.path.split(file_dir)
        file_size=os.stat(file_dir).st_size    #計算文件大小
        sk.send(cmd + '|' + file_name + '-' + str(file_size))
        send_size = 0
        with open(file_dir,'rb') as f:
            Flag = True
            while Flag:
                if send_size + 1024 > file_size:
                    data = f.read(file_size - send_size)
                    Flag = False
                else:
                    data = f.read(1024)
                    send_size += 1024
                sk.send(data)
            print sk.recv(1024)  # 接收服務端返回文件上傳成功的消息

    elif cmd=='get':
        sk.send(cmd + '|' + file_dir)   #把要下載服務端某個文件的具體路徑發送到服務端
        file_name,file_size=sk.recv(1024).split('|')   #接收服務端返回的文件名和文件大小
        file_dir = os.path.join(os.getcwd(), file_name)
        recv_size = 0
        with open(file_dir, 'wb') as f:
            Flag = True
            while Flag:
                if int(file_size) > recv_size:
                    data = sk.recv(1024)
                    recv_size += len(data)
                    f.write(data)
                else:
                    recv_size = 0
                    Flag = False
                    print 'download successed,文件已保存到 %s' % file_dir
    else:
        print "請輸入put或get命令，確認是上傳還是下載文件"

sk.close()