Python3 Socket文件传输系统：如何获取用户指定open()使用的路径？

嘿，这个问题其实很好解决——核心就是把硬编码的文件名换成用户输入的完整文件路径，直接传给open()函数就行。我给你分发送端和接收端分别调整代码，你可以直接参考：

首先，咱们可以用input()函数让用户输入要发送的文件路径，另外最好加个简单的检查，确保用户输入的路径真的存在，避免程序报错。

修改后的发送端关键代码大概是这样：

import socket
import os

def send_file():
    # 创建socket连接（这部分你已经实现了，我就不重复写了）
    client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    client_socket.connect(("服务器IP", 端口号))

    # 让用户输入待发送文件的完整路径
    file_path = input("请输入要发送的文件完整路径：")
    
    # 检查文件是否存在
    if not os.path.isfile(file_path):
        print("错误：指定的文件不存在！")
        client_socket.close()
        return
    
    # 获取文件名（用来告诉接收端保存的文件名）
    file_name = os.path.basename(file_path)
    # 发送文件名给接收端
    client_socket.send(file_name.encode())
    
    # 打开指定路径的文件并发送内容
    with open(file_path, "rb") as f:
        while True:
            data = f.read(1024)
            if not data:
                break
            client_socket.send(data)
    
    print("文件发送完成！")
    client_socket.close()

if __name__ == "__main__":
    send_file()

这里的关键点：

• 用os.path.isfile()验证用户输入的是一个存在的文件
• 用os.path.basename()提取文件名，方便接收端知道默认的保存文件名（如果接收端要完全自定义文件名，这部分可以调整）

接收端的情况稍微复杂一点：用户可能输入的是文件夹路径（比如C:\downloads或者/home/user/downloads），也可能直接输入带文件名的完整路径（比如C:\downloads\received_file.txt）。咱们可以做个判断，兼容这两种情况。

修改后的接收端关键代码：

import socket
import os

def receive_file():
    # 创建socket监听（这部分你已经实现了）
    server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server_socket.bind(("本地IP", 端口号))
    server_socket.listen(1)
    print("等待连接...")
    client_socket, addr = server_socket.accept()
    print(f"已连接：{addr}")

    # 接收发送端传来的文件名
    file_name = client_socket.recv(1024).decode()
    
    # 让用户输入保存路径
    save_path = input("请输入文件保存路径（可以是文件夹或带文件名的完整路径）：")
    
    # 判断用户输入的是文件夹还是完整文件路径
    if os.path.isdir(save_path):
        # 如果是文件夹，就用原文件名保存
        full_save_path = os.path.join(save_path, file_name)
    else:
        # 如果是带文件名的路径，直接用这个路径
        full_save_path = save_path
        # 确保保存路径的文件夹存在，不存在就创建
        save_dir = os.path.dirname(full_save_path)
        if not os.path.exists(save_dir):
            os.makedirs(save_dir)
    
    # 开始接收文件并保存到指定路径
    with open(full_save_path, "wb") as f:
        while True:
            data = client_socket.recv(1024)
            if not data:
                break
            f.write(data)
    
    print(f"文件已保存到：{full_save_path}")
    client_socket.close()
    server_socket.close()

if __name__ == "__main__":
    receive_file()

这里的关键点：

• 用os.path.isdir()判断用户输入的是文件夹还是文件路径
• 用os.path.join()拼接文件夹和文件名，避免跨系统的路径格式错误
• 用os.makedirs()自动创建不存在的保存文件夹，防止因为路径不存在导致保存失败

一些小提醒 #

• 路径格式：Windows系统下可以直接输入C:\Users\XXX\file.txt，Python会自动处理反斜杠；Linux/macOS输入/home/XXX/file.txt就行
• 权限问题：如果保存路径是系统目录（比如Windows的C:\Windows），可能会因为权限不足保存失败，记得提醒用户选择有权限的目录
• 扩展功能：你还可以加个文件存在校验，询问用户是否覆盖已存在的文件，这个可以根据需求自行扩展

内容的提问来源于stack exchange，提问作者gman1230321