Load directory.

load_dir.py

# Non-recursive
import os
def load_directory(data_path):

	files_list = []
	try:
		for file_name in os.listdir(data_path):
			if file_name.endswith(".html"):
				files_list.append(file_name)

	except IndexError:
		sys.stderr.write( "Something went wrong with " + file_name + ".")
	
	return files_list

# Recursive
from os import walk

def load_directory(data_path):

	files_list = []
	try:
		for (dir_path, dir_names, file_names) in walk(data_path):
			for file_name in file_names:
				if file_name.endswith(".html"):
					files_list.extend(file_name)
			break 
			#Remove break for recursive.
	except IndexError:
		sys.stderr.write( "Something went wrong with " + filenames + ".")
	
	return files_list