pysyft 博客看完

2026-02-03 18:44:19 +08:00 · 2021-05-09 16:21:02 +08:00
parent f55d48db3f
commit 4df1fa4d7c
13 changed files with 288 additions and 163 deletions
--- a/Python/web_socket_test/client.py
+++ b/Python/web_socket_test/client.py
@@ -0,0 +1,35 @@
+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()
--- a/Python/web_socket_test/index.html
+++ b/Python/web_socket_test/index.html
@@ -0,0 +1,24 @@
+<!DOCTYPE html>
+<html>
+<head>
+    <title>WebSocket Test</title>
+</head>
+<body>
+<script>
+    var wsObj = new WebSocket("ws://127.0.0.1:5678");   //建立连接
+    wsObj.onopen = function(){  //发送请求
+        alert("open");
+        wsObj.send("admin:123456");
+    };
+    wsObj.onmessage = function(ev){  //获取后端响应
+        alert(ev.data);
+    };
+    wsObj.onclose = function(ev){
+        alert("close");
+    };
+    wsObj.onerror = function(ev){
+        alert("error");
+    };
+</script>
+</body>
+</html>
--- a/Python/web_socket_test/server.py
+++ b/Python/web_socket_test/server.py
@@ -0,0 +1,42 @@
+import asyncio
+import websockets
+
+# 检测客户端权限，用户名密码通过才能退出循环
+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):
+    while True:
+        recv_text = await websocket.recv()
+        print("recv_mesage:",recv_text)
+        response_text = f"your submit context: {recv_text}"
+        await websocket.send(response_text)
+
+# 服务器端主逻辑
+# websocket和path是该函数被回调时自动传过来的，不需要自己传
+async def main_logic(websocket, path):
+    await check_permit(websocket)
+
+    await recv_msg(websocket)
+
+# 把ip换成自己本地的ip
+start_server = websockets.serve(main_logic, '127.0.0.1', 5678)
+# 如果要给被回调的main_logic传递自定义参数，可使用以下形式
+# 一、修改回调形式
+# import functools
+# start_server = websockets.serve(functools.partial(main_logic, other_param="test_value"), '10.10.6.91', 5678)
+# 修改被回调函数定义，增加相应参数
+# async def main_logic(websocket, path, other_param)
+
+asyncio.get_event_loop().run_until_complete(start_server)
+asyncio.get_event_loop().run_forever()