Сетевой чат TCP/UDP

import socket
import threading
 
# Определяем хост и порт для соединения
HOST = 'localhost'
PORT = 5555
 
# Создаем сокет
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
 
# Связываем сокет с хостом и портом
server.bind((HOST, PORT))
 
# Слушаем входящие соединения
server.listen()
 
# Список клиентов и их идентификаторов
clients = {}
client_ids = 0
 
# Функция для обработки входящих сообщений от клиентов
def handle_client(client_socket, client_address):
    global clients, client_ids
 
    # Присваиваем идентификатор клиенту
    client_id = client_ids
    client_ids += 1
 
    # Добавляем клиента в список
    clients[client_id] = client_socket
 
    # Отправляем клиенту его идентификатор
    client_socket.send(str(client_id).encode())
 
    # Принимаем сообщения от клиента и пересылаем их другим клиентам
    while True:
        try:
            message = client_socket.recv(1024)
            if message:
                message_str = message.decode()
                print(f"Received message from {client_id}: {message_str}")
 
                # Отправляем сообщение всем клиентам, кроме отправителя
                for id, socket in clients.items():
                    if id != client_id:
                        socket.send(message)
            else:
                # Если клиент отключился, удаляем его из списка
                client_socket.close()
                del clients[client_id]
                print(f"Client {client_id} disconnected")
                break
        except:
            # Если произошла ошибка, удаляем клиента из списка
            client_socket.close()
            del clients[client_id]
            print(f"Client {client_id} disconnected")
            break
 
 
# Функция для запуска сервера
def start_server():
    print(f"Server started on {HOST}:{PORT}")
    while True:
        # Принимаем входящие соединения и запускаем новый поток для обработки каждого клиента
        client_socket, client_address = server.accept()
        print(f"New client connected: {client_address}")
 
        client_thread = threading.Thread(target=handle_client, args=(client_socket, client_address))
        client_thread.start()
 
 
if __name__ == '__main__':
    start_server()

import socket
import threading
 
# Определяем хост и порт для соединения  
HOST = 'localhost'
PORT = 5555
 
# Выбираем протокол (TCP или UDP)
protocol = input("Choose protocol (TCP/UDP): ").lower()
 
# Создаем сокет
if protocol == "tcp":
    client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
elif protocol == "udp":
    client_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
else:
    print("Invalid protocol")
    exit()
 
# Функция для отправки сообщений
def send_message():
    while True:
        message = input()
        # Добавляем к сообщению идентификатор клиента
        message_with_id = f"{client_id}: {message}"
        if protocol == "tcp":
            client_socket.send(message_with_id.encode())
        elif protocol == "udp":
            client_socket.sendto(message_with_id.encode(), (HOST, PORT))
 
# Функция для приема сообщений
def receive_message():
    while True:
        if protocol == "tcp":
            message = client_socket.recv(1024).decode()
        elif protocol == "udp":
            data, address = client_socket.recvfrom(1024)
            if data:
                message = data.decode()
            else:
                continue
        print(message)
 
# Подключаемся к серверу
if protocol == "tcp":
    client_socket.connect((HOST, PORT))
    # Получаем идентификатор клиента от сервера
    client_id = int(client_socket.recv(1024).decode())
    print(f"You are client {client_id}")
elif protocol == "udp":
    # Отправляем пустое сообщение, чтобы получить идентификатор клиента от сервера
    client_socket.sendto(b"", (HOST, PORT))
    data, address = client_socket.recvfrom(1024)
    if data:
        client_id = int(data.decode())
        print(f"You are client {client_id}")
    else:
        print("Error getting client ID from server")
        exit()
 
 
# Запускаем потоки для отправки и приема сообщений
send_thread = threading.Thread(target=send_message)
send_thread.start()
 
receive_thread = threading.Thread(target=receive_message)
receive_thread.start()

import socket
import threading
 
# Определяем хост и порты для соединения
HOST = 'localhost'
TCP_PORT = 5555
UDP_PORT = 3334
 
# Создаем сокеты
tcp_server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
udp_server = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
 
# Связываем сокеты с хостом и портами
tcp_server.bind((HOST, TCP_PORT))
udp_server.bind((HOST, UDP_PORT))
 
# Слушаем входящие соединения
tcp_server.listen()
 
# Список клиентов и их идентификаторов для TCP сокета
tcp_clients = {}
tcp_client_ids = 0
 
