from django.core.management.base import NoArgsCommand
from django_extensions.management.utils import get_project_root
from random import choice
from optparse import make_option
from os.path import join as _j
import py_compile 
import os

class Command(NoArgsCommand):
    option_list = NoArgsCommand.option_list + (
        make_option('--path', '-p', action='store', dest='path', 
            help='Specify path to recurse into'),
    )
    help = "Compile python bytecode files for the project."
    
    requires_model_validation = False
    
    def handle_noargs(self, **options):
	project_root = options.get("path", None)
	if not project_root:
    	    project_root = get_project_root()
	verbose = int(options.get("verbosity", 1))>1
	
	for root, dirs, files in os.walk(project_root):
	    for file in files:
		ext = os.path.splitext(file)[1]
		if ext==".py":
		    full_path = _j(root, file)
		    if verbose:
			print "%sc" % full_path
		    py_compile.compile(full_path)
		    

# Backwards compatibility for Django r9110
if not [opt for opt in Command.option_list if opt.dest=='verbosity']:
    Command.option_list += (
	make_option('--verbosity', '-v', action="store", dest="verbosity",
	    default='1', type='choice', choices=['0', '1', '2'],
	    help="Verbosity level; 0=minimal output, 1=normal output, 2=all output"),
    )
