scripts/rm.py

   1 #! /usr/bin/env python
   2
   3 import argparse
   4 import os
   5 import shutil
   6
   7 if __name__ == '__main__':
   8     parser = argparse.ArgumentParser(description='Remove old files')
   9     parser.add_argument('-r', '--recursive', action='store_true',
  10                         help='remove directories recursively')
  11     parser.add_argument('names', nargs='+',
  12                         help='files/directories names to remove')
  13     args = parser.parse_args()
  14
  15     for name in args.names:
  16         if os.path.isdir(name):
  17             if args.recursive:
  18                 shutil.rmtree(name)
  19             else:
  20                 os.rmdir(name)
  21         else:
  22             os.unlink(name)