python 实现实时聊天
Python3 socket编程,并与多线程实现最简单的聊天工具之一
socket通讯必须有服务端和客户端
创建服务端:
【第一步】:先创建一个socket类型的对象s:
s=socket.socket(familly,type)
family参数可以是AF_UNIX(Unix域,用于同一台机器上的进程间通讯),也可以是AF_INET(用于IPV4协议的TCP和 UDP)。
type参数一般为SOCK_STREAM(流套接字)或者 SOCK_DGRAM(数据报文套接字),很少用SOCK_RAW(raw套接字)。可以简单的认为:
SOCK_STREAM :能确保数据达到,用于发送文件数据。
SOCK_DGRAM :不能确保数据到达,用于局域网广播消息。
SOCK_RAW :需要自定义IP包,暂不解释。
其实后面还有参数proto用来指明要接收的协议包?fileno参数大概是内存地址一类的东西?默认都不填。
【第二步】:调用socket对象里的bind方法绑定IP和端口:
s.bind((host,port))
bind只有一个tuple(元组)类型的参数,host为客户端的ip,0.0.0.0表示任何ip,本地测试也可以用127.0.0.1,端口最好大于1024。
【第三步】:设置监听数:
s.listen( )
参数填最大监听的连接数。
【第四步】:accept方法接收客户端的连接:
connect , address = s.accept( )
无参数,程序进入阻塞模式直到有客户端连接,接收到连接后会返回一个元组形式的参数:connect为新产生的socket对象,address为客户端的ip。
【第五步】:收发数据及关闭连接:
必须使accept返回的新socket对象connect来收发数据,connect里的send和recv方法
connect.send(" string ".encode('utf8'))
connect.recv(size).decode('utf8')
send方法在Python2里参数是字符串,在Python3里参数是二进制对象,所以必须先用encode('utf8')将字符串重新编码
recv方法参数为读取的缓冲区大小size,与send方法同理,接收到数据后必须用decode('utf8')将数据解码成可读的字符串
最后会话结束s.close()关闭连接
创建客户端:
【第一步】:先创建一个socket类型的对象s:
s=socket.socket(familly,type)
与服务端同理
【第二步】:连接服务端:
s.connect((host,port))
host为服务端主机的ip,端口为服务端定义的端口。
【第三步】:收发数据及关闭连接:
与服务端同理的send和recv方法
s.send()、s.recv()、s.close()
具体代码(必须在CMD里运行,每次服务端或客户端仅能发送一条信息,if内代码可精简):
服务端:
-
import socket
-
s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
-
s.bind(("127.0.0.1",1234))
-
s.listen(2)
-
sock,addr=s.accept()
-
while True:
-
t=sock.recv(1024).decode('utf8') #服务端先接收信息
-
if t == "exit":
-
break
-
print(t)
-
t=input()
-
if t == "exit":
-
break
-
sock.send(t.encode('utf8'))
-
s.close()
客户端:
-
import socket
-
s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
-
s.connect(("127.0.0.1",1234))
-
while True:
-
t=input()
-
s.send(t.encode('utf8')) #客户端先发送信息
-
if t == "exit":
-
break
-
t=s.recv(1024).decode("utf8")
-
if t == "exit":
-
break
-
print(t)
-
s.close()
由于input函数的阻塞作用,以上的代码发完一条信息,只能等待另一端的信息发过来才能继续发送。
这时就要考虑将输入与接收分开来,将接收的函数(或方法)从主线程里抓出来丢到另一个线程里单独运行,为实现这一功能,必须引入多线程。
多线程的使用别人的教程写得都太杂乱,什么select都来了……,其实很简单,Python里两句话搞定,需要import threading
接收的方法为recv,就把s.recv()写到其他函数里,然后主函数里让它自己跑起来:
trd=threading.Thread(target=rec,args=(sock,))
trd.start()
target参数为需要跑起来的函数名,仅函数名,不需要括号,args为传递到target函数里的参数(元组类型),这里仅传入收发数据用的socket对象即可
改良后的代码(须在CMD里运行):
服务端:
-
import socket
-
import threading
-
s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
-
s.bind(("127.0.0.1",9999))
-
s.listen(2)
-
sock,addr=s.accept()
-
true=True
-
def rec(sock):
-
global true
-
while true:
-
t=sock.recv(1024).decode('utf8') #函数的核心语句就一条接收方法
-
if t == "exit":
-
true=False
-
print(t)
-
trd=threading.Thread(target=rec,args=(sock,))
-
trd.start()
-
while true:
-
t=input()
-
sock.send(t.encode('utf8'))
-
if t == "exit":
-
true=False
-
s.close()
客户端:
-
import socket
-
import threading
-
s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
-
s.connect(("127.0.0.1",9999))
-
true=True
-
def rec(s):
-
global true
-
while true:
-
t=s.recv(1024).decode("utf8") #客户端也同理
-
if t == "exit":
-
true=False
-
print(t)
-
trd=threading.Thread(target=rec,args=(s,))
-
trd.start()
-
while true:
-
t=input()
-
s.send(t.encode('utf8'))
-
if t == "exit":
-
true=False
-
s.close()