python利用socketserver实现并发套接字功能_Python

本文实现利用python的socketserver这个强大的模块实现套接字的并发，具体内容如下

目录结构如下：

python利用socketserver实现并发套接字功能

测试文件请放在server_file文件夹里面

server.py

									#!/usr/bin/env python

									# -*- coding: gbk -*-

									# @Version : Python 3.5.2

									# @Time : 2018/1/24 10:29

									# @Author : Ncp

									# @File : server.py

									# @Software: PyCharm

									import json

									import time

									import hashlib

									import struct

									import os

									from socketserver import *

									FILE_PATH = os.path.dirname(os.path.abspath(__file__))+'\\server_file'

									class MYserver(BaseRequestHandler): # 设置一个类，基础BaseRequestHandler这个类

									 def handle(self):     # 这个方法下添加通信功能（和上面创建类一样，这是socketserver的固定模式）

									  print(self.client_address)

									  '''

									  :functions: 使用socketserver的并发套接字，提供客户端下载文件，并对文件进行MD5加密

									  '''

									  while True:

									   try:

									    data = self.request.recv(1024)

									    data_recv = data.decode('gbk').split()

									    if not os.path.exists(FILE_PATH+r'\%s' %data_recv[1]):

									     self.request.send('file is not found'.encode('gbk'))

									     continue

									    else:

									     data = self.request.send('1'.encode('gbk')) # 这里发现小问题，不回复一个信息的话，发送给客户端的包头居然成了没有封装

									     FILE_SIZE = os.path.getsize(FILE_PATH+r'\%s' %data_recv[1])

									     with open(FILE_PATH+r'\%s' %data_recv[1],'rb')as f:

									      hash_file = f.read()

									     m = hashlib.md5()

									     m.update(hash_file)

									     m_hex = m.hexdigest()

									     file_header = {'filename':data_recv[1],

									         'filesize':FILE_SIZE,

									         'md5':m_hex,

									         'time':time.strftime('%Y-%m-%d-%X',time.localtime())

									         }

									     # 包头信息序列化

									     file_header_dump = json.dumps(file_header)

									     # 编译成2进制

									     file_header_bytes = file_header_dump.encode('gbk')

									     # 封装报头

									     file_header_struct = struct.pack('i',len(file_header_bytes))

									     # 发送报头

									     self.request.send(file_header_struct)

									     # 发送报文内容

									     self.request.send(file_header_bytes)

									     # 发送文件数据

									     send_size = 0

									     with open(FILE_PATH+r'\%s' %data_recv[1] , 'rb')as f:

									      for i in f:

									       self.request.send(i)

									       send_size += len(i) # 这里后续可以拓展一个进度或者网速显示功能

									   except Exception:

									    self.request.close()

									if __name__ == '__main__':

									 server = ThreadingTCPServer(('127.0.0.1',8080),MYserver) # windows下只能开启多线程

									 server.serve_forever()

client.py

									#!/usr/bin/env python

									# -*- coding: gbk -*-

									# @Version : Python 3.5.2

									# @Time : 2018/1/24 10:29

									# @Author : Ncp

									# @File : client.py

									# @Software: PyCharm

									from socket import *

									import os,sys

									import hashlib

									import struct

									import math

									import json

									FILE_PATH = os.path.dirname(os.path.abspath(__file__))+'\\client_file'

									# 显示下载进度条功能，可以拓展为显示下载速度（提示，因为每次传输4096个字节，那么下载网速为KB/S,1KB个字节=1024B（字节），那么1s传输了多少个字节呢？）

									def progress(recvd, total):

									 fraction = '{:.0%}'.format(recvd / total)

									 sys.stdout.write('\r[%-30s] %s' % ('#' * int(math.floor(recvd * 30 / total)), fraction))

									 sys.stdout.flush()

									 if recvd == total:

									  sys.stdout.write('\n')

									# 主函数

									def run(ip,addr):

									 client = socket(AF_INET,SOCK_STREAM)

									 client.connect((ip,addr))

									 while True:

									  user_input = input('>>').strip()

									  cmd = user_input.split()

									  if len(cmd) != 2:

									   print('input format is error please use:get xx')

									   continue

									  if cmd[0] == 'get':

									   client.send(user_input.encode('gbk'))

									   data = client.recv(1024)

									   data_recv = data.decode('gbk')

									   if data_recv == 'file is not found':

									    print(data_recv)

									    continue

									  else:

									   print('commands is not found')

									   continue

									  # 收包头，然后一系列处理

									  header = client.recv(4)

									  if not header:break

									  header_json_len = struct.unpack('i', header)[0]

									  header_json_bytes = client.recv(header_json_len)

									  header_josn = header_json_bytes.decode('gbk')

									  header_dic = json.loads(header_josn)

									  # 去除包头内容进行下载

									  print(header_dic)

									  data_len = header_dic['filesize']

									  data_file = header_dic['filename']

									  data_md5 = header_dic['md5']

									  recv_size = 0

									  with open(FILE_PATH+r'\%s' %data_file,'wb')as fw:

									   while recv_size < data_len:

									    recv_data = client.recv(4096)

									    recv_size += len(recv_data)

									    fw.write(recv_data)

									    progress(recv_size,data_len)

									   print('Download completion, start validation')

									  # 收到文件后，读取文件进行加密，看是否与服务端下载的文件一致！

									  with open(FILE_PATH+r'\%s' %data_file,'rb')as fr:

									   data_total = fr.read()

									  m = hashlib.md5()

									  m.update(data_total)

									  m_hex = m.hexdigest()

									  if m_hex == data_md5:

									   print('文件验证通过')

									  else:

									   print('文件损坏,删除文件')

									   os.remove(FILE_PATH+r'\%s' %data_file)

									if __name__ == '__main__':

									 run('127.0.0.1',8080)