# _*_ 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(f"服务端已经启动{port}端口......")

    # 4. 接收客户端连接
    sock_obj, address = server.accept()
    sock_obj.settimeout(3)
    print(f"客户端：{address}，超时时间：{sock_obj.gettimeout()}")

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

            # 6. 给客户端回复消息
            send_data = f"received[{recv_data}]"
            sock_obj.send(send_data.encode("utf-8"))
            print(f"服务端 -> 客户端: {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(f"服务端 -> 客户端: {recv_data}")

    # 5. 关闭socket连接
    client.close()


if __name__ == '__main__':
    flag = sys.argv[1]
    if flag == "server":
        server_func(9901)
    else:
        client_func(9901)