Young87

SmartCat's Blog

So happy to code my life!

游戏开发交流QQ群号60398951

当前位置:首页 >跨站数据测试

初学python网络编程socket,并实现文件传输【含源代码】

什么是Socket

socket又称“套接字”, 应用程序通常通过“套接字”向网络发出请求或者应答网络请求,使主机间或者一台计算机上的进程间可以通讯。

socket()函数

Python 中,我们用 socket()函数来创建套接字,语法格式如下:

socket.socket(family, type, protocol)

参数: family: 套接字家族可以使AF_UNIX或者AF_INET
type:套接字类型可以根据是面向连接的还是非连接分为SOCK_STREAM或SOCK_DGRAM
protocol: 一般不填默认为0.

socket 对象内建方法

函数 描述
服务器端套接字:

s.bind() 绑定地址(host,port)到套接字, 在AF_INET下,以元组(host,port)的形式表示地址。

s.listen() 开始TCP监听。backlog指定在拒绝连接之前,操作系统可以挂起的最大连接数量。该值至少为1,大部分应用程序设为5就可以了。

s.accept() 被动接受TCP客户端连接,(阻塞式)等待连接的到来 客户端套接字

s.connect() 主动初始化TCP服务器连接,。一般address的格式为元组(hostname,port),如果连接出错,返回socket.error错误。

s.connect_ex() connect()函数的扩展版本,出错时返回出错码,而不是抛出异常

公共用途的套接字函数

s.recv() 接收TCP数据,数据以字符串形式返回,bufsize指定要接收的最大数据量。flag提供有关消息的其他信息,通常可以忽略。

s.send() 发送TCP数据,将string中的数据发送到连接的套接字。返回值是要发送的字节数量,该数量可能小于string的字节大小。

s.sendall() 完整发送TCP数据,完整发送TCP数据。将string中的数据发送到连接的套接字,但在返回之前会尝试发送所有数据。成功返回None,失败则抛出异常。

s.recvfrom() 接收UDP数据,与recv()类似,但返回值是(data,address)。其中data是包含接收数据的字符串,address是发送数据的套接字地址。

s.sendto() 发送UDP数据,将数据发送到套接字,address是形式为(ipaddr,port)的元组,指定远程地址。返回值是发送的字节数。

s.close() 关闭套接字

s.getpeername() 返回连接套接字的远程地址。返回值通常是元组(ipaddr,port)。

s.getsockname() 返回套接字自己的地址。通常是一个元组(ipaddr,port)

s.setsockopt(level,optname,value) 设置给定套接字选项的值。

s.getsockopt(level,optname[.buflen]) 返回套接字选项的值。

s.settimeout(timeout) 设置套接字操作的超时期,timeout是一个浮点数,单位是秒。值为None表示没有超时期。一般,超时期应该在刚创建套接字时设置,因为它们可能用于连接的操作(如connect())

s.gettimeout() 返回当前超时期的值,单位是秒,如果没有设置超时期,则返回None。

s.fileno() 返回套接字的文件描述符。

s.setblocking(flag) 如果flag为0,则将套接字设为非阻塞模式,否则将套接字设为阻塞模式(默认值)。非阻塞模式下,如果调用recv()没有发现任何数据,或send()调用无法立即发送数据,那么将引起socket.error异常。

s.makefile() 创建一个与该套接字相关连的文件

-----**** 上述来源于网络 ****-----------

实例:实现文件传输

实例演示:

服务器端打开
在这里插入图片描述
客户端运行,选择文件上传
在这里插入图片描述
客户端文件上传完毕
在这里插入图片描述
服务器端收到文件
在这里插入图片描述
注:本次演示不是在局域网上,而是在实际上的服务器上的文件传输演示

源代码:

客户端

"""
author: 全洛

"""
import socket
import easygui as eg
import json
import os
import time

'''读取配置文件'''
f = open('clientconf.json', 'r', encoding='utf-8')
date = json.load(f)
ip = date.get('IP')
port = date.get('port')
f.close()

'''创建socket对象'''
client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

try:
    '''连接服务器, 指定IP和端口'''
    client.connect((ip, port))

    '''接收小于 1024 字节的数据'''
    msg = client.recv(1024)

    '''输出连接成功信息'''
    print(msg.decode('utf-8'))
except Exception as e:
    print(f'连接服务器失败', e)
