From f7c5321964172b72fbfcbc28cccebf7769980ea4 Mon Sep 17 00:00:00 2001
From: Robert Wittams <robert@wittams.com>
Date: Tue, 20 Dec 2005 22:48:18 +0000
Subject: [PATCH] magic-removal:Move some field specific stuff out of the
 options preparation.

git-svn-id: http://code.djangoproject.com/svn/django/branches/magic-removal@1750 bcc190cf-cafb-0310-a4f2-bffc1f526a37
---
 django/db/models/fields/__init__.py |  6 ++++++
 django/db/models/fields/related.py  |  5 +++--
 django/db/models/options.py         | 33 +++++++----------------------
 3 files changed, 17 insertions(+), 27 deletions(-)

diff --git a/django/db/models/fields/__init__.py b/django/db/models/fields/__init__.py
index b0fb245830..a5ecb73e9b 100644
--- a/django/db/models/fields/__init__.py
+++ b/django/db/models/fields/__init__.py
@@ -339,6 +339,12 @@ class AutoField(Field):
             return None
         return Field.get_manipulator_new_data(self, new_data, rel)
 
+    def contribute_to_class(self, cls, name):
+        if cls._meta.has_auto_field:
+            raise AssertionError, "A model can't have more than one AutoField." 
+        super(AutoField, self).contribute_to_class(cls, name)
+        cls._meta.has_auto_field = True
+
 class BooleanField(Field):
     def __init__(self, *args, **kwargs):
         kwargs['blank'] = True
diff --git a/django/db/models/fields/related.py b/django/db/models/fields/related.py
index 24433e0dfe..ccdb59e799 100644
--- a/django/db/models/fields/related.py
+++ b/django/db/models/fields/related.py
@@ -197,8 +197,9 @@ class OneToOneField(SharedMethods, IntegerField):
         setattr(cls, 'get_%s' % rel_obj_name,
                 curry(cls._get_related, method_name='get_object',
                       rel_class=related.model, rel_field=related.field))
-
-
+        if not cls._meta.one_to_one_field:
+           cls._meta.one_to_one_field = self
+           
 class ManyToManyField(RelatedField, Field):
     def __init__(self, to, **kwargs):
         kwargs['verbose_name'] = kwargs.get('verbose_name', None)
diff --git a/django/db/models/options.py b/django/db/models/options.py
index cdab02e88d..dfe54de00c 100644
--- a/django/db/models/options.py
+++ b/django/db/models/options.py
@@ -1,5 +1,5 @@
 from django.db.models.related import RelatedObject
-from django.db.models.fields.related import OneToOne, ManyToMany
+from django.db.models.fields.related import ManyToMany
 from django.db.models.fields import AutoField
 from django.db.models.loading import get_installed_model_modules
 from django.db.models.query import orderlist2sql
@@ -33,7 +33,11 @@ class Options:
         self.order_with_respect_to = None
         self.module_constants = {}
         self.admin = None
+        
         self.meta = meta
+        self.pk = None
+        self.has_auto_field = False
+        self.one_to_one_field = None
 
     def merge_meta(self):
         meta_attrs = self.meta.__dict__
@@ -63,34 +67,11 @@ class Options:
         else:
             self.order_with_respect_to = None
 
-        # Calculate one_to_one_field.
-        self.one_to_one_field = None
-        for f in self.fields:
-            if isinstance(f.rel, OneToOne):
-                self.one_to_one_field = f
-                break
-        # Cache the primary-key field.
-        self.pk = None
-        for f in self.fields:
-            if f.primary_key:
-                self.pk = f
-                break
-        # If a primary_key field hasn't been specified, add an
-        # auto-incrementing primary-key ID field automatically.
         if self.pk is None:
             auto = AutoField(verbose_name='ID', primary_key=True)
             auto.creation_counter = -1
             model.add_to_class('id', auto)
-            self.pk = self.fields[0]
-        # Cache whether this has an AutoField.
-        self.has_auto_field = False
-        for f in self.fields:
-            is_auto = isinstance(f, AutoField)
-            if is_auto and self.has_auto_field:
-                raise AssertionError, "A model can't have more than one AutoField."
-            elif is_auto:
-                self.has_auto_field = True
-        #HACK
+
         self.limit_choices_to = {}
 
         # If the db_table wasn't provided, use the app_label + module_name.
@@ -105,6 +86,8 @@ class Options:
             self.many_to_many.insert(bisect(self.many_to_many, field), field)
         else:
             self.fields.insert(bisect(self.fields, field), field)
+            if not self.pk and field.primary_key:
+                self.pk = field
 
     def __repr__(self):
         return '<Options for %s>' % self.module_name