1.基础版(供学习了解原理使用,low)
server服务端
import socket import struct import json server = socket.socket() ip_port = ('127.0.0.1', 8001) server.bind(ip_port) server.listen() conn, addr = server.accept() head_len_msg = conn.recv(4) head_len = struct.unpack('i',head_len_msg)[0] head_byte = conn.recv(head_len) head = json.loads(head_byte.decode('utf-8')) file_size = head['size'] size = 0 with open('文件副本',mode='wb') as f1: while file_size: if file_size >= 1024: line = conn.recv(1024) f1.write(line) file_size -= 1024 else: content = conn.recv(1024) break conn.close() server.close()
client客户端
import os import socket import struct import json client = socket.socket() client.connect(('127.0.0.1', 8001)) buffer = 1024 filepath = input('请输入文件路径') filesize = os.path.getsize(filepath) head = {'size':os.path.getsize(filepath)} head_json = json.dumps(head) head_byte = head_json.encode('utf-8') head_len = len(head_byte) pack_len = struct.pack('i', head_len) client.send(pack_len) client.send(head_byte) with open(filepath,mode='rb') as f: while filesize: if filesize>buffer: content = f.read(buffer) client.send(content) filesize -= buffer else: content = f.read(filesize) client.send(content) break
2.升级版(正常使用)
封装了报头,解决了粘包问题的FTP传文件
server服务端
import struct,json,socket import subprocess import os # 引入所需模块
# 面向对象,定义server类 class MyTcpServer: address_family = socket.AF_INET # 定义socket的family属性,使用ipv4 socket_type = socket.SOCK_STREAM # 定义面向类型,面向流,是tcp传输 allow_reuse_address = False # 服务器地址复用, 默认不允许 max_packet_size = 8192 # 传包最大缓冲长度 coding = 'utf-8' # 编码类型 request_queue_size = 5 # 等待队列数,listen()时使用 server_dir = 'file_upload' # 服务器存放上传文件的文件夹名 , 测试时 要新建一个名为file_upload的文件夹 def __init__(self, server_address, bind_and_active = True): # 初始化方法, 传入服务器地址, 是否要开始监听 self.server_address = server_address self.socket = socket.socket() # 创建socket对象 if bind_and_active: # 如果需要连接 try: # 异常处理 self.server_bind() # 绑定端口 self.server_activate() # 监听开始 except: self.server_close() # 服务器关闭 raise def server_bind(self): # bind() 方法 if self.allow_reuse_address: # 如果允许复用 self.socket.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR, 1) # 服务器允许端口复用写这句话 self.socket.bind(self.server_address) # socket对象绑定服务器地址 self.server_address = self.socket.getsockname() # 获取套接字的名称 ,服务器的地址 def server_activate(self): # 服务器listen()方法 self.socket.listen(self.request_queue_size) def server_close(self): # close()方法 的函数 self.socket.close() def get_request(self): # 得到请求 accpt()方法的函数 return self.socket.accept() def close_request(self,request): # 关闭请求的方法 request.close()
def run(self): # run方法 while 1: self.conn,self.client_addr = self.get_request() # 获得conn ,addr print('from client ', self.client_addr) while 1: try: head_struct = self.conn.recv(4) # 接收4位 这4位是自定义报头的长度 if not head_struct: break # 如果 不是, 结束 head_len = struct.unpack('i', head_struct)[0] # 报头的长度 通过struct解包得出,解包为元组 第0项是报头长度,此时类型为字节 head_json = self.conn.recv(head_len).decode(self.coding) # 将长度解码 head_dic = json.loads(head_json) # 并用json转化为字典 print(head_dic) # 打印报头 cmd = head_dic['cmd'] # cmd 这一项等于 字典中的cmd选项 就是要实现什么功能 if hasattr(self,cmd): # 如果自己有cmd 这个功能 func = getattr(self,cmd) # 得到这个方法 func(head_dic) # 运行这个功能
except Exception: break def put(self,args): # 接收文件 file_path = os.path.normpath(os.path.join(self.server_dir,args['filename'])) # 文件路径 将接收的文件名和自己的路径拼接 并标准化 filesize = args['filesize'] # 文件大小为报头中标出的 recv_size = 0 # 接收的长度 先定义为0 print('----->',file_path) with open(file_path,'wb') as f: # 开始接收长度 while recv_size < filesize: recv_data = self.conn.recv(self.max_packet_size) # recv最大可接收缓存为定义好的最大长度 f.write(recv_data) # 写入文件 recv_size += len(recv_data) print('recvsize: %s filesize: %s' %(recv_size,filesize)) tcpserver1 = MyTcpServer(('127.0.0.1', 8001)) tcpserver1.run()
client客户端
import os import socket,json,subprocess,struct class MyTcpClient: addresss_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.servver_address = server_address self.socket = socket.socket(self.addresss_family,self.socket_type) if connect: try: self.client_connect() except: self.client_close() raise def client_connect(self): self.socket.connect(self.servver_address) def client_close(self): self.socket.close() def run(self): while True: inp = input('>>: ').strip() if not inp:continue l = inp.split() cmd = l[0] # 运行的功能 为切出的第一项 if hasattr(self,cmd): func = getattr(self,cmd) func(l) def put(self,args): # 传文件 cmd = args[0] # 报头执行的命令 filename = args[1] # 文件名 if not os.path.isfile(filename): # 如果这个路径是文件 print('file:%s is not exists' % filename) # 如果不是为空 else: filesize = os.path.getsize(filename) # 得到文件大小 head_dic = {'cmd':cmd,'filename':os.path.basename(filename),'filesize':filesize} # 定义报头 print(head_dic) # 打印一下报头看看 head_json = json.dumps(head_dic) # 用json转化为字符串 head_json_bytes = bytes(head_json, encoding=self.coding) # 编码成字节准备传输 head_struct = struct.pack('i', len(head_json_bytes)) # 用struct模块封包 把报头长度转化为4个字节 self.socket.send(head_struct) # 发送报头长度 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('upload successful') client = MyTcpClient(('127.0.0.1', 8001)) client.run()