add some cases

xianhu · xianhu · commit bba8d082a82f · 2019-09-09T13:35:18.000+08:00
diff --git a/README.md b/README.md
@@ -60,7 +60,12 @@
 
 ### python_redis.py: Python操作Redis实现消息的发布与订阅
 
+### python_schedule.py: Python进行调度开发
+
+### python_socket.py: Python的socket开发实例
+
 ### Plotly目录: 一些plotly画图的实例，使用jupyter notebook编写
+
 ===================================================================================================
 
 ### 您可以fork该项目, 并在修改后提交Pull request, 看到后会尽量进行代码合并
diff --git a/python_socket.py b/python_socket.py
@@ -0,0 +1,76 @@
+# _*_ coding: utf-8 _*_
+
+"""
+Socket编程
+"""
+
+import sys
+import socket
+
+
+def server_func(port):
+    """
+    服务端
+    """
+    # 1. 创建socket对象
+    server = socket.socket()
+
+    # 2. 绑定ip和端口
+    server.bind(("127.0.0.1", port))
+
+    # 3. 监听是否有客户端连接
+    server.listen(10)
+    print("服务端已经启动%s端口......" % port)
+
+    # 4. 接收客户端连接
+    sock_obj, address = server.accept()
+    print("客户端来自于：", address)
+
+    while True:
+        # 5. 接收客户端发送的消息
+        recv_data = sock_obj.recv(1024).decode("utf-8")
+        print("客户端端(%s) -> 服务端: %s" % (address, recv_data))
+        if recv_data == "quit":
+            break
+
+        # 6. 给客户端回复消息
+        send_data = "received[%s]" % recv_data
+        sock_obj.send(send_data.encode("utf-8"))
+        print("服务端 -> 客户端(%s): %s" % (address, send_data))
+
+    # 7. 关闭socket对象
+    sock_obj.close()
+    server.close()
+
+
+def client_func(port):
+    """
+    客户端
+    """
+    # 1. 创建客户端的socket对象
+    client = socket.socket()
+
+    # 2. 连接服务端， 需要指定端口和IP
+    client.connect(("127.0.0.1", port))
+
+    while True:
+        # 3. 给服务端发送数据
+        send_data = input("客户端>").strip()
+        client.send(send_data.encode("utf-8"))
+        if send_data == "quit":
+            break
+    
+        # 4. 获取服务端返回的消息
+        recv_data = client.recv(1024).decode("utf-8")
+        print("服务端 -> 客户端: %s" % recv_data)
+
+    # 5. 关闭socket连接
+    client.close()
+
+
+if __name__ == '__main__':
+    flag = sys.argv[1]
+    if flag == "server":
+        server_func(9901)
+    else:
+        client_func(9901)
