python socket 网络聊天室_python socket 聊天室

这篇博客分享了如何使用Python的socket库创建一个全双工网络聊天室。通过两个脚本,`chat_server.py`和`chat_client.py`,实现客户端与服务器之间的实时通信。服务器端使用epoll进行事件监听,而客户端则通过epoll处理输入输出事件,确保了聊天室的正常运行。后续计划添加更多功能,如昵称设定和私聊等。
摘要由CSDN通过智能技术生成

http://www.oschina.net/code/snippet_1992504_44586

这位兄台写了个半双工的, 花半小时写了个全双工的

测试了一下,应该实现了基本功能, 后续会添加更多功能,包括取昵称,更改昵称,私聊等

1.[代码]python chat_server

__author__ = '’

import socket

import select

HOST = ""

PORT = 9008

BACKLOG = 100

BUFF = 1024

ADDR = (HOST, PORT)

CONNECT_MAP = {}

def broadcast_msg(_srv_sock, my_sock, msg):

for _file_no, _sock in CONNECT_MAP.iteritems():

if _sock != _srv_sock and _sock != my_sock:

try:

_sock.send(msg)

except socket.error:

pass

def chat_server():

srv_sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

srv_sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)

srv_sock.bind(ADDR)

srv_sock.listen(BACKLOG)

_epoll = select.epoll()

_epoll.register(srv_sock.fileno(), select.EPOLLIN)

print "Start chat server on port %d" % PORT

while True:

events = _epoll.poll(timeout=0.5)

for _file_no, event in events:

if _file_no == srv_sock.fileno():

_cli_sock, _addr = srv_sock.accept()

_cli_sock.setblocking(0)

print "Client (%s, %s) connected" % _addr

broadcast_msg(srv_sock, _cli_sock, "[%s:%s] entered root\n\r" % _addr)

_epoll.register(_cli_sock.fileno(), select.EPOLLIN)

CONNECT_MAP[_cli_sock.fileno()] = _cli_sock

elif event & select.EPOLLIN:

try:

_sock = CONNECT_MAP[_file_no]

_data = _sock.recv(BUFF)

if _data:

_msg = "" + _data + '\n\r'

broadcast_msg(srv_sock, _sock, _msg)

except:

CONNECT_MAP[_file_no].close()

CONNECT_MAP.pop(_file_no)

_epoll.unregister(_file_no)

elif event & select.EPOLLHUP:

CONNECT_MAP[_file_no].close()

CONNECT_MAP.pop(_file_no)

_epoll.unregister(_file_no)

if __name__ == "__main__":

chat_server()

2.[代码]python chat_client

__author__ = ''

import select

import string

import sys

import socket

# default server address

HOST = "localhost"

PORT = 9008

BUFF = 4096

if len(sys.argv) == 3:

HOST = sys.argv[1]

PORT = int(sys.argv[2])

def chat_client():

cli_sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

try:

cli_sock.connect((HOST, PORT))

except:

print "Connect to chat server failed"

sys.exit(-1)

print "succ connected to chat server, now please enter your words"

sys.stdout.write("I>")

sys.stdout.flush()

_epoll = select.epoll()

_epoll.register(sys.stdin.fileno(), select.EPOLLIN)

_epoll.register(cli_sock.fileno(), select.EPOLLIN | select.EPOLLOUT | select.EPOLLERR)

while True:

events = _epoll.poll(0.5)

for _file_no, event in events:

if _file_no == cli_sock.fileno() and event & select.EPOLLIN:

_data = cli_sock.recv(BUFF)

if not _data:

print "Disconnected from chat server"

sys.exit(-1)

else:

sys.stdout.write(_data)

sys.stdout.write("I>")

sys.stdout.flush()

elif event & select.EPOLLIN:

_msg = sys.stdin.readline()

cli_sock.send(_msg)

sys.stdout.write("I>")

sys.stdout.flush()

if __name__ == "__main__":

chat_client()

以下是一个简单的 Python 网络聊天室的代码示例: 服务端代码: ```python import socket import threading HOST = '127.0.0.1' PORT = 65432 server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) server_socket.bind((HOST, PORT)) server_socket.listen() clients = [] def broadcast(message): for client in clients: client.sendall(message.encode()) def handle_client(client_socket, client_addr): clients.append(client_socket) print(f'Connected by {client_addr}') while True: message = client_socket.recv(1024).decode() if not message: break print(f'Received from {client_addr}: {message}') broadcast(f'{client_addr}: {message}') clients.remove(client_socket) client_socket.close() print(f'Connection closed by {client_addr}') while True: client_socket, client_addr = server_socket.accept() client_thread = threading.Thread(target=handle_client, args=(client_socket, client_addr)) client_thread.start() ``` 客户端代码: ```python import socket import threading HOST = '127.0.0.1' PORT = 65432 client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) client_socket.connect((HOST, PORT)) def receive(): while True: message = client_socket.recv(1024).decode() print(message) receive_thread = threading.Thread(target=receive) receive_thread.start() while True: message = input() client_socket.sendall(message.encode()) ``` 这个聊天室采用了基于TCP协议的socket通信,它允许多个客户端连接到服务器并相互通信。服务端在客户端连接时将客户端socket对象添加到一个客户端列表中,然后在收到消息时广播给所有连接的客户端。客户端连接到服务器后,将启动一个接收线程来监听服务器的消息,同时等待用户输入并将消息发送给服务器。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值