Python socket实现简单聊天室_Python

Python socket实现简单聊天室

2021-01-26 00:25萧-十一 Python

这篇文章主要为大家详细介绍了Python socket实现简单聊天室，具有一定的参考价值，感兴趣的小伙伴们可以参考一下

本文实例为大家分享了Python socket实现简单聊天室的具体代码，供大家参考，具体内容如下

服务端使用了select模块，实现了对多个socket的监控。客户端由于select在Windows下只能对socket使用，所以使用了多线程来实现对客户端输入和socket连接的同时监控。注意这里的socket设置为了非阻塞。这样就实现了在一个线程中同时进行socket的接收和发送。

服务器代码：

									# -*- coding: utf-8 -*-

									import socket,select

									connection_list = []

									host = ''

									port = 10001

									def board_cast(sock,message):

									 for socket in connection_list:

									  if socket != server_sock and socket != sock:

									   try:

									    socket.send(message)

									   except:

									    socket.close()

									    print str(socket.getpeername())+' is offline'

									    connection_list.remove(socket)

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

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

									server_sock.setblocking(0)

									server_sock.bind((host,port))

									server_sock.listen(10)

									connection_list.append(server_sock)

									while 1:

									 readable,writable,error = select.select(connection_list,[],[])

									 for sock in readable:

									  if sock == server_sock:

									   connection,connection_add = sock.accept()

									   message = str(connection_add)+'enter room'

									   board_cast(connection,message)

									   print connection_add,'%s connect'

									   connection_list.append(connection)

									  else:

									   try:

									    date = sock.recv(1024)

									    print date

									    board_cast(sock,'('+str(sock.getpeername())+') :'+date)

									   except:

									    message2 = str(sock.getpeername())+ 'is offline'

									    board_cast(sock,message2)

									    print str(sock.getpeername())+ ' is offline'

									    sock.close()

									    connection_list.remove(sock)

									    continue

客户端代码：

									# -*- coding: utf-8 -*-

									import socket,threading,time

									flag = 0

									date = ''

									lock = threading.Lock()

									host = 'localhost'

									port = 10001

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

									client_sock.setblocking(0)

									class Mythread1(threading.Thread):

									 def __init__(self):

									  threading.Thread.__init__(self)

									 def run(self):

									  global flag, date

									  while 1:

									   date = raw_input()

									   if len(date):

									    lock.acquire()

									    flag = 1

									    lock.release()

									class Mythread2(threading.Thread):

									 def __init__(self):

									  threading.Thread.__init__(self)

									 def run(self):

									  global flag

									  global date

									  while 1:

									   try:

									    buf = client_sock.recv(1024)

									    if len(buf):

									     print buf

									   except:

									    pass

									   if flag:

									    try:

									     client_sock.send(date)

									    except socket.error, e:

									     print e

									    lock.acquire()

									    flag = 0

									    lock.release()

									try:

									 client_sock.connect((host,port))

									 print"连接成功"

									except socket.error,e:

									 print e

									t1 = Mythread1()

									t2 = Mythread2()

									t1.start()

									t2.start()