From 9308377fda5da5d85c5b86f9eb6bab672f2cea85 Mon Sep 17 00:00:00 2001
From: Adrian Holovaty <adrian@holovaty.com>
Date: Thu, 26 Jan 2006 21:22:43 +0000
Subject: [PATCH] magic-removal: Removed SharedMethods hack in
 db.models.fields.related -- there's some duplicate code now, but it's easier
 to understand

git-svn-id: http://code.djangoproject.com/svn/django/branches/magic-removal@2129 bcc190cf-cafb-0310-a4f2-bffc1f526a37
---
 django/db/models/fields/related.py | 42 ++++++++++++++++++------------
 1 file changed, 26 insertions(+), 16 deletions(-)

diff --git a/django/db/models/fields/related.py b/django/db/models/fields/related.py
index faea35c99e..f227273689 100644
--- a/django/db/models/fields/related.py
+++ b/django/db/models/fields/related.py
@@ -61,21 +61,7 @@ class RelatedField(object):
         related = RelatedObject(other, cls, self)
         self.contribute_to_related_class(other, related)
 
-#HACK
-class SharedMethods(RelatedField):
-    def get_attname(self):
-        return '%s_id' % self.name
-
-    def get_validator_unique_lookup_type(self):
-        return '%s__%s__exact' % (self.name, self.rel.get_related_field().name)
-
-    def contribute_to_class(self, cls, name):
-        super(SharedMethods, self).contribute_to_class(cls, name)
-        # Add methods for many-to-one related objects.
-        # EXAMPLES: Choice.get_poll(), Story.get_dateline()
-        setattr(cls, 'get_%s' % self.name, curry(cls._get_foreign_key_object, field_with_rel=self))
-
-class ForeignKey(SharedMethods, Field):
+class ForeignKey(RelatedField, Field):
     empty_strings_allowed = False
     def __init__(self, to, to_field=None, **kwargs):
         try:
@@ -106,6 +92,12 @@ class ForeignKey(SharedMethods, Field):
             if name in kwargs:
                 self.deprecated_args.append(name)
 
+    def get_attname(self):
+        return '%s_id' % self.name
+
+    def get_validator_unique_lookup_type(self):
+        return '%s__%s__exact' % (self.name, self.rel.get_related_field().name)
+
     def prepare_field_objs_and_params(self, manipulator, name_prefix):
         params = {'validator_list': self.validator_list[:], 'member_name': name_prefix + self.attname}
         if self.rel.raw_id_admin:
@@ -152,6 +144,12 @@ class ForeignKey(SharedMethods, Field):
                     return {self.attname: choice_list[1][0]}
         return Field.flatten_data(self, follow, obj)
 
+    def contribute_to_class(self, cls, name):
+        super(ForeignKey, self).contribute_to_class(cls, name)
+        # Add methods for many-to-one related objects.
+        # EXAMPLES: Choice.get_poll(), Story.get_dateline()
+        setattr(cls, 'get_%s' % self.name, curry(cls._get_foreign_key_object, field_with_rel=self))
+
     def contribute_to_related_class(self, cls, related):
         rel_obj_name = related.get_accessor_name()
         # Add "get_thingie" methods for many-to-one related objects.
@@ -170,7 +168,7 @@ class ForeignKey(SharedMethods, Field):
             func = lambda self, *args, **kwargs: self._add_related(related.model, related.field, *args, **kwargs)
             setattr(cls, 'add_%s' % rel_obj_name, func)
 
-class OneToOneField(SharedMethods, IntegerField):
+class OneToOneField(RelatedField, IntegerField):
     def __init__(self, to, to_field=None, **kwargs):
         kwargs['verbose_name'] = kwargs.get('verbose_name', 'ID')
         to_field = to_field or to._meta.pk.name
@@ -195,6 +193,18 @@ class OneToOneField(SharedMethods, IntegerField):
             if name in kwargs:
                 self.deprecated_args.append(name)
 
+    def get_attname(self):
+        return '%s_id' % self.name
+
+    def get_validator_unique_lookup_type(self):
+        return '%s__%s__exact' % (self.name, self.rel.get_related_field().name)
+
+    def contribute_to_class(self, cls, name):
+        super(OneToOneField, self).contribute_to_class(cls, name)
+        # Add methods for many-to-one related objects.
+        # EXAMPLES: Choice.get_poll(), Story.get_dateline()
+        setattr(cls, 'get_%s' % self.name, curry(cls._get_foreign_key_object, field_with_rel=self))
+
     def contribute_to_related_class(self, cls, related):
         rel_obj_name = related.get_accessor_name()
         # Add "get_thingie" methods for one-to-one related objects.