简易FTP

　　老男孩----上海校区

　　简易FTP

客户端端

import  socket
import  struct
import json
import os
class MYTCPCLIENT:
    address_family = socket.AF_INET
    socket_type = socket.SOCK_STREAM
    allow_reuse_address = False
    max_packet_size = 8192
    coding ='utf-8'
    request_queue_size = 5
    def __init__(self,server_address,connect = True):
        self.server_address = server_address
        self.socket = socket.socket(self.address_family,self.socket_type)

        if connect:
            try:
                self.client_connect()
            except:
                self.client_close()
                raise
    def client_connect(self):
        self.socket.connect(self.server_address)

    def client_close(self):
        self.socket.close()

    def run(self):
        while True:
            choice = input(">>>>:").strip()
            if not choice:continue
            f = choice.split()
            cmd=f[0]
            if hasattr(self,cmd):
                func = getattr(self,cmd)
                func(f)

    def put(self,args):
        cmd = args[0]
        filename = args[1]
        if not os.path.isfile(filename):
            print('file:%s is not exists' %filename)
            return
        else:
            filesize = os.path.getsize(filename)

        head_dic = {'cmd':cmd,'filename':os.path.basename(filename),'filesize':filesize}
        head_json=json.dumps(head_dic)
        head_json_bytes = bytes(head_json,encoding=self.coding)
        head_struce = struct.pack('i',len(head_json_bytes))
        self.socket.send(head_struce)
        self.socket.send(head_json_bytes)
        send_size = 0
        with open(filename,'rb') as f:
            for line in f:
                self.socket.send(line)
                send_size+=len(line)
                print(send_size)
            else:
                print('下载成功')

client = MYTCPCLIENT(('127.0.0.1',8080))
client.run()

服务端

import  socketserver
import struct
import  json
import os

class FtpServer(socketserver.BaseRequestHandler):
    coding='utf-8'
    server_dir ='file_upload'
    max_packet_size = 1024
    BASE_DIR=os.path.dirname(os.path.abspath(__file__))
    def handle(self):
        print(self.request)
        while True:
            data = self.request.recv(4)
            data_len=struct.unpack('i',data)[0]
            head_json= self.request.recv(data_len).decode(self.coding)
            head_dic = json.loads(head_json)
            cmd = head_dic['cmd']
            if hasattr(self,cmd):      # 判断这个对象中有没有这个属性
                func=getattr(self,cmd) #获取对象属性值
                func(head_dic)
    def put(self,args):
        flie_path =  os.path.normpath(os.path.join(
            self.BASE_DIR,
            self.server_dir,
            args['filename']
        ))
        filesize=args['filesize']
        recv_size = 0
        print('------>',flie_path)
        with open(flie_path,'wb') as f:
            while recv_size <filesize:
                recv_data = self.request.recv(self.max_packet_size)
                f.write(recv_data)
                recv_size += len(recv_data)
                print('recvsize:%s filesize:%s' % (recv_size, filesize))
ftpserver = socketserver.ThreadingTCPServer(('127.0.0.1',8080),FtpServer)
ftpserver.serve_forever()