python websockets 客户端与服务端

---

简易 websockets 客户端与服务端

服务端

代码如下（示例）：

import asyncio
import websockets

__wstocken = {}

# 通知给所有的客户端
async def notify_users(msg):
    try:
		for key in __wstocken:
			await  __wstocken[key].send(msg)
    except Exception as e:
		print("error: {}".format(e))

# 检测客户端权限，用户名密码通过才能退出循环
async def check_permit(websocket):
    while True:
        recv_str = await websocket.recv()
        cred_dict = recv_str.split(":")
        if cred_dict[0] == "admin" and cred_dict[1] == "123456":
            response_str = "congratulation, you have connect with server\r\nnow, you can do something else"
            await websocket.send(response_str)
            return True
        else:
            response_str = "sorry, the username or password is wrong, please submit again"
            await websocket.send(response_str)

# 接收客户端消息并处理，这里只是简单把客户端发来的返回回去
async def recv_msg(websocket, path):
    async for message in websocket:
		print(message)
        response_text = f"your submit context: {message}"
        await websocket.send(response_text)


async def register(websocket, path):
    __wstocken[path] = websocket
	if len(__wstocken) > 80:
		print("------waring wsclient to much")
	print(__wstocken)

#需要以websocket作为主键去查找，无法使用path。有path携带错误的问题
async def unregister(websocket, path):
    del __wstocken[path]
	print(__wstocken)


# 服务器端主逻辑
# websocket和path是该函数被回调时自动传过来的，不需要自己传
async def logic(websocket, path):
    await check_permit(websocket)
	await register(websocket, path)
	try：
		await recv_msg(websocket, path)
	except Exception as e:
		print("error: {}".format(e))
	finally:
		await unregister(websocket, path)
		
def wsserver_task(host, port):
	start_server = websockets.server(login, host, port, ping_interval=None)
	asyncio.get_event_loop().run_until_complete(start_server)
	asyncio.get_event_loop().run_forever()
	# 如果要给被回调的logic传递自定义参数，可使用以下形式
	# 修改回调形式
	# import functools
	# start_server = websockets.serve(functools.partial(logic, other_param="test_value"), '10.10.6.91', 5678)
	# 修改被回调函数定义，增加相应参数
	# async def logic(websocket, path, other_param)
	
if __name__ == "__main__":
	wsserver_task("192.168.1.30", 5678)

客户端

代码如下（示例）：

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()
        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://192.168.1.30:5678', ping_interval=None) as websocket:
        await auth_system(websocket)

        await send_msg(websocket)

asyncio.get_event_loop().run_until_complete(main_logic())