mirror of
https://github.com/Estom/notes.git
synced 2026-02-03 10:33:35 +08:00
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
import asyncio
|
|
import websockets
|
|
|
|
# 向服务器端认证，用户名密码通过才能退出循环
|
|
async def auth_system(websocket):
|
|
while True:
|
|
cred_text = input("please enter your username and password: ")
|
|
await websocket.send(cred_text)
|
|
response_str = await websocket.recv()
|
|
print("receive_message",response_str)
|
|
if "congratulation" in response_str:
|
|
return True
|
|
|
|
# 向服务器端发送认证后的消息
|
|
async def send_msg(websocket):
|
|
while True:
|
|
_text = input("please enter your context: ")
|
|
if _text == "exit":
|
|
print(f'you have enter "exit", goodbye')
|
|
await websocket.close(reason="user exit")
|
|
return False
|
|
await websocket.send(_text)
|
|
recv_text = await websocket.recv()
|
|
print(f"{recv_text}")
|
|
|
|
# 客户端主逻辑
|
|
async def main_logic():
|
|
async with websockets.connect('ws://127.0.0.1:5678') as websocket:
|
|
await auth_system(websocket)
|
|
|
|
await send_msg(websocket)
|
|
|
|
asyncio.get_event_loop().run_until_complete(main_logic())
|
|
|
|
asyncio.get_event_loop().run_forever() |