select模块

select模块

select的ftp server端，不过有问题，select只是实现数据的一次请求发送，但是这个模块里用了多次，所以会报错

import json,queue,select,socket,syssys.path.append("..")from core import methodserver = socket.socket()server.bind(("localhost",9000))server.listen(1000)server.setblocking(False)   #不阻塞msg_dic = {}inputs = [server,]#inputs = [server,conn,conn2] #[conn2,]outputs = []while True:    readable,writable,exceptional = select.select(inputs,outputs,inputs)    print(readable,writable,exceptional)    for r in readable:        if r is server:            conn,addr = server.accept()            print("来了一个新连接",addr)            inputs.append(conn)            msg_dic[conn] = queue.Queue()        else:            data = r.recv(1024).strip()            print("收到数据",data)            cmd_dic = json.loads(data.decode())            # msg_dic[r].put(cmd_dic)            # action = cmd_dic["action"]            # msg_dic[r].put(action)            func_name = cmd_dic["action"]            msg_dic[r].put(func_name)            file_information = method.A(r,cmd_dic)            msg_dic[r].put(file_information)            outputs.append(r)    for w in writable:        print(w,777777777777)        func_name2 = msg_dic[w].get()        data_to_client = msg_dic[w].get()        func = getattr(method,func_name2)        print(func,type(func))        func(w,data_to_client)        outputs.remove(w)    for e in exceptional:        if e in outputs:            outputs.remove(e)        inputs.remove(e)        del msg_dic[e]

ftp server端的方法模块

import jsonimport osdef put(r, filename,filesize):    """接收客户端文件"""    if os.path.isfile(filename):        f = open(filename + ".new", "wb")    else:        f = open(filename, "wb")    r.send(b"200 ok")    received_size = 0    while received_size < filesize:        data = r.recv(1024)        f.write(data)        received_size += len(data)    else:        print("file [%s] has uploaded..." % filename)def get(r, filename):    '''发送给客户端文件'''    f = open(filename, "rb")    for line in f:        r.send(line)    else:        print("file transport success!")def A(r,*args):    cmd_dic = args[0]    action = cmd_dic["action"]    filename = cmd_dic["filename"]    # filesize = cmd_dic["size"]    if action == "get":        if os.path.isfile(filename):            r.send("ok".encode())            print("wait....")            client_ack = r.recv(1024)            client_ack = client_ack.decode()            if client_ack == "ok":                cmd_dic["filesize"] = os.stat(filename).st_size                r.send(json.dumps(cmd_dic).encode())                client_response = r.recv(1024)                client_response = client_response.decode()                print(client_response, type(client_response))                if client_response == "ok":                    return filename                else:                    return "client收到filesize但没有回应......."            else:                return "client没有确认file是否存在....."        else:            r.send("no".encode())            return "file is not exist!"    elif action == "put":        filesize = cmd_dic["size"]        # put(r,filename,filesize)        return filename,filesize

ftp client端

import socketimport osimport jsonclass FTPclient(object):    def __init__(self):        self.client = socket.socket()    def help(self):        msg = '''        ls        pwd        cd ../..        get filename        put filename        '''        print(msg)    def connect(self,ip,port):        self.client.connect((ip,port))    def interactive(self):        while True:            cmd = input(">>:").strip()            if len(cmd) == 0:continue            cmd_str = cmd.split()[0]            if hasattr(self,"cmd_%s" % cmd_str):                function = getattr(self,"cmd_%s" % cmd_str)                function(cmd)            else:                self.help()    def cmd_put(self,*args):        cmd_split = args[0].split()        if len(cmd_split) > 1:            filename = cmd_split[1]            if os.path.isfile(filename):                filesize = os.stat(filename).st_size                msg_dic = {                    "action":"put",                    "filename":filename,                    "size":filesize,                    "overridden":True                }                self.client.send(json.dumps(msg_dic).encode())                #f防止粘包，服务器确认                server_response = self.client.recv(1024)                f = open(filename,"rb")                for line in f:                    self.client.send(line)                else:                    print("file upload success")                    f.close()            else:                print(filename,"is not exist!")    def cmd_get(self,*args):        cmd_split = args[0].split()        if len(cmd_split) > 1:            filename = cmd_split[1]            msg_dic = {                "action":"get",                "filename":filename,                "filesize":""            }            self.client.send(json.dumps(msg_dic).encode())            #f防止粘包，服务器确认            file_exist = self.client.recv(1024)            file_exist = file_exist.decode()            if file_exist == "ok":                self.client.send("ok".encode())                data = self.client.recv(1024)                msg_dic = json.loads(data.decode())                filesize = msg_dic["filesize"]                print(filesize)                if os.path.isfile(filename):                    f = open(filename+".new", "wb")                else:                    f = open(filename,"wb")                self.client.send("ok".encode())  #向server发送ack，可以收包了                received_size = 0                while received_size < filesize:                    data = self.client.recv(1024)                    f.write(data)                    received_size += len(data)                else:                    print("file download completed!")                    f.close()            else:                print("file is not exist in server!")ftp = FTPclient()ftp.connect("localhost",9000)ftp.interactive()