Python构建TCP来传输图片

python构建TCP的Client和Server来实现图片传输：

Client：

# -*- coding=utf-8 -*-

import socket
import os
import sys
import struct


def socket_client():
    try:
        client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        client.connect(('172.16.1.102', 2001))
    except socket.error as msg:
        print(msg)
        sys.exit(1)
 
    while True:
        # inputData = input()

        # if(inputData == 'quit'):
        #     print("I Quit")
        #     break
        # sendBytes = client.send(inputData.encode())

        # filepath = input("please input file path: ")
        filepath = '/home/magic/test.jpg'
        if os.path.isfile(filepath):
            # 定义定义文件信息。128s表示文件名为128bytes长，l表示一个int或log文件类型，在此为文件大小
            fileinfo_size = struct.calcsize('128sl')
            # 定义文件头信息，包含文件名和文件大小
            fhead = struct.pack('128sl', bytes(os.path.basename(filepath).encode('utf-8')), os.stat(filepath).st_size)
            client.send(fhead)
            print('client filepath: {0}'.format(filepath))
            with open(filepath, 'rb') as fp:
                while True:
                    data = fp.read(1024)
                    if not data:
                        print('{0} file send over...'.format(filepath))
                        break
                    client.send(data)
        client.close()
        break


if __name__ == '__main__':
    socket_client()

Server：

# -*- coding=utf-8 -*-
import socket
import threading
import sys
import os
import struct
import time

class MyServer:
    def __init__(self):
        super().__init__()
        self.time_start = 0
        self.time_end = 0

        # self.server

    def deal_data(self,client, addr):
        print('Accept new connection from {0}'.format(self.addr))
        self.time_start = time.time()
        try:
            while True:
                # data = client.recv(1024*1024)
                # if not data:
                #     print('Data is empty,Quit')
                #     break
                # localTime = time.asctime( time.localtime(time.time()))
                # print(localTime,' The Bytes of RecvData',len(data))
                # print(data.decode())

                fileinfo_size = struct.calcsize('128sl')  # linux 和 windows 互传 128sl 改为 128sq  机器位数不一样，一个32位一个64位
                recvData_buf = self.client.recv(fileinfo_size)
                print('Receive Bytes:',recvData_buf, type(recvData_buf))

                if recvData_buf:
                    print(recvData_buf, type(recvData_buf))
                    filename, filesize = struct.unpack('128sl', recvData_buf)
                    fn = filename.strip(str.encode('\00'))
                    new_filename = os.path.join(str.encode('./'), str.encode('new_') + fn)
                    print('file new name is {0}, filesize is {1}'.format(new_filename, filesize))
                    
                    recvd_size = 0  # 定义已接收文件的大小
                    with open(new_filename, 'wb') as fp:
                        print("start receiving...")
                        while not recvd_size == filesize:
                            if filesize - recvd_size > 1024:
                                data = self.client.recv(1024)
                                recvd_size += len(data)
                            else:
                                data = self.client.recv(filesize - recvd_size)
                                recvd_size = filesize

                            fp.write(data)
                    print("end receive...")
                    self.time_end = time.time()
                    print("Server transform image cast time: %.2f"%(self.time_end - self.time_start) )
        

                print("Server receive image over!")
                self.client.close()
                break


        except BaseException as e:
            print("Exception")
            print(repr(e))
        finally:
            # self.server.close()
            print("waiting for connected....")
           


    def socket_service(self):
        try:
            self.server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
            self.server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
            self.server.bind(('192.168.1.100', 2001)) # 这里换上自己的ip和端口
            self.server.listen(10)
        except socket.error as msg:
            print(msg)
            sys.exit(1)
        print("Waiting...")

        while True:
            self.client, self.addr = self.server.accept()
            self.t = threading.Thread(target = self.deal_data, args=(self.client, self.addr))
            self.t.start()


if __name__ == '__main__':
    myserver = MyServer()
    myserver.socket_service()
    # socket_service()