服務(wù)端.py

from socket import *
from fileManager import*
from threading import*


class ServerThread(Thread):
    """創(chuàng)建子線程"""
    def __init__(self, connection, address):
        super().__init__()
        self.connection = connection
        self.address = address

    def run(self):
        picture_nums = Number()
        # 讓通信一直進(jìn)行
        while True:
            # 接收信息
            re_message = self.connection.recv(1024).decode(encoding='utf-8')
            if re_message == '1':
                # print('發(fā)送文字')
                message = input('請(qǐng)輸入：')
                self.connection.send(message.encode())
                print('文字發(fā)送成功!')
            elif re_message == '2':
                # 發(fā)送圖片
                picture_message = binary_read('./server_files./圖片.jpg')
                self.connection.send(picture_message)
                print('圖片發(fā)送成功！')
            elif re_message == '3':
                # 發(fā)送視頻
                video_message = binary_read('./server_files/視頻.mp4')
                self.connection.send(video_message)
                print('視頻發(fā)送成功！')
            elif re_message == '4':
                # 接收?qǐng)D片
                picture_data = bytes()
                while True:
                    picture_message = self.connection.recv(1024)
                    if len(picture_message) < 1024:
                        picture_data += picture_message
                        binary_write('./server_files/圖片%d.jpg' % picture_nums.create_num(), picture_data)
                        break
                    picture_data += picture_message
                print('圖片接收成功！')
            elif re_message == '5':
                self.connection.close()
                print('%s已離開(kāi)...' % self.address[0])
                break
            else:
                continue


def create_server():
    server = socket()
    server.bind(('10.7.160.92', 1234))
    server.listen(512)
    print('開(kāi)始監(jiān)聽(tīng)...')
    while True:
        connection, address = server.accept()
        print('歡迎%s...' % address[0])
        t1 = ServerThread(connection, address)
        t1.start()


if __name__ == '__main__':
    create_server()

客戶端.py

from socket import *
from fileManager import *


def clients():
    client = socket()
    client.connect(('10.7.160.92', 1234))
    text_nums = Number()
    picture_nums = Number()
    video_nums = Number()
    while True:
        print('1.接收文字 2.接收?qǐng)D片 3.接收視頻 4.發(fā)送圖片 5.退出')
        message = input('請(qǐng)選擇：')
        client.send(message.encode())
        if message == '1':
            # 接收文字
            all_message = client.recv(1024).decode(encoding='utf-8')
            text_write('./client_files/聊天記錄%d.txt' % text_nums.create_num(), all_message)
            print('文字接收成功！')
        elif message == '2':
            # 接收?qǐng)D片
            picture_data = bytes()
            while True:
                picture_message = client.recv(1024)
                if len(picture_message) < 1024:
                    picture_data += picture_message
                    binary_write('./client_files/圖片%d.jpg' % picture_nums.create_num(), picture_data)
                    print('圖片接收成功！')
                    break
                picture_data += picture_message
        elif message == '3':
            # 接收視頻
            all_data = bytes()
            while True:
                data1 = client.recv(1024)
                if len(data1) < 1024:
                    all_data += data1
                    all_data = binary_write('./client_files/視頻%d.mp4' % video_nums.create_num(), all_data)
                    print('視頻接收成功！')
                    break
                all_data += data1
        elif message == '4':
            # 發(fā)送圖片
            picture_message = binary_read('./server_files./圖片.jpg')
            client.send(picture_message)
            print('圖片發(fā)送成功！')
        elif message == '5':
            break
        else:
            print('輸入錯(cuò)誤！')


if __name__ == '__main__':
    clients()

fileManager.py

class Number:
    """產(chǎn)生序號(hào)"""
    def __init__(self):
        self._nums = 0

    @property
    def nums(self):
        return self._nums

    def create_num(self):
        self._nums += 1
        return self._nums


def text_write(file_path: str, all_message: str):
    """
    寫(xiě)入文本文件
    :param file_path:
    :return: None
    """
    if not all_message:
        return
    with open(file_path, 'a', encoding='utf-8') as f:
        f.write(all_message+'\n')


def binary_read(file_path: str):
    """
    獲得二進(jìn)制文件
    :param file_path:
    :return:
    """
    try:
        with open(file_path, 'rb') as f:
            return f.read()
    except:
        return None


def binary_write(file_path: str, all_message: bytes):
    """
    寫(xiě)入二進(jìn)制文件
    :param file_path:
    :param all_message:
    :return:
    """
    if not all_message:
        return
    with open(file_path, 'wb') as f:
        f.write(all_message)