Colin's git trees - cmccabe-bin/blob - pickrand.py

   1 #!/usr/bin/python
   2
   3 import os
   4 import random
   5 import sys
   6 import time
   7
   8 print_to_stderr = False
   9 random.seed(os.getpid() + int(time.time()))
  10
  11 file_name = None
  12 if (len(sys.argv) == 1):
  13     allfiles = []
  14     for root, dirs, files in os.walk("."):
  15         for f in files:
  16             allfiles.append(os.path.join(root, f))
  17     if (len(allfiles) == 0):
  18        sys.exit(1)
  19     r = random.randint(0,len(allfiles) - 1)
  20     file_name = allfiles[r]
  21 elif (len(sys.argv) == 2) and (sys.argv[1] == "-S"):
  22     print_to_stderr = True
  23 else:
  24     allfiles = sys.argv[1:]
  25     r = random.randint(0,len(allfiles) - 1)
  26     file_name = allfiles[r]
  27
  28 print(file_name)
  29 if (print_to_stderr):
  30     print >>sys.stderr, file_name
  31 sys.exit(0)