else:
    '''选择发送的文件'''
    filepath = eg.fileopenbox(title='选择文件')
    '''获取文件名,文件大小'''
    filename = filepath.split("\\")[-1]
    filesize = os.path.getsize(filepath)
    # 先将文件名传过去
    # 编码文件名
    client.send(filename.encode())
    time.sleep(0.5)
    # 再将将文件大小传过去
    # 编码文件大小
    client.send(filesize.to_bytes(filesize.bit_length(), byteorder='big'))
    try:
        '''传输文件'''
        start_time = time.time()
        with open(filepath, 'rb') as f:
            size = 0
            while True:
                #读取文件数据,每次1024KB
                f_data = f.read(1024)
                # 数据长度不为零,传输继续
                if f_data:
                    client.send(f_data)
                    size += len(f_data)
                    if time.time() - start_time == 0:
                        time.sleep(0.5)
                    speed = (size) / (time.time() - start_time)
                    print('\r' + '【上传进度】:%s%.2f%%, Speed: %.2fMB/s' % ('>' * int(size * 50 / filesize), float(size / filesize * 100), float(speed / 1024 / 1024)), end=' ')
                # 数据长度为零传输完成
                else:
                    print(f'{filename},{float(filesize/1024/1024):.2f}MB, 传输完成')
                    break
    except Exception as e:
        print(f'传输异常', e)
finally:
    '''关闭客户端'''
    client.close()

客户端配置:
{
“IP”: “182.92.112.155”,
“port”: 3636
}

服务器端

"""
author: 全洛

"""
import socket
import os
import json
import threading
import time


'''导入配置文件'''
f = open('serverconf.json', 'r', encoding='utf-8')
dirsave = json.load(f).get('dirSave')
f.close()
if not os.path.exists(dirsave):  # 看是否有该文件夹,没有则创建文件夹
    os.mkdir(dirsave)

class server:
    def __init__(self):
        ''''''
        ''' 创建socket对象'''
        self.server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

        '''获取本地'''
        self.host = socket.gethostname()

        '''设置端口'''
        self.port = 3636

        '''绑定地址'''
        self.server.bind((self.host, self.port))

        '''设置最大连接数, 超过后排队'''
        self.server.listen(12)

        print('服务器已上线\n等待客户端连接...\n')

    def server100(self):
        while True:
            '''建立客户端连接'''
            client, addr = self.server.accept()
            print(f'客户端: {str(addr)}, 连接上服务器')
            msg = f'已连接到服务器{self.host}!\r\n'
            client.send(msg.encode('utf-8'))
            # 有客户端连接后,创建一个线程给客户端
            t1 = threading.Thread(self.taskfilethread(client))
            # 设置线程守护
            t1.setDaemon(True)
            # 启动线程
            t1.start()

    def taskfilethread(self, client):
        '''接收文件名,文件大小'''
        filename = client.recv(1024)
        # time.sleep(0.5)
        filesize = client.recv(1024)
        # 解码文件名,文件大小
        filename = filename.decode()
        filesize = int.from_bytes(filesize, byteorder='big')
        '''接收文件'''
        try:
            f = open(dirsave+"\\"+ filename, 'wb')
            size = 0
            start_time = time.time()
            while True:
                # 接收数据
                f_data = client.recv(1024)
                # 数据长度不为零,接收继续
                if f_data:
                    f.write(f_data)
                    size += len(f_data)
                    if time.time() - start_time == 0:
                        time.sleep(0.5)
                    speed = (size) / (time.time() - start_time)
                    print('\r' + '【下载进度】:%s%.2f%%, Speed: %.2fMB/s' % ('>' * int(size * 50 / filesize), float(size / filesize * 100), float(speed/1024/1024)),end=' ')
                # 数据长度为零接收完成
                else:
                    break
        except Exception as e:
            print(f'接收异常', e)
        else:
            f.flush()
            print(f'{filename},{float(filesize/1024/1024):.2f}MB, 接收完成')
            f.close()

if __name__ =='__main__':
    server = server()
    server.server100()

服务器端配置:
{
“dirSave”: “C:\Users\Administrator\Desktop\temp”
}


最后

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:

除特别声明,本站所有文章均为原创,如需转载请以超级链接形式注明出处:SmartCat's Blog

上一篇: RocketMQ一行代码造成大量消息发送失败

下一篇: 华为 5G、阿里检测病毒算法、腾讯 AI 一分钟诊断,国内抗疫科技大阅兵!

精华推荐