Find files recursively using given file extensions and run given command on random 'n' files

randomrun.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
Find files recursively using given file extensions and run given command on random 'n' files:

   * python randomrun.py -e mkv,avi,ogm -r vlc -c 4 /some/dir/ # run 4 random moves using vlc
   * python randomrun.py -e txt,dat -r cat -c 3 -p  /some/dir/ # run cat on 3 random files parallelly
"""

import argparse
import os
import sys
import random
import subprocess

parser = argparse.ArgumentParser()
parser.add_argument("-r", "--run", help="run command with chosen files; default ls", default="ls")
parser.add_argument("-e", "--extensions", help="valid extensions (comma separated); default txt,text", default="txt,text")
parser.add_argument("-c", "--count", help="how many files use; default 10", default=10)
parser.add_argument("-v", "--verbose", help="print messages; default false", action="store_true")
parser.add_argument("-p", "--parallelly", help="run command parallelly; default false", action="store_true")
parser.add_argument("path", help="directory to find files from (recursively)")
args = parser.parse_args()

# validate given path
if not os.path.exists(args.path) or not os.path.isdir(args.path):
    sys.exit('%s not exists or is not a directory' % args.path)

# find files recursively according to given extensions
files = [os.path.join(path, filename)
        for path, dirs, files in os.walk(args.path)
        for filename in files
        if filename.lower().endswith(tuple(args.extensions.split(',')))]

count = int(args.count) if int(args.count) <= len(files) else len(files)
files = random.sample(files, count)

for filename in files:
    cmd = args.run.split(' ')
    cmd.append(filename)
    if args.verbose:
        print "running command: %s" % " ".join(cmd)

    if args.parallelly:
        subprocess.Popen(cmd)
    else:
        subprocess.call(cmd)