# Функция для обработки входящих сообщений от TCP клиентов
def handle_tcp_client(client_socket, client_address):
    global tcp_clients, tcp_client_ids
 
    # Присваиваем идентификатор клиенту
    client_id = tcp_client_ids
    tcp_client_ids += 1
 
    # Добавляем клиента в список
    tcp_clients[client_id] = client_socket
 
    # Отправляем клиенту его идентификатор
    client_socket.send(str(client_id).encode())
 
    # Принимаем сообщения от клиента и пересылаем их другим клиентам
    while True:
        try:
            message = client_socket.recv(1024)
            if message:
                message_str = message.decode()
                print(f"Received message from {client_id}: {message_str}")
 
                # Отправляем сообщение всем клиентам, кроме отправителя
                for id, socket in tcp_clients.items():
                    if id != client_id:
                        socket.send(message)
            else:
                # Если клиент отключился, удаляем его из списка
                client_socket.close()
                del tcp_clients[client_id]
                print(f"TCP client {client_id} disconnected")
                break
        except:
            # Если произошла ошибка, удаляем клиента из списка
            client_socket.close()
            del tcp_clients[client_id]
            print(f"TCP client {client_id} disconnected")
            break
 
# Функция для обработки входящих сообщений от UDP клиентов
def handle_udp_client():
    while True:
        data, address = udp_server.recvfrom(1024)
        if data:
            message = data.decode()
            print(f"Received message from {address}: {message}")
 
            # Отправляем сообщение всем клиентам, включая отправителя
            for id, socket in tcp_clients.items():
                socket.send(message.encode())
 
# Функция для запуска сервера
def start_server():
    print(f"Server started on {HOST}:{TCP_PORT} (TCP) and {HOST}:{UDP_PORT} (UDP)")
    # Запускаем поток для обработки входящих сообщений от UDP клиентов
    udp_thread = threading.Thread(target=handle_udp_client)
    udp_thread.start()
 
    while True:
        # Принимаем входящие соединения и запускаем новый поток для обработки каждого TCP клиента
        tcp_client_socket, tcp_client_address = tcp_server.accept()
        print(f"New TCP client connected: {tcp_client_address}")
 
        tcp_client_thread = threading.Thread(target=handle_tcp_client, args=(tcp_client_socket, tcp_client_address))
        tcp_client_thread.start()
 
if __name__ == '__main__':
    start_server()

import socket
import threading
 
# Определяем хост и порты для соединения  
HOST = 'localhost'
TCP_PORT =5555
UDP_PORT = 3334
 
# Выбираем протокол (TCP или UDP)
protocol = input("Choose protocol (TCP/UDP): ").lower()
 
# Создаем сокет
if protocol == "tcp":
    client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    port = TCP_PORT
elif protocol == "udp":
    client_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    port = UDP_PORT
else:
    print("Invalid protocol")
    exit()
 
# Функция для отправки сообщений
def send_message():
    while True:
        message = input()
        # Добавляем к сообщению идентификатор клиента
        message_with_id = f"{client_id}: {message}"
        if protocol == "tcp":
            client_socket.send(message_with_id.encode())
        elif protocol == "udp":
            client_socket.sendto(message_with_id.encode(), (HOST, port))
 
# Функция для приема сообщений
def receive_message():
    while True:
        if protocol == "tcp":
            message = client_socket.recv(1024).decode()
        elif protocol == "udp":
            data, address = client_socket.recvfrom(1024)
            if data:
                message = data.decode()
            else:
                continue
        print(message)
 
# Подключаемся к серверу
if protocol == "tcp":
    client_socket.connect((HOST, port))
    # Получаем идентификатор клиента от сервера
    client_id = int(client_socket.recv(1024).decode())
    print(f"You are client {client_id}")
elif protocol == "udp":
    # Отправляем пустое сообщение, чтобы получить идентификатор клиента от сервера
    client_socket.sendto(b"", (HOST, port))
    data, address = client_socket.recvfrom(1024)
    if data:
        client_id = int(data.decode())
        print(f"You are client {client_id}")
    else:
        print("Error getting client ID from server")
        exit()
 
# Запускаем потоки для отправки и приема сообщений
send_thread = threading.Thread(target=send_message)
send_thread.start()
 
receive_thread = threading.Thread(target=receive_message)
receive_thread.start()