Quick utility to pick n random files from a directory

file_sample.py

#!/usr/bin/python
import os
import shutil
import sys
import random

def pick_random(src, dest, N):
	all_files = []
	for path, dirs, files in os.walk(src):
		for f in files:
			full_path = os.path.join(path, f)
			all_files.append(full_path)
	random.shuffle(all_files)
	for i in range(min(len(all_files), N)):
		shutil.copy(all_files[i], dest)

if __name__ == "__main__":
	if len(sys.argv) != 4:
		print("Arguments: <src> <dest> <n>")
	_, src, dest, n = sys.argv
	if not os.path.exists(dest):
		os.mkdir(dest)
	pick_random(src, dest, int(n))