gpt4 book ai didi

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

转载 作者:qq735679552 更新时间:2022-09-28 22:32:09 25 4
gpt4 key购买 nike

CFSDN坚持开源创造价值,我们致力于搭建一个资源共享平台,让每一个IT人在这里找到属于你的精彩世界.

这篇CFSDN的博客文章python利用socketserver实现并发套接字功能由作者收集整理,如果你对这篇文章有兴趣,记得点赞哟.

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

目录结构如下:

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

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

server.py 。

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
#!/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 。

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
#!/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 )

自己可以设置一个多用户登录,然后测试,用户下载同一个文件,分别存入每个用户自己的家目录下面,效果更好.

当然现在也能测试,开启两个客户端同时下载文件.

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持我.

原文链接:https://www.cnblogs.com/encp/archive/2018/01/26/8358432.html 。

最后此篇关于python利用socketserver实现并发套接字功能的文章就讲到这里了,如果你想了解更多关于python利用socketserver实现并发套接字功能的内容请搜索CFSDN的文章或继续浏览相关文章,希望大家以后支持我的博客! 。

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com