[TOC] 模拟ftp中的get和put命令,实现文件的上传和下载,用法了`struct`模块和`socket`模块 ## 服务端代码 ``` import socket import subprocess import struct import json import os share_dir=r'/Users/XXX/server/share' def get(conn,cmds): filename = cmds[1] # 以读的方式打开文件,读取文件内容发送给客户端 # 第一步：制作固定长度的报头 header_dic = { 'filename': filename, # 'filename':'1.mp4' 'md5': 'xxdxxx', 'file_size': os.path.getsize(r'%s/%s' % (share_dir, filename)) } header_json = json.dumps(header_dic) header_bytes = header_json.encode('utf-8') # 第二步：先发送报头的长度 conn.send(struct.pack('i', len(header_bytes))) # 第三步：再发报头 conn.send(header_bytes) # 第四步：再发送真实的数据 with open('%s/%s' % (share_dir, filename), 'rb') as f: # conn.send(f.read()) for line in f: conn.send(line) def put(conn,cmds): pass def run(): phone=socket.socket(socket.AF_INET,socket.SOCK_STREAM) phone.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1) phone.bind(('127.0.0.1',8912)) phone.listen(5) print('starting...') while True: # 链接循环 conn,client_addr=phone.accept() print(client_addr) while True: #通信循环 try: #1、收命令 res=conn.recv(8096) # b'put 1.mp4' if not res:break #适用于linux操作系统 #2、解析命令，提取相应命令参数 cmds=res.decode('utf-8').split() #['put','xxx.mp4'] if cmds[0] == 'get': get(conn,cmds) elif cmds[0] == 'put': input(conn,cmds) except ConnectionResetError: #适用于windows操作系统 break conn.close() phone.close() if __name__ == '__main__': run() ``` ## clinet端代码 ``` import socket import struct import json download_dir=r'/Users/XXX/client/download' def get(phone,cmds): # 以写的方式打开一个新文件，接收服务端发来的文件的内容写入客户的新文件 # 第一步：先收报头的长度 obj = phone.recv(4) header_size = struct.unpack('i', obj)[0] # 第二步：再收报头 header_bytes = phone.recv(header_size) # 第三步：从报头中解析出对真实数据的描述信息 header_json = header_bytes.decode('utf-8') header_dic = json.loads(header_json) ''' header_dic={ 'filename': filename, #'filename':'1.mp4' 'md5':'xxdxxx', 'file_size': os.path.getsize(filename) } ''' print(header_dic) total_size = header_dic['file_size'] filename = header_dic['filename'] # 第四步：接收真实的数据 with open('%s/%s' % (download_dir, filename), 'wb') as f: recv_size = 0 while recv_size < total_size: line = phone.recv(1024) # 1024是一个坑 f.write(line) recv_size += len(line) print('总大小：%s 已下载大小：%s' % (total_size, recv_size)) def put(phone,cmds): pass def run(): phone=socket.socket(socket.AF_INET,socket.SOCK_STREAM) phone.connect(('127.0.0.1',8912)) while True: #1、发命令 inp=input('>>: ').strip() #get a.txt if not inp:continue phone.send(inp.encode('utf-8')) cmds=inp.split() #['get','a.txt'] if cmds[0] == 'get': get(phone,cmds) elif cmds[0] == 'put': put(phone,cmds) phone.close() if __name__ == '__main__': run() ```