python网络编程基（4）-tcp案例

tcp实现文件下载客户端

import socket

def main():
	# 创建tcp套接字
	tcp_client_socket = socket.socket(AF_INET, socket.SOCK_STREAM)
	# 获取服务器的ip和port
	server_ip = input("请输入服务器的ip：")
	try:
		server_port = int(input("请输入服务器的port:"))
	except Exception as e:
		print("输入错误")
	# 链接服务器
	tcp_client_socket.connect((ip, port))
	# 获取下载的文件名字
	file_name = input("请输入您要下载的文件名字:")
	# 发送下载请求到客户端
	tcp_client_socket.send(file_name.encode("utf-8")
	# 接收客户端发来的文件
	recv_data = tcp_client_socket.recv(1024)
	# 如果接收到消息，放入本地文件当中
	if(recv_data):
		with open("新"+ file_name, "wb") as f:
			f.write(recv_data)
	# 关闭套接字
	tcp_client_socket.close()


if __name__ == "__main__":
	main()

tcp实现文件下载服务器

import socket

def main():
	# 创建tcp套接字
	tcp_server_socket = socket.socket(socket.AF_INT, socket.SOCK_STREAM)
	# 绑定本地信息
	tcp_server_socket.bind(("", 9000))
	# 变为监听套接字
	tcp_server_socket.listen(128)
	# 为客户端提供服务
	tcp_client_socket, tcp_client_addr = tcp_server_socket.accept()
	# 接收客户端发来的请求
	file_name = tcp_client_socket.recv(1024)
	# 将本地文件发送给客户端
	try:
		f = open(file_name, "rb")
		send_data = f.read()
		f.close()
	except Exception as e:
		print("没有这个文件")
	if send_data:
		tcp_client_socket.send(send_data.encode("utf-8"))
	# 关闭套接字
	tcp_client_socket.close()
	tcp_server_socket.close()

if __name__ == "__main__":
	main()

下面是一段升级代码：

import socket
def send_file_2_client(new_client_socket, client_addr):
    # 1. 接收客户端 需要下载的文件名
    # 接收客户端发送过来的 要下载的文件名
    file_name = new_client_socket.recv(1024).decode("utf-8")
    print("客户端(%s)需要下载文件是：%s" % (str(client_addr), file_name))
    file_content = None
    # 2. 打开这个文件，读取数据
    try:
        f = open(file_name, "rb")
        file_content = f.read()
        f.close()
    except Exception as ret:
        print("没有要下载的文件(%s)" % file_name)


    # 3. 发送文件的数据给客户端
    if file_content:
        # new_client_socket.send("hahahghai-----ok-----".encode("utf-8"))
        new_client_socket.send(file_content)


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

    # 2. 插入手机卡(绑定本地信息 bind)
    tcp_server_socket.bind(("", 7890))

    # 3. 将手机设置为正常的 响铃模式(让默认的套接字由主动变为被动 listen)
    tcp_server_socket.listen(128)

    while True:
        # 4. 等待别人的电话到来(等待客户端的链接 accept)
        new_client_socket, client_addr = tcp_server_socket.accept()

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

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


if __name__ == "__main__":
    main()