快捷導(dǎo)航

Python開發(fā)一個功能齊全的IM聊天工具（附實例代碼）

更新時間：2024年12月28日 15:55:40 作者：zhh157

即時通訊（IM）工具現(xiàn)在已經(jīng)很常見了,從簡單的文本聊天到文件傳輸、音視頻通話,IM 工具功能豐富,那么,本文使用Python開發(fā)一個基礎(chǔ)的IM聊天工具,包括：客戶端和服務(wù)端架構(gòu)、實時消息發(fā)送與接收、多用戶聊天支持、一個簡單的圖形用戶界面（GUI）

1. IM 工具的架構(gòu)設(shè)計

IM 工具通常使用客戶端-服務(wù)器架構(gòu)。其基本邏輯如下：

服務(wù)端：負(fù)責(zé)管理用戶的連接，轉(zhuǎn)發(fā)消息到指定用戶，確保消息的實時性和可靠性。
客戶端：負(fù)責(zé)用戶輸入和消息發(fā)送，同時接收來自服務(wù)端的消息并展示給用戶。

我們將采用 Python 的 socket 模塊構(gòu)建網(wǎng)絡(luò)通信，使用 threading 實現(xiàn)多線程來處理多個用戶的連接，并使用 tkinter 開發(fā)客戶端的圖形界面。

2. 服務(wù)端的實現(xiàn)

服務(wù)端負(fù)責(zé)監(jiān)聽客戶端連接，接收和轉(zhuǎn)發(fā)消息。我們將創(chuàng)建一個簡單的服務(wù)端程序：

服務(wù)端代碼

import socket
import threading

# 保存已連接的客戶端
clients = []

# 處理客戶端消息
def handle_client(client_socket, client_address):
    print(f"[連接成功] {client_address}")
    while True:
        try:
            # 接收消息
            message = client_socket.recv(1024).decode('utf-8')
            if message:
                print(f"[收到消息] {client_address}: {message}")
                # 廣播消息給其他客戶端
                broadcast(message, client_socket)
            else:
                remove_client(client_socket)
                break
        except:
            remove_client(client_socket)
            break

# 廣播消息
def broadcast(message, sender_socket):
    for client in clients:
        if client != sender_socket:
            try:
                client.send(message.encode('utf-8'))
            except:
                remove_client(client)

# 移除客戶端
def remove_client(client_socket):
    if client_socket in clients:
        clients.remove(client_socket)
        print(f"[斷開連接] {client_socket.getpeername()}")

# 服務(wù)端主程序
def server_program():
    server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server.bind(('0.0.0.0', 12345))  # 監(jiān)聽所有 IP，端口號 12345
    server.listen(5)
    print("[服務(wù)器啟動] 正在監(jiān)聽連接...")

    while True:
        client_socket, client_address = server.accept()
        clients.append(client_socket)
        threading.Thread(target=handle_client, args=(client_socket, client_address)).start()

if __name__ == "__main__":
    server_program()

代碼解析

監(jiān)聽連接：服務(wù)端通過 server.bind() 和 server.listen() 來監(jiān)聽客戶端的連接請求。
多線程處理：每個連接都會啟動一個新的線程來處理該客戶端的消息，確保服務(wù)端能同時處理多個用戶。
消息廣播：使用 broadcast() 將消息轉(zhuǎn)發(fā)給其他連接的客戶端。
客戶端管理：通過 clients 列表存儲當(dāng)前所有連接的客戶端，并在連接斷開時移除。

3. 客戶端的實現(xiàn)

客戶端需要能夠發(fā)送消息給服務(wù)端，同時接收來自服務(wù)端的廣播消息。我們還需要一個簡單的 GUI 來提升用戶體驗。

客戶端代碼

import socket
import threading
import tkinter as tk
from tkinter import scrolledtext

class ChatClient:
    def __init__(self, host='127.0.0.1', port=12345):
        self.client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.client_socket.connect((host, port))

        # 創(chuàng)建主窗口
        self.root = tk.Tk()
        self.root.title("IM 聊天工具")
        self.root.geometry("400x500")

        # 消息顯示區(qū)域
        self.chat_area = scrolledtext.ScrolledText(self.root, wrap=tk.WORD, state='disabled')
        self.chat_area.pack(padx=10, pady=10, fill=tk.BOTH, expand=True)

        # 消息輸入框
        self.message_entry = tk.Entry(self.root)
        self.message_entry.pack(padx=10, pady=5, fill=tk.X)
        self.message_entry.bind("<Return>", self.send_message)

        # 發(fā)送按鈕
        self.send_button = tk.Button(self.root, text="發(fā)送", command=self.send_message)
        self.send_button.pack(padx=10, pady=5)

        # 啟動接收線程
        threading.Thread(target=self.receive_messages, daemon=True).start()

        # 運行主窗口
        self.root.mainloop()

    def send_message(self, event=None):
        message = self.message_entry.get()
        if message:
            self.client_socket.send(message.encode('utf-8'))
            self.message_entry.delete(0, tk.END)

    def receive_messages(self):
        while True:
            try:
                message = self.client_socket.recv(1024).decode('utf-8')
                self.display_message(message)
            except:
                break

    def display_message(self, message):
        self.chat_area.config(state='normal')
        self.chat_area.insert(tk.END, message + '\n')
        self.chat_area.yview(tk.END)
        self.chat_area.config(state='disabled')

if __name__ == "__main__":
    ChatClient()