1 from django.core.management.base import copy_helper, CommandError, LabelCommand

     2 import os

     3 import re

     4 from random import choice

     5 

     6 INVALID_PROJECT_NAMES = ('django', 'site', 'test')

     7 

     8 class Command(LabelCommand):

     9     help = "Creates a Django project directory structure for the given project name in the current directory."

    10     args = "[projectname]"

    11     label = 'project name'

    12 

    13     requires_model_validation = False

    14     # Can't import settings during this command, because they haven't

    15     # necessarily been created.

    16     can_import_settings = False

    17 

    18     def handle_label(self, project_name, **options):

    19         # Determine the project_name a bit naively -- by looking at the name of

    20         # the parent directory.

    21         directory = os.getcwd()

    22 

    23         try:

    24             proj_name = __import__(project_name)

    25             if proj_name:

    26                 raise CommandError("%r conflicts with the name of an existing Python module and cannot be used as a project name. Please try another name." % project_name)

    27         except ImportError:

    28             if project_name in INVALID_PROJECT_NAMES:

    29                 raise CommandError("%r contains an invalid project name. Please try another name." % project_name)

    30 

    31         copy_helper(self.style, 'project', project_name, directory)

    32 

    33         # Create a random SECRET_KEY hash, and put it in the main settings.

    34         main_settings_file = os.path.join(directory, project_name, 'settings.py')

    35         settings_contents = open(main_settings_file, 'r').read()

    36         fp = open(main_settings_file, 'w')

    37         secret_key = ''.join([choice('abcdefghijklmnopqrstuvwxyz0123456789!@#$%^&*(-_=+)') for i in range(50)])

    38         settings_contents = re.sub(r"(?<=SECRET_KEY = ')'", secret_key + "'", settings_contents)

    39         fp.write(settings_contents)

    40         fp.close()