时间:2022-04-27 10:27:57 | 栏目:Python代码 | 点击:次
简述
Python可以通过网络套接字来进行传输图片
这点还是比较有意思的。
下面用TCP来进行实现,适用于Python3
代码功能简述
下面代码
代码
【后附解析】
Sender.py (发送者)
import socket import os import sys import struct def sock_client(): try: s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) s.connect(('127.0.0.1', 6666)) except socket.error as msg: print(msg) print(sys.exit(1)) while True: filepath = input('input the file: ') # filepath = 'test.png' fhead = struct.pack(b'128sl', bytes(os.path.basename(filepath), encoding='utf-8'), os.stat(filepath).st_size) s.send(fhead) print('client filepath: {0}'.format(filepath)) fp = open(filepath, 'rb') while 1: data = fp.read(1024) if not data: print('{0} file send over...'.format(filepath)) break s.send(data) s.close() break if __name__ == '__main__': sock_client()
Reciever.py (接受者)
import socket import os import sys import struct def socket_service(): try: s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) s.bind(('127.0.0.1', 6666)) s.listen(10) except socket.error as msg: print(msg) sys.exit(1) print("Wait") while True: sock, addr = s.accept() deal_data(sock, addr) break s.close() def deal_data(sock, addr): print("Accept connection from {0}".format(addr)) while True: fileinfo_size = struct.calcsize('128sl') buf = sock.recv(fileinfo_size) if buf: filename, filesize = struct.unpack('128sl', buf) fn = filename.decode().strip('\x00') new_filename = os.path.join('./', 'new_' + fn) recvd_size = 0 fp = open(new_filename, 'wb') while not recvd_size == filesize: if filesize - recvd_size > 1024: data = sock.recv(1024) recvd_size += len(data) else: data = sock.recv(1024) recvd_size = filesize fp.write(data) fp.close() sock.close() break if __name__ == '__main__': socket_service()
解析:
发送者代码:
还是先创建TCP套接字,并发起连接:
具体代码如下:
try: s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) s.connect(('127.0.0.1', 6666)) except socket.error as msg: print(msg) print(sys.exit(1))
接下来的While循环中,就开始进行图片发送
fhead = struct.pack(b'128sl', bytes(os.path.basename(filepath), encoding='utf-8'), os.stat(filepath).st_size) s.send(fhead)
之后再通过二进制方式来获取文件。
然后再发送出去,每次发送固定长度的数据。先读,再判断是否为空,再发送。
接受端:
还是一样,先创建套接字(不过这个需要绑定对应的ip以及端口)
具体代码如下:(这里写的是监听10个套接字对象,但是在我们代码中并没哟用到)
try: s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) s.bind(('127.0.0.1', 6666)) s.listen(10) except socket.error as msg: print(msg) sys.exit(1)
之后就进入到循环,来判断,是否获取到sock对象,这里是会进入阻塞状态的。
在deal_data中就开始通过连接的对象来获取数据。
一样,还是一个无限循环,但其实本质上还是空的。。。没什么特别的。
之后就是对于传过来的文件名处理一下,这里的这些“128sl”,其实就是一个大小规模(类似于打包的箱子大小的样子)。然后就是得到新的文件名了。
对应的代码如下:
filename, filesize = struct.unpack('128sl', buf) fn = filename.decode().strip('\x00') new_filename = os.path.join('./', 'new_' + fn)
之后的recvd_size,其实就是为了来计数的标签
之后就不断地收,边收边写入文件,一直到收完了之后。就关闭掉对应的文件流
最后关闭套接字,退出循环