python_socket.py

# _*_ 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()
    sock_obj.settimeout(3)
    print("客户端：%s，超时时间：%s" % (address, sock_obj.gettimeout()))

    while True:
        try:
            # 5. 接收客户端发送的消息
            recv_data = sock_obj.recv(1024).decode("utf-8")
            print("客户端端 -> 服务端: %s" % recv_data)
            if recv_data == "quit":
                break

            # 6. 给客户端回复消息
            send_data = "received[%s]" % recv_data
            sock_obj.send(send_data.encode("utf-8"))
            print("服务端 -> 客户端: %s" % send_data)
        except Exception as excep:
            print("error: ", excep)

    # 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)