初学python网络编程socket,并实现文件传输【含源代码】
日期: 2020-05-18 分类: 跨站数据测试 308次阅读
什么是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
精华推荐