From 9893fa12b735f3f47b35d4063d86dddf3145cb25 Mon Sep 17 00:00:00 2001 From: Claude Paroz Date: Sat, 13 Oct 2012 16:05:34 +0200 Subject: [PATCH] Fixed #19125 -- The startproject command should validate the name earlier MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Thanks Ɓukasz Rekucki for the report and the patch. --- django/core/management/commands/startapp.py | 3 +-- .../core/management/commands/startproject.py | 3 +-- django/core/management/templates.py | 25 +++++++++++-------- tests/regressiontests/admin_scripts/tests.py | 14 ++++++----- 4 files changed, 25 insertions(+), 20 deletions(-) diff --git a/django/core/management/commands/startapp.py b/django/core/management/commands/startapp.py index 5581331b7b..692ad09a43 100644 --- a/django/core/management/commands/startapp.py +++ b/django/core/management/commands/startapp.py @@ -9,8 +9,7 @@ class Command(TemplateCommand): "directory.") def handle(self, app_name=None, target=None, **options): - if app_name is None: - raise CommandError("you must provide an app name") + self.validate_name(app_name, "app") # Check that the app_name cannot be imported. try: diff --git a/django/core/management/commands/startproject.py b/django/core/management/commands/startproject.py index a1a9b33a43..b143e6c380 100644 --- a/django/core/management/commands/startproject.py +++ b/django/core/management/commands/startproject.py @@ -10,8 +10,7 @@ class Command(TemplateCommand): "given directory.") def handle(self, project_name=None, target=None, *args, **options): - if project_name is None: - raise CommandError("you must provide a project name") + self.validate_name(project_name, "project") # Check that the project_name cannot be imported. try: diff --git a/django/core/management/templates.py b/django/core/management/templates.py index f522097b8c..927dbc13ac 100644 --- a/django/core/management/templates.py +++ b/django/core/management/templates.py @@ -67,16 +67,7 @@ class TemplateCommand(BaseCommand): self.paths_to_remove = [] self.verbosity = int(options.get('verbosity')) - # If it's not a valid directory name. - if not re.search(r'^[_a-zA-Z]\w*$', name): - # Provide a smart error message, depending on the error. - if not re.search(r'^[_a-zA-Z]', name): - message = ('make sure the name begins ' - 'with a letter or underscore') - else: - message = 'use only numbers, letters and underscores' - raise CommandError("%r is not a valid %s name. Please %s." % - (name, app_or_project, message)) + self.validate_name(name, app_or_project) # if some directory is given, make sure it's nicely expanded if target is None: @@ -211,6 +202,20 @@ class TemplateCommand(BaseCommand): raise CommandError("couldn't handle %s template %s." % (self.app_or_project, template)) + def validate_name(self, name, app_or_project): + if name is None: + raise CommandError("you must provide %s %s name" % ( + "an" if app_or_project == "app" else "a", app_or_project)) + # If it's not a valid directory name. + if not re.search(r'^[_a-zA-Z]\w*$', name): + # Provide a smart error message, depending on the error. + if not re.search(r'^[_a-zA-Z]', name): + message = 'make sure the name begins with a letter or underscore' + else: + message = 'use only numbers, letters and underscores' + raise CommandError("%r is not a valid %s name. Please %s." % + (name, app_or_project, message)) + def download(self, url): """ Downloads the given URL and returns the file name. diff --git a/tests/regressiontests/admin_scripts/tests.py b/tests/regressiontests/admin_scripts/tests.py index d0ca9d26df..df2326e163 100644 --- a/tests/regressiontests/admin_scripts/tests.py +++ b/tests/regressiontests/admin_scripts/tests.py @@ -1430,13 +1430,15 @@ class StartProject(LiveServerTestCase, AdminScriptTestCase): def test_invalid_project_name(self): "Make sure the startproject management command validates a project name" - args = ['startproject', '7testproject'] - testproject_dir = os.path.join(test_dir, '7testproject') - self.addCleanup(shutil.rmtree, testproject_dir, True) + for bad_name in ('7testproject', '../testproject'): + args = ['startproject', bad_name] + testproject_dir = os.path.join(test_dir, bad_name) + self.addCleanup(shutil.rmtree, testproject_dir, True) - out, err = self.run_django_admin(args) - self.assertOutput(err, "Error: '7testproject' is not a valid project name. Please make sure the name begins with a letter or underscore.") - self.assertFalse(os.path.exists(testproject_dir)) + out, err = self.run_django_admin(args) + self.assertOutput(err, "Error: '%s' is not a valid project name. " + "Please make sure the name begins with a letter or underscore." % bad_name) + self.assertFalse(os.path.exists(testproject_dir)) def test_simple_project_different_directory(self): "Make sure the startproject management command creates a project in a specific directory"