Question

我和这个问题类似。

How to copy a file from a network share to local disk with variables?

唯一的区别是我的网络驱动器有用户名和密码保护密码。

我需要使用Samba将文件复制到Python份额并验证。

如果我手动登录，那么代码可以正常工作，但是没有登录shutil命令就不起作用。

Answer 1

我尝试使用NET USE调用os.system命令将共享映射到未使用的驱动器号（假设您在Windows上）：

os.system(r"NET USE P: \\ComputerName\ShareName %s /USER:%s\%s" % (password, domain_name, user_name))

将共享映射到驱动器号后，可以使用shutil.copyfile将文件复制到给定的驱动器。最后，您应该卸载共享：

os.system(r"NET USE P: /DELETE")

当然这仅适用于Windows，您必须确保驱动器号P可用。您可以检查NET USE命令的返回码以查看挂载是否成功;如果没有，你可以尝试不同的驱动器号，直到你成功。

由于两个NET USE命令成对出现，第二个命令应始终在执行第一个时执行（即使在其间的某处引发了异常），您可以将这两个调用包装在上下文中管理员，如果您使用的是Python 2.5或更高版本：

from contextlib import contextmanager

@contextmanager
def network_share_auth(share, username=None, password=None, drive_letter='P'):
    """Context manager that mounts the given share using the given
    username and password to the given drive letter when entering
    the context and unmounts it when exiting."""
    cmd_parts = ["NET USE %s: %s" % (drive_letter, share)]
    if password:
        cmd_parts.append(password)
    if username:
        cmd_parts.append("/USER:%s" % username)
    os.system(" ".join(cmd_parts))
    try:
        yield
    finally:
        os.system("NET USE %s: /DELETE" % drive_letter)

with network_share_auth(r"\\ComputerName\ShareName", username, password):
     shutil.copyfile("foo.txt", r"P:\foo.txt")

Answer 2

如果您有pywin32库（例如，它是ActiveState Python发行版的一部分），那么您可以在几行中完成它，而无需映射驱动器：

import win32wnet
win32wnet.WNetAddConnection2(0, None, '\\\\'+host, None, username, password)
shutil.copy(source_file, '\\\\'+host+dest_share_path+'\\')
win32wnet.WNetCancelConnection2('\\\\'+host, 0, 0) # optional disconnect

有a more complete example on ActiveState Code

如何使用Python将文件复制到网络路径或驱动器

2 个答案: