socket编程实现文件的上传和下载

#!/usr/bin/env python# coding:utf-8import socketimport structimport jsonbuffer = 1024sk = socket.socket()sk.bind((‘127.0.0.1‘,8888))sk.listen()conn,addr = sk.accept()ret = conn.recv(4)len_operate_c = struct.unpack(‘i‘,ret)[0]operate_c = conn.recv(len_operate_c)operate_c = operate_c.decode(‘utf-8‘)if operate_c == ‘upload‘: ???res = conn.recv(4) ???len_head = struct.unpack(‘i‘,res)[0] ???head = conn.recv(len_head).decode(‘utf-8‘) ???head = json.loads(head) ???filesize = head[‘filesize‘] ???filename = head[‘filename‘] ???with ?open (filename,‘wb‘) as ?f: ???????while filesize: ???????????if filesize >= buffer: ???????????????content = conn.recv(buffer) ???????????????f.write(content) ???????????????filesize -= buffer ???????????else : ???????????????content = conn.recv(filesize) ???????????????f.write(content) ???????????????filesize = 0conn.close()sk.close()

#!/usr/bin/env python# coding:utf-8def get_filename(file_path): ???return os.path.basename(file_path)import socketimport osimport jsonimport structbuffer = 1024sk = socket.socket()sk.connect((‘127.0.0.1‘,8888))operate = [‘download‘,‘upload‘]for ?k,j ?in ?enumerate(operate,1) : ???print(k,j)num = int(input(‘请输入你的选项： ‘))operate_c = operate[(num-1)]len_operate_c = len(operate_c)print(len_operate_c)res = struct.pack(‘i‘,len_operate_c)sk.send(res)sk.send(operate_c.encode(‘utf-8‘))if operate_c == ‘upload‘ : ???head = {‘filename‘:None,‘filesize‘:None,} ???file_path = input("请输入要上传的文件绝对路径 ：") ???filename = get_filename(file_path) ???head[‘filename‘] = filename ???filesize = os.path.getsize(file_path) ???head[‘filesize‘]=filesize ???json_head = json.dumps(head) ???len_head = len(json_head) ???ret = struct.pack(‘i‘,len_head) ???sk.send(ret) ???sk.send(json_head.encode(‘utf-8‘)) ???with open (file_path,‘rb‘) as f: ???????while filesize: ???????????if filesize >= buffer : ???????????????content = ?f.read(buffer) ???????????????sk.send(content) ???????????????filesize -= buffer ???????????else : ???????????????content = f.read(filesize) ???????????????sk.send(content) ???????????????filesize = 0sk.close()