最近看到Python黑帽子socket模块,看到网上很多人利用它写了一个简单的聊天小程序,我也去试了下,但是发现一个bug,当client端在打字的时候,若server端发出一个数据包,client端就会自动打印出来(我用了threading模块),导致client端在命令行的输入到一半就断行了,强迫症表示很不爽。所以我写了一个python小程序解决。
源码奉上
chat_client.py
import socket
import threading
import time
import sys
target_host=“192.168.2.140”
target_port=5555
client=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
client.connect((target_host,target_port))
print ("[*]Successful Connection")
def server_handle(client):
while True:
response=""
data=""
while True:
data+=bytes.decode(client.recv(4096))
recv_len=len(data)
response+=data
if recv_len<4096:
break
print (“djb1:”+response)
server_handler=threading.Thread(target=server_handle,args=(client,))
server_handler.start()
while True:
files=open(“enter.txt”,“r”)
buffer_check=files.read()
if len(buffer_check)0:
pass
else:
if buffer_check"exit":
client.send(str.encode(“see you!”))
client.close()
sys.exit(0)
print(“WZ:>”+buffer_check)
time.sleep(0.2)
client.send(str.encode(buffer_check))
enter.py
import time
while True:
data=input(“DJB:”)
file1=open(“enter.txt”,“w”)
file1.write(data)
file1.close()
time.sleep(0.1)
file2=open(“enter.txt”,“w”)
file2.close()
原理其实很简单,用enter程序输出数据,用chat_client来打印,enter将数据储存在enter.txt中,chat_client导出数据,为了避免多次导出导致多次打印,设置enter程序在储存后0.1秒将数据删除,而chat_client在导出数据后会暂停0.1秒,在chat_client导出数据并再次检查enter.txt是否含有数据时,enter程序已经删除了数据。
很弱的一个想法,希望看到的大佬可以指点一下