From bafe879188f3cd268f81e426bcdf4addc2d9a10a Mon Sep 17 00:00:00 2001
From: Ramiro Morales <cramm0@gmail.com>
Date: Mon, 24 Jan 2011 14:58:05 +0000
Subject: [PATCH] Fixed 14796 -- Modified order of operations performed on
 field names by the inspectdb command so it doesn't generates model fields
 with names equal to Python keywords. Thanks pappjm at gmail dot com for the
 report and mmcnickle for the fix.

git-svn-id: http://code.djangoproject.com/svn/django/trunk@15296 bcc190cf-cafb-0310-a4f2-bffc1f526a37
---
 django/core/management/commands/inspectdb.py  | 11 ++++---
 tests/regressiontests/inspectdb/__init__.py   |  1 +
 .../regressiontests/inspectdb/bug/__init__.py |  1 +
 tests/regressiontests/inspectdb/bug/models.py |  7 +++++
 tests/regressiontests/inspectdb/models.py     |  1 +
 tests/regressiontests/inspectdb/tests.py      | 29 +++++++++++++++++++
 6 files changed, 46 insertions(+), 4 deletions(-)
 create mode 100644 tests/regressiontests/inspectdb/__init__.py
 create mode 100644 tests/regressiontests/inspectdb/bug/__init__.py
 create mode 100644 tests/regressiontests/inspectdb/bug/models.py
 create mode 100644 tests/regressiontests/inspectdb/models.py
 create mode 100644 tests/regressiontests/inspectdb/tests.py

diff --git a/django/core/management/commands/inspectdb.py b/django/core/management/commands/inspectdb.py
index e45f22c2876..5f0e278c61a 100644
--- a/django/core/management/commands/inspectdb.py
+++ b/django/core/management/commands/inspectdb.py
@@ -20,7 +20,7 @@ class Command(NoArgsCommand):
     def handle_noargs(self, **options):
         try:
             for line in self.handle_inspection(options):
-                print line
+                self.stdout.write("%s\n" % line)
         except NotImplementedError:
             raise CommandError("Database inspection isn't supported for the currently selected database backend.")
 
@@ -66,12 +66,11 @@ class Command(NoArgsCommand):
                 if ' ' in att_name:
                     att_name = att_name.replace(' ', '_')
                     comment_notes.append('Field renamed to remove spaces.')
+                    
                 if '-' in att_name:
                     att_name = att_name.replace('-', '_')
                     comment_notes.append('Field renamed to remove dashes.')
-                if keyword.iskeyword(att_name):
-                    att_name += '_field'
-                    comment_notes.append('Field renamed because it was a Python reserved word.')
+                    
                 if column_name != att_name:
                     comment_notes.append('Field name made lowercase.')
 
@@ -97,6 +96,10 @@ class Command(NoArgsCommand):
                             extra_params['unique'] = True
 
                     field_type += '('
+                    
+                if keyword.iskeyword(att_name):
+                    att_name += '_field'
+                    comment_notes.append('Field renamed because it was a Python reserved word.')
 
                 # Don't output 'id = meta.AutoField(primary_key=True)', because
                 # that's assumed if it doesn't exist.
diff --git a/tests/regressiontests/inspectdb/__init__.py b/tests/regressiontests/inspectdb/__init__.py
new file mode 100644
index 00000000000..8b137891791
--- /dev/null
+++ b/tests/regressiontests/inspectdb/__init__.py
@@ -0,0 +1 @@
+
diff --git a/tests/regressiontests/inspectdb/bug/__init__.py b/tests/regressiontests/inspectdb/bug/__init__.py
new file mode 100644
index 00000000000..8b137891791
--- /dev/null
+++ b/tests/regressiontests/inspectdb/bug/__init__.py
@@ -0,0 +1 @@
+
diff --git a/tests/regressiontests/inspectdb/bug/models.py b/tests/regressiontests/inspectdb/bug/models.py
new file mode 100644
index 00000000000..6fec2273e0b
--- /dev/null
+++ b/tests/regressiontests/inspectdb/bug/models.py
@@ -0,0 +1,7 @@
+from django.db import models
+
+class People(models.Model):
+    name = models.CharField(max_length=255)
+
+class Message(models.Model):
+    from_field = models.ForeignKey(People, db_column='from_id')
diff --git a/tests/regressiontests/inspectdb/models.py b/tests/regressiontests/inspectdb/models.py
new file mode 100644
index 00000000000..8b137891791
--- /dev/null
+++ b/tests/regressiontests/inspectdb/models.py
@@ -0,0 +1 @@
+
diff --git a/tests/regressiontests/inspectdb/tests.py b/tests/regressiontests/inspectdb/tests.py
new file mode 100644
index 00000000000..30c7948b221
--- /dev/null
+++ b/tests/regressiontests/inspectdb/tests.py
@@ -0,0 +1,29 @@
+import os
+import sys
+from StringIO import StringIO
+
+from django.conf import settings
+from django.core.management import call_command
+from django.db.models.loading import load_app
+from django.test import TestCase
+
+class InspectDBTestCase(TestCase):
+    
+    def setUp(self):
+        self.old_sys_path = sys.path[:]
+        sys.path.append(os.path.dirname(os.path.abspath(__file__)))
+        self.old_installed_apps = settings.INSTALLED_APPS
+        settings.INSTALLED_APPS = ('bug',)
+        map(load_app, settings.INSTALLED_APPS)
+        call_command('syncdb', verbosity=0)
+        
+    def test_attribute_name_not_python_keyword(self):
+        out = StringIO()
+        call_command('inspectdb', stdout=out)
+        error_message = "inspectdb generated an attribute name which is a python keyword"
+        self.assertNotIn("from = models.ForeignKey(BugPeople)", out.getvalue(), msg=error_message)
+        out.close()
+        
+    def tearDown(self):
+        settings.INSTALLED_APPS = self.old_installed_apps
+        sys.path = self.old_sys_path