使用两个变量删除文件夹中的文件夹

时间:2012-01-27 09:21:20

标签: python

我在公司文件夹中有多个分支,然后每个分支有多个用户。我需要在离开分支时删除某些用户。我没有收到错误,但整个分支文件夹被删除,而不仅仅是指定的用户文件夹。有人能指出我正确的方向

#!/usr/bin/python
import os, sys, getopt, shutil
branch, user = "",""

if __name__ == "__main__":
    argv = sys.argv[1:]
    try:
       opts, args = getopt.getopt(argv, 'hd:e', ['help', 'branch=', 'user='])
    except getopt.GetoptError:
       print __doc__
       sys.exit()

    for opt, arg in opts:
       if opt in ('-h', '--help'):
           print __doc__
           sys.exit()
       if opt in ('-d', '--branch'):
           branch = arg
       if opt in ('-e', '--user'):
           user = arg

shutil.rmtree("/home/company/%s/%s" % (branch , user))

1 个答案:

答案 0 :(得分:0)

  • 看起来您的rmtree()来电错误,它应该只能在“”模式下运行。按原样,它将始终在加载模块时运行,branchuser都为空。这似乎很危险。

  • 在致电branch之前,您似乎应该检查选项中是否确实同时拥有userrmtree()