TCP案例

案例一：循环为客户服务

import socket


def main():
    # 1.创建套接字
    tcp_client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    # 2.绑定本地信息
    tcp_client_socket.bind(("", 7890))

    # 3.listen使套接字变为被动链接
    tcp_client_socket.listen(128)

    while True:
        # 4.监听
        print('等待一个新的客户端到来...')
        new_tcp_socket, socket_addr = tcp_client_socket.accept()
        print('一个新的客户端已经到了{}'.format(socket_addr))

        while True:
            # 5.接收发送数据
            recv_data = new_tcp_socket.recv(1024)
            print('接收的数据是{},'.format(recv_data.decode('gbk')))
            if recv_data:
                # 回送一部分数据给客户端
                new_tcp_socket.send('Hello'.encode('gbk'))
            else:
                break
        # 6，关闭套接字
        # 关闭accept返回的套接字，意味着不会再为这个客户端服务
        new_tcp_socket.close()
        print('已经服务完毕！')
    # 如果将监听套接字关闭了，那么会导致不能再次等待新的客户端的到来即xxx.accept就会失败
    tcp_client_socket.close()


if __name__ == "__main__":
    main()

 

案例二：下载文件

（1）客户端

import socket


def main():
    # 1.创建套接字
    tcp_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    # 2.获取服务器的ip和端口
    dest_ip = input('请输入下载服务器的ip：')
    try:
        dest_port = int(input('请输入下载服务器的port：'))
    except:
        print('输入的端口号有误，请重新输入：')
        dest_port = int(input('请输入下载服务器的port：'))

    # 3.连接服务器
    tcp_socket.connect((dest_ip, dest_port))

    # 4.获取下载文件名称
    download_file_name = input('请输入需要下载的文件名称：')

    # 5.将文件名字发送到服务器
    tcp_socket.send(download_file_name.encode('gbk'))

    # 6.接收文件的数据
    recv_data = tcp_socket.recv(1024 * 1024)

    # 7.保存接收的数据到一个文件中
    if recv_data:
        with open('[接收]'+download_file_name, 'wb') as f:
            f.write(recv_data)

    # 8.关闭套接字
    tcp_socket.close()


if __name__ == '__main__':
    main()

（2）服务端

# -*- coding: utf-8 -*-
import socket


def send_file_2_client(new_client_socket, client_addr):
    # 5.接收客户端发过来的要下载的文件名
    file_name = new_client_socket.recv(1024)
    print('客户端{}需要下载的文件是{}'.format(str(client_addr), file_name))

    # 6.打开文件，读取数据
    file_content = None
    try:
        f = open(file_name,'rb')
        file_content = f.read()
        f.close()
    except Exception as ret:
        print('没有要下载的文件{}'.format(file_name))

    # 7.发送数据给客户端
    if file_content:
       new_client_socket.send(file_content)

def main():
    # 1.创建套接字
    tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    # 2.绑定本地信息
    tcp_server_socket.bind(("", 7800))

    # 3.让默认的套接字有主动变为被动listen
    tcp_server_socket.listen(128)

    # 4.等待客服端的连接
    new_client_socket, client_addr = tcp_server_socket.accept()

    while True:
        # 调用发送文件函数，完成为客户端服务
        send_file_2_client(new_client_socket, client_addr)

    # 7.关闭套接字
    new_client_socket.close()
    tcp_server_socket.close()


if __name__ == '__main__':
    main()