Python传输图片（同一局域网下版本）

深藏阁楼爱情的钟 2022-05-21 10:05 227阅读 0赞

## 简述 ##

Python可以通过网络套接字来进行传输图片  
这点还是比较有意思的。

下面用TCP来进行实现，适用于Python3

## 代码功能简述 ##

下面代码

*  实现了在一个电脑上发送图片（Sender）
 *  实现了在另外一个电脑（也可以是同一个电脑上的不同端口）接受图片（Reciever）
 *  只发一次，然后只收一次
 *  新图片会在旧图片上多一个 **“new\_”前缀**
 *  必须要先开接受者代码才行，之后再启动发送者代码（不然发送者代码会直接被拒绝）
 *  下面是固定了在一个电脑上（**看ip地址，下面写的是127.0.0.1**，如果是其他电脑就写其他电脑上的ip地址）
 *  另外本程序只能在同一个局域网下使用（就算是广域网上，那就必须要使用广域网的ip地址）

## 代码 ##

**【后附解析】**

**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循环中，就开始进行图片发送

*  通过struct库进行打包，然后再用套接字来发送  
    **注意：网络上的有些代码是不能直接用的，比如在这里，就需要把str变成bytes** 
 *  通过os库来获取文件的基本名（这个是为了避免有些文件中包含具体路径）
 *  通过os库来获取文件的大小（通过文件状态中关于大小的描述中获取到）

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**，其实就是为了来计数的标签  
之后就不断地收，边收边写入文件，一直到收完了之后。就关闭掉对应的文件流  
最后关闭套接字，退出循环