版本:Python3
知识点:Socket,threading
一、服务器端
1,服务器绑定地址与端口号,保持监听状态。
2,无限循环接收客户端的连接,验证身份判断是否是客户端,如果不是立即关闭连接,如果是则创建一个新线程管理对客户端的所有操作,并将其加入到连接池。
3,当客户端主动终止连接时,通知其他用户其已离开聊天室,关闭连接,从连接池里将其剔除。
代码:
# encoding: utf-8
# Author: Timeashore
# Time: 2017-12-31
# Email: 1274866364@qq.com
# 服务器端
import threading
import socket
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.bind(('127.0.0.1',5000))
s.listen(5)
print('Server',socket.gethostbyname('127.0.0.1'),'listening ...')
mydict = dict() # fileno:nickname
mylist = list() # 连接池列表
# 向在线人发送消息(除了发消息本人)
def send_everybody(number, message):
for user in mylist:
if user.fileno() != number:
user.send(message.encode())
# 每个线程的调用方法,管理对客户端的所有操作
def fun(conn, number):
mylist.append(conn) # 添加到连接池
nickname = conn.recv(1024).decode()
conn.send(('Hello,'+nickname).encode())
mydict[number] = nickname # 添加到fileno:nickname
print('connection',number,'has nickname : ',nickname) # 打印链接信息
send_everybody(number, '【系统提示:'+nickname+' 进入聊天室】')
while True:
try:
conn.send(''.encode())
getmessage = conn.recv(1024).decode()
if getmessage:
print(nickname,':',getmessage)
send_everybody(number,nickname+': '+getmessage)
except (OSError, ConnectionResetError):
try:
mylist.remove(conn) # 从连接池列表剔除
except:
pass
print(nickname, 'exit, ', len(mylist), ' person left') # 服务器终端打印离开信息
send_everybody(number, '【系统提示:' + nickname + ' 离开聊天室】') # 告知所有人nickname已经离开聊天室
conn.close() # 关闭连接
return
while True:
conn, addr = s.accept()
print('Accept new connection',conn.getsockname(),conn.fileno())
try:
verification = conn.recv(1024).decode()
if verification == 'user':
conn.send('welcome to server!\n'.encode())
conn.send('nickname:\n'.encode())
t = threading.Thread(target=fun, args=(conn, conn.fileno()))
t.setDaemon(True)
t.start()
else:
conn.send('Your verification NOT pass'.encode())
conn.close()
except:
pass
二、客户端
与服务器端相比,客户端比较简单。
1,与服务器建立连接,发送身份验证信息。
2,启动两个线程,一个发送消息,一个接收消息。
代码:
# encoding: utf-8
# Author: Timeashore
# Time: 2017-12-31
# Email: 1274866364@qq.com
import socket
import threading
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.connect(('127.0.0.1',5000))
s.send('user'.encode()) # 验证身份
print(s.recv(1024).decode()) # 接受欢迎消息
print(s.recv(1024).decode()) # 接受欢迎消息
nickname = input()
s.send(nickname.encode())
# 接收消息
def recv():
while True:
try:
message = s.recv(1024).decode()
if message:
print(message)
else:
pass
except ConnectionAbortedError:
print('Server closed this connection!')
except ConnectionResetError:
print('Server is closed!')
# 发送消息
def send():
while True:
try:
ready_send = input('')
s.send(ready_send.encode())
except ConnectionAbortedError:
print('Server closed this connection!')
except ConnectionResetError:
print('Server is closed!')
# 启动两个线程,一个发送一个接收
t1 = threading.Thread(target=recv)
t2 = threading.Thread(target=send)
for x in [t1,t2]:
x.setDaemon(True)
x.start()
t1.join()
t2.join()
以上是在终端运行,可以使用Python GUI工具 Tkinter 给客户端增加一个界面,客户端收到的消息实时显示在界面上,需要处理线程和GUI的配合。