From a01eab5049070498788e735bbd0b9175c93332a2 Mon Sep 17 00:00:00 2001 From: Robert Wittams Date: Fri, 16 Dec 2005 02:16:59 +0000 Subject: [PATCH] Minimal support for change forms in the admin git-svn-id: http://code.djangoproject.com/svn/django/branches/magic-removal@1683 bcc190cf-cafb-0310-a4f2-bffc1f526a37 --- .../admin/templates/widget/foreign.html | 4 +- .../admin/templatetags/admin_modify.py | 9 +- django/contrib/admin/views/main.py | 36 +- django/core/formfields.py | 2 +- django/db/models/__init__.py | 738 +++++++----------- django/db/models/fields.py | 3 +- 6 files changed, 319 insertions(+), 473 deletions(-) diff --git a/django/contrib/admin/templates/widget/foreign.html b/django/contrib/admin/templates/widget/foreign.html index c2d06d578a..d305066066 100644 --- a/django/contrib/admin/templates/widget/foreign.html +++ b/django/contrib/admin/templates/widget/foreign.html @@ -1,8 +1,8 @@ {% load admin_modify adminmedia %} {% output_all bound_field.form_fields %} {% if bound_field.raw_id_admin %} - Lookup + FIXURL Lookup {% else %} {% if bound_field.needs_add_label %} - Add Another + FIXURL Add Another {% endif %}{% endif %} diff --git a/django/contrib/admin/templatetags/admin_modify.py b/django/contrib/admin/templatetags/admin_modify.py index d5fcc8081c..48d016164a 100644 --- a/django/contrib/admin/templatetags/admin_modify.py +++ b/django/contrib/admin/templatetags/admin_modify.py @@ -1,10 +1,11 @@ -from django.core import template, template_loader, meta +from django.core import template, template_loader from django.utils.html import escape from django.utils.text import capfirst from django.utils.functional import curry from django.contrib.admin.views.main import AdminBoundField from django.db.models.fields import BoundField, Field from django.db.models import BoundRelatedObject, TABULAR, STACKED +from django.db import models from django.conf.settings import ADMIN_MEDIA_PREFIX import re @@ -42,7 +43,7 @@ submit_row = register.inclusion_tag('admin/submit_line', takes_context=True)(sub #@register.simple_tag def field_label(bound_field): class_names = [] - if isinstance(bound_field.field, meta.BooleanField): + if isinstance(bound_field.field, models.BooleanField): class_names.append("vCheckboxLabel") colon = "" else: @@ -135,6 +136,8 @@ class StackedBoundRelatedObject(BoundRelatedObject): def __init__(self, related_object, field_mapping, original): super(StackedBoundRelatedObject, self).__init__(related_object, field_mapping, original) fields = self.relation.editable_fields() + self.field_mappings.fill() + print self.field_mappings.__dict__ self.form_field_collection_wrappers = [FormFieldCollectionWrapper(field_mapping ,fields) for field_mapping in self.field_mappings] self.show_url = original and hasattr(self.relation.opts, 'get_absolute_url') @@ -241,7 +244,7 @@ def admin_field_line(context, argument_val): break # Assumes BooleanFields won't be stacked next to each other! - if isinstance(bound_fields[0].field, meta.BooleanField): + if isinstance(bound_fields[0].field, models.BooleanField): class_names.append('checkbox-row') return { diff --git a/django/contrib/admin/views/main.py b/django/contrib/admin/views/main.py index 3624436445..d0b0be3e73 100644 --- a/django/contrib/admin/views/main.py +++ b/django/contrib/admin/views/main.py @@ -67,7 +67,6 @@ def find_model(mod, remaining): if hasattr(mod, '_MODELS'): name = remaining[0] for model in mod._MODELS: - print "'%s'" % name, model if model.__name__.lower() == name: return model raise Http404 @@ -391,29 +390,32 @@ class AdminBoundFieldSet(BoundFieldSet): super(AdminBoundFieldSet, self).__init__(field_set, field_mapping, original, AdminBoundFieldLine) class BoundManipulator(object): - def __init__(self, opts, manipulator, field_mapping): - self.inline_related_objects = opts.get_followed_related_objects(manipulator.follow) + def __init__(self, model, manipulator, field_mapping): + self.model = model + self.opts = model._meta + self.inline_related_objects = self.opts.get_followed_related_objects(manipulator.follow) self.original = hasattr(manipulator, 'original_object') and manipulator.original_object or None self.bound_field_sets = [field_set.bind(field_mapping, self.original, AdminBoundFieldSet) - for field_set in opts.admin.get_field_sets(opts)] - self.ordered_objects = opts.get_ordered_objects()[:] + for field_set in self.opts.admin.get_field_sets(self.opts)] + self.ordered_objects = self.opts.get_ordered_objects()[:] class AdminBoundManipulator(BoundManipulator): - def __init__(self, opts, manipulator, field_mapping): - super(AdminBoundManipulator, self).__init__(opts, manipulator, field_mapping) - field_sets = opts.admin.get_field_sets(opts) + def __init__(self, model, manipulator, field_mapping): + super(AdminBoundManipulator, self).__init__(model, manipulator, field_mapping) + field_sets = self.opts.admin.get_field_sets(self.opts) - self.auto_populated_fields = [f for f in opts.fields if f.prepopulate_from] - self.javascript_imports = get_javascript_imports(opts, self.auto_populated_fields, self.ordered_objects, field_sets); + self.auto_populated_fields = [f for f in self.opts.fields if f.prepopulate_from] + self.javascript_imports = get_javascript_imports(self.opts, self.auto_populated_fields, self.ordered_objects, field_sets); self.coltype = self.ordered_objects and 'colMS' or 'colM' - self.has_absolute_url = hasattr(opts.get_model_module().Klass, 'get_absolute_url') - self.form_enc_attrib = opts.has_field_type(models.FileField) and \ + self.has_absolute_url = hasattr(model, 'get_absolute_url') + self.form_enc_attrib = self.opts.has_field_type(models.FileField) and \ 'enctype="multipart/form-data" ' or '' self.first_form_field_id = self.bound_field_sets[0].bound_field_lines[0].bound_fields[0].form_fields[0].get_id(); self.ordered_object_pk_names = [o.pk.name for o in self.ordered_objects] + opts = self.opts self.save_on_top = opts.admin.save_on_top self.save_as = opts.admin.save_as @@ -428,11 +430,12 @@ class AdminBoundManipulator(BoundManipulator): return str(getattr(ordered_obj, name)) return "" -def render_change_form(opts, manipulator, app_label, context, add=False, change=False, show_delete=False, form_url=''): +def render_change_form(model, manipulator, app_label, context, add=False, change=False, show_delete=False, form_url=''): + opts = model._meta extra_context = { 'add': add, 'change': change, - 'bound_manipulator': AdminBoundManipulator(opts, manipulator, context['form']), + 'bound_manipulator': AdminBoundManipulator(model, manipulator, context['form']), 'has_delete_permission': context['perms'][app_label][opts.get_delete_permission()], 'form_url': form_url, 'app_label': app_label, @@ -527,7 +530,8 @@ def change_stage(request, path, object_id): if request.POST and request.POST.has_key("_saveasnew"): return add_stage(request, path, form_url='../add/') try: - manipulator = mod.ChangeManipulator(object_id) + manipulator_class = model.ChangeManipulator + manipulator = manipulator_class(object_id) except ObjectDoesNotExist: raise Http404 @@ -596,7 +600,7 @@ def change_stage(request, path, object_id): 'is_popup' : request.REQUEST.has_key('_popup') }) - return render_change_form(opts,manipulator, app_label, c, change=True) + return render_change_form(model,manipulator, app_label, c, change=True) def _nest_help(obj, depth, val): current = obj diff --git a/django/core/formfields.py b/django/core/formfields.py index c1084fad95..bde4b1abbd 100644 --- a/django/core/formfields.py +++ b/django/core/formfields.py @@ -10,7 +10,7 @@ class EmptyValue(Exception): "This is raised when empty data is provided" pass -class Manipulator: +class Manipulator(object): # List of permission strings. User must have at least one to manipulate. # None means everybody has permission. required_permission = '' diff --git a/django/db/models/__init__.py b/django/db/models/__init__.py index aaf1860cb9..88a3a5ce23 100644 --- a/django/db/models/__init__.py +++ b/django/db/models/__init__.py @@ -243,9 +243,11 @@ class RelatedObject(object): def get_manipulator_fields(self, opts, manipulator, change, follow): # TODO: Remove core fields stuff. - if change: + + if manipulator.original_object: meth_name = 'get_%s_count' % self.get_method_name_part() count = getattr(manipulator.original_object, meth_name)() + count += self.field.rel.num_extra_on_change if self.field.rel.min_num_in_admin: count = max(count, self.field.rel.min_num_in_admin) @@ -253,7 +255,6 @@ class RelatedObject(object): count = min(count, self.field.rel.max_num_in_admin) else: count = self.field.rel.num_in_admin - fields = [] for i in range(count): for f in self.opts.fields + self.opts.many_to_many: @@ -416,6 +417,9 @@ class Options: raise AssertionError, "A model can't have more than one AutoField." elif is_auto: self.has_auto_field = True + #HACK + self.limit_choices_to = {} + def __repr__(self): return '' % self.module_name @@ -426,8 +430,11 @@ class Options: def get_content_type_id(self): "Returns the content-type ID for this object type." if not hasattr(self, '_content_type_id'): - mod = get_module('core', 'contenttypes') - self._content_type_id = mod.get_object(python_module_name__exact=self.module_name, package__label__exact=self.app_label).id + import django.models.core + manager = django.models.core.ContentType.objects + self._content_type_id = \ + manager.get_object(python_module_name__exact=self.module_name, + package__label__exact=self.app_label).id return self._content_type_id def get_field(self, name, many_to_many=True): @@ -505,11 +512,12 @@ class Options: "Returns a list of Options objects that are ordered with respect to this object." if not hasattr(self, '_ordered_objects'): objects = [] - for klass in get_app(self.app_label)._MODELS: - opts = klass._meta - if opts.order_with_respect_to and opts.order_with_respect_to.rel \ - and self == opts.order_with_respect_to.rel.to._meta: - objects.append(opts) + #HACK + #for klass in get_app(self.app_label)._MODELS: + # opts = klass._meta + # if opts.order_with_respect_to and opts.order_with_respect_to.rel \ + # and self == opts.order_with_respect_to.rel.to._meta: + # objects.append(opts) self._ordered_objects = objects return self._ordered_objects @@ -740,6 +748,268 @@ class Manager(object): # objects -- MySQL returns the values as strings, instead. return [typecast_timestamp(str(row[0])) for row in cursor.fetchall()] + +class ManipulatorDescriptor(object): + class empty: + pass + + def __init__(self, name, base): + self.man = None + self.name = name + self.base = base + + def __get__(self, instance, type=None): + if instance != None: + raise "Manipulator accessed via instance" + else: + class Man(self.get_base_manipulator(type), self.base): + pass + Man.classinit(type) + Man.__name__ = self.name + return Man + + def get_base_manipulator(self, type): + if hasattr(type, 'MANIPULATOR'): + man = type.MANIPULATOR + else: + man = self.empty + return man + + +class AutomaticManipulator(formfields.Manipulator): + def classinit(cls, model): + cls.model = model + cls.manager = model._default_manager + cls.opts = model._meta + for field_name_list in cls.opts.unique_together: + setattr(cls, 'isUnique%s' % '_'.join(field_name_list), curry(manipulator_validator_unique_together, field_name_list, opts)) + for f in cls.opts.fields: + if f.unique_for_date: + setattr(cls, 'isUnique%sFor%s' % (f.name, f.unique_for_date), curry(manipulator_validator_unique_for_date, f, opts.get_field(f.unique_for_date), opts, 'date')) + if f.unique_for_month: + setattr(cls, 'isUnique%sFor%s' % (f.name, f.unique_for_month), curry(manipulator_validator_unique_for_date, f, opts.get_field(f.unique_for_month), opts, 'month')) + if f.unique_for_year: + setattr(cls, 'isUnique%sFor%s' % (f.name, f.unique_for_year), curry(manipulator_validator_unique_for_date, f, opts.get_field(f.unique_for_year), opts, 'year')) + classinit = classmethod(classinit) + + def __init__(self, original_object= None, follow=None): + self.follow = self.model._meta.get_follow(follow) + self.fields = [] + self.original_object = original_object + for f in self.opts.fields + self.opts.many_to_many: + if self.follow.get(f.name, False): + self.fields.extend(f.get_manipulator_fields(self.opts, self, self.change)) + + + # Add fields for related objects. + for f in self.opts.get_all_related_objects(): + if self.follow.get(f.name, False): + print f.name + fol = self.follow[f.name] + fields = f.get_manipulator_fields(self.opts, self, self.change, fol) + print fields + self.fields.extend(fields) + + + def save(self, new_data): + add, change, opts, klass = self.add, self.change, self.opts, self.model + # TODO: big cleanup when core fields go -> use recursive manipulators. + from django.utils.datastructures import DotExpandedDict + params = {} + for f in opts.fields: + # Fields with auto_now_add should keep their original value in the change stage. + auto_now_add = change and getattr(f, 'auto_now_add', False) + if self.follow.get(f.name, None) and not auto_now_add: + param = f.get_manipulator_new_data(new_data) + else: + if change: + param = getattr(self.original_object, f.attname) + else: + param = f.get_default() + params[f.attname] = param + + if change: + params[opts.pk.attname] = self.obj_key + + # First, save the basic object itself. + new_object = klass(**params) + new_object.save() + + # Now that the object's been saved, save any uploaded files. + for f in opts.fields: + if isinstance(f, FileField): + f.save_file(new_data, new_object, change and self.original_object or None, change, rel=False) + + # Calculate which primary fields have changed. + if change: + self.fields_added, self.fields_changed, self.fields_deleted = [], [], [] + for f in opts.fields: + if not f.primary_key and str(getattr(self.original_object, f.attname)) != str(getattr(new_object, f.attname)): + self.fields_changed.append(f.verbose_name) + + # Save many-to-many objects. Example: Poll.set_sites() + for f in opts.many_to_many: + if self.follow.get(f.name, None): + if not f.rel.edit_inline: + if f.rel.raw_id_admin: + new_vals = new_data.get(f.name, ()) + else: + new_vals = new_data.getlist(f.name) + was_changed = getattr(new_object, 'set_%s' % f.name)(new_vals) + if change and was_changed: + self.fields_changed.append(f.verbose_name) + + expanded_data = DotExpandedDict(dict(new_data)) + # Save many-to-one objects. Example: Add the Choice objects for a Poll. + for related in opts.get_all_related_objects(): + # Create obj_list, which is a DotExpandedDict such as this: + # [('0', {'id': ['940'], 'choice': ['This is the first choice']}), + # ('1', {'id': ['941'], 'choice': ['This is the second choice']}), + # ('2', {'id': [''], 'choice': ['']})] + child_follow = self.follow.get(related.name, None) + + if child_follow: + obj_list = expanded_data[related.var_name].items() + obj_list.sort(lambda x, y: cmp(int(x[0]), int(y[0]))) + params = {} + + # For each related item... + for _, rel_new_data in obj_list: + + # Keep track of which core=True fields were provided. + # If all core fields were given, the related object will be saved. + # If none of the core fields were given, the object will be deleted. + # If some, but not all, of the fields were given, the validator would + # have caught that. + all_cores_given, all_cores_blank = True, True + + # Get a reference to the old object. We'll use it to compare the + # old to the new, to see which fields have changed. + old_rel_obj = None + if change: + if rel_new_data[related.opts.pk.name][0]: + try: + old_rel_obj = getattr(self.original_object, 'get_%s' % related.get_method_name_part() )(**{'%s__exact' % related.opts.pk.name: rel_new_data[related.opts.pk.attname][0]}) + except ObjectDoesNotExist: + pass + + for f in related.opts.fields: + if f.core and not isinstance(f, FileField) and f.get_manipulator_new_data(rel_new_data, rel=True) in (None, ''): + all_cores_given = False + elif f.core and not isinstance(f, FileField) and f.get_manipulator_new_data(rel_new_data, rel=True) not in (None, ''): + all_cores_blank = False + # If this field isn't editable, give it the same value it had + # previously, according to the given ID. If the ID wasn't + # given, use a default value. FileFields are also a special + # case, because they'll be dealt with later. + + if f == related.field: + param = getattr(new_object, related.field.rel.field_name) + elif add and isinstance(f, AutoField): + param = None + elif change and (isinstance(f, FileField) or not child_follow.get(f.name, None)): + if old_rel_obj: + param = getattr(old_rel_obj, f.column) + else: + param = f.get_default() + else: + param = f.get_manipulator_new_data(rel_new_data, rel=True) + if param != None: + params[f.attname] = param + + # Create the related item. + new_rel_obj = related.opts.get_model_module().Klass(**params) + + # If all the core fields were provided (non-empty), save the item. + if all_cores_given: + new_rel_obj.save() + + # Save any uploaded files. + for f in related.opts.fields: + if child_follow.get(f.name, None): + if isinstance(f, FileField) and rel_new_data.get(f.name, False): + f.save_file(rel_new_data, new_rel_obj, change and old_rel_obj or None, old_rel_obj is not None, rel=True) + + # Calculate whether any fields have changed. + if change: + if not old_rel_obj: # This object didn't exist before. + self.fields_added.append('%s "%s"' % (related.opts.verbose_name, new_rel_obj)) + else: + for f in related.opts.fields: + if not f.primary_key and f != related.field and str(getattr(old_rel_obj, f.attname)) != str(getattr(new_rel_obj, f.attname)): + self.fields_changed.append('%s for %s "%s"' % (f.verbose_name, related.opts.verbose_name, new_rel_obj)) + + # Save many-to-many objects. + for f in related.opts.many_to_many: + if child_follow.get(f.name, None) and not f.rel.edit_inline: + was_changed = getattr(new_rel_obj, 'set_%s' % f.name)(rel_new_data[f.attname]) + if change and was_changed: + self.fields_changed.append('%s for %s "%s"' % (f.verbose_name, related.opts.verbose_name, new_rel_obj)) + + # If, in the change stage, all of the core fields were blank and + # the primary key (ID) was provided, delete the item. + if change and all_cores_blank and old_rel_obj: + new_rel_obj.delete() + self.fields_deleted.append('%s "%s"' % (related.opts.verbose_name, old_rel_obj)) + + # Save the order, if applicable. + if change and opts.get_ordered_objects(): + order = new_data['order_'] and map(int, new_data['order_'].split(',')) or [] + for rel_opts in opts.get_ordered_objects(): + getattr(new_object, 'set_%s_order' % rel_opts.object_name.lower())(order) + return new_object + + def get_related_objects(self): + return self.opts.get_followed_related_objects(self.follow) + + def flatten_data(self): + new_data = {} + for f in self.opts.get_data_holders(self.follow): + fol = self.follow.get(f.name) + new_data.update(f.flatten_data(fol, self.original_object)) + return new_data + +class ModelAddManipulator(AutomaticManipulator): + change = False + add = True + def __init__(self, follow=None): + super(ModelAddManipulator, self).__init__(follow=follow) + +class ModelChangeManipulator(AutomaticManipulator): + change = False + add = True + + def __init__(self, obj_key=None, follow=None): + assert obj_key is not None, "ChangeManipulator.__init__() must be passed obj_key parameter." + self.obj_key = obj_key + try: + original_object = self.__class__.manager.get_object(pk=obj_key) + except ObjectDoesNotExist: + # If the object doesn't exist, this might be a manipulator for a + # one-to-one related object that hasn't created its subobject yet. + # For example, this might be a Restaurant for a Place that doesn't + # yet have restaurant information. + if opts.one_to_one_field: + # Sanity check -- Make sure the "parent" object exists. + # For example, make sure the Place exists for the Restaurant. + # Let the ObjectDoesNotExist exception propogate up. + lookup_kwargs = opts.one_to_one_field.rel.limit_choices_to + lookup_kwargs['%s__exact' % opts.one_to_one_field.rel.field_name] = obj_key + _ = opts.one_to_one_field.rel.to._meta.get_model_module().get_object(**lookup_kwargs) + params = dict([(f.attname, f.get_default()) for f in opts.fields]) + params[opts.pk.attname] = obj_key + original_object = opts.get_model_module().Klass(**params) + else: + raise + print "calling super" + super(ModelChangeManipulator, self).__init__(original_object=original_object, follow=follow) + print "Back" + self.original_object = original_object + + if self.opts.get_ordered_objects(): + self.fields.append(formfields.CommaSeparatedIntegerField(field_name="order_")) + + class ModelBase(type): "Metaclass for all models" def __new__(cls, name, bases, attrs): @@ -850,11 +1120,16 @@ class ModelBase(type): related = RelatedObject(other._meta, new_class, field) field.contribute_to_related_class(other, related) + return new_class + class Model(object): __metaclass__ = ModelBase - + + AddManipulator = ManipulatorDescriptor('AddManipulator', ModelAddManipulator) + ChangeManipulator = ManipulatorDescriptor('ChangeManipulator', ModelChangeManipulator) + def __repr__(self): return '<%s object>' % self.__class__.__name__ @@ -1051,33 +1326,7 @@ class Model(object): delete.alters_data = True - def __get_manipulator(cls): - if hasattr(cls, 'MANIPULATOR'): - man = cls.MANIPULATOR - else: - class man: - pass - return man - __get_manipulator=classmethod(__get_manipulator) - - def __get_add_manipulator(cls): - class AddManipulator(cls.__get_manipulator(), ModelAddManipulator): - pass - AddManipulator.classinit(cls) - cls.AddManipulator = AddManipulator - return AddManipulator - AddManipulator = property(classmethod(__get_add_manipulator)) - def __get_change_manipulator(cls): - class ChangeManipulator(cls.__get_manipulator(), ModelChangeManipulator): - pass - ChangeManipulator.classinit(cls) - cls.ChangeManipulator = ChangeManipulator - return ChangeManipulator - ChangeManipulator = property(classmethod(__get_change_manipulator)) - - - def __get_FIELD_display(self, field): value = getattr(self, field.attname) return dict(field.choices).get(value, value) @@ -1286,6 +1535,7 @@ class Model(object): connection.commit() + ############################################ @@ -1510,6 +1760,8 @@ def _parse_lookup(kwarg_items, opts, table_count=0): # HELPER FUNCTIONS (MANIPULATORS) # ################################### + + def get_manipulator(opts, klass, extra_methods, add=False, change=False): "Returns the custom Manipulator (either add or change) for the given opts." assert (add == False or change == False) and add != change, "get_manipulator() can be passed add=True or change=True, but not both" @@ -1522,420 +1774,6 @@ def get_manipulator(opts, klass, extra_methods, add=False, change=False): return man -class AutomaticManipulator(formfields.Manipulator): - def classinit(cls, model): - cls.model = model - cls.manager = model._default_manager - cls.opts = model._meta - for field_name_list in cls.opts.unique_together: - setattr(cls, 'isUnique%s' % '_'.join(field_name_list), curry(manipulator_validator_unique_together, field_name_list, opts)) - for f in cls.opts.fields: - if f.unique_for_date: - setattr(man, 'isUnique%sFor%s' % (f.name, f.unique_for_date), curry(manipulator_validator_unique_for_date, f, opts.get_field(f.unique_for_date), opts, 'date')) - if f.unique_for_month: - setattr(man, 'isUnique%sFor%s' % (f.name, f.unique_for_month), curry(manipulator_validator_unique_for_date, f, opts.get_field(f.unique_for_month), opts, 'month')) - if f.unique_for_year: - setattr(man, 'isUnique%sFor%s' % (f.name, f.unique_for_year), curry(manipulator_validator_unique_for_date, f, opts.get_field(f.unique_for_year), opts, 'year')) - classinit = classmethod(classinit) - - def __init__(self, follow=None): - self.follow = self.model._meta.get_follow(follow) - self.fields = [] - - for f in opts.fields + opts.many_to_many: - if self.follow.get(f.name, False): - self.fields.extend(f.get_manipulator_fields(opts, self, change)) - - # Add fields for related objects. - for f in opts.get_all_related_objects(): - if self.follow.get(f.name, False): - fol = self.follow[f.name] - self.fields.extend(f.get_manipulator_fields(opts, self, change, fol)) - self.original_object = None - - def save(self, new_data): - add, change, opts, klass = self.add, self.change, self.opts, self.model - # TODO: big cleanup when core fields go -> use recursive manipulators. - from django.utils.datastructures import DotExpandedDict - params = {} - for f in opts.fields: - # Fields with auto_now_add should keep their original value in the change stage. - auto_now_add = change and getattr(f, 'auto_now_add', False) - if self.follow.get(f.name, None) and not auto_now_add: - param = f.get_manipulator_new_data(new_data) - else: - if change: - param = getattr(self.original_object, f.attname) - else: - param = f.get_default() - params[f.attname] = param - - if change: - params[opts.pk.attname] = self.obj_key - - # First, save the basic object itself. - new_object = klass(**params) - new_object.save() - - # Now that the object's been saved, save any uploaded files. - for f in opts.fields: - if isinstance(f, FileField): - f.save_file(new_data, new_object, change and self.original_object or None, change, rel=False) - - # Calculate which primary fields have changed. - if change: - self.fields_added, self.fields_changed, self.fields_deleted = [], [], [] - for f in opts.fields: - if not f.primary_key and str(getattr(self.original_object, f.attname)) != str(getattr(new_object, f.attname)): - self.fields_changed.append(f.verbose_name) - - # Save many-to-many objects. Example: Poll.set_sites() - for f in opts.many_to_many: - if self.follow.get(f.name, None): - if not f.rel.edit_inline: - if f.rel.raw_id_admin: - new_vals = new_data.get(f.name, ()) - else: - new_vals = new_data.getlist(f.name) - was_changed = getattr(new_object, 'set_%s' % f.name)(new_vals) - if change and was_changed: - self.fields_changed.append(f.verbose_name) - - expanded_data = DotExpandedDict(dict(new_data)) - # Save many-to-one objects. Example: Add the Choice objects for a Poll. - for related in opts.get_all_related_objects(): - # Create obj_list, which is a DotExpandedDict such as this: - # [('0', {'id': ['940'], 'choice': ['This is the first choice']}), - # ('1', {'id': ['941'], 'choice': ['This is the second choice']}), - # ('2', {'id': [''], 'choice': ['']})] - child_follow = self.follow.get(related.name, None) - - if child_follow: - obj_list = expanded_data[related.var_name].items() - obj_list.sort(lambda x, y: cmp(int(x[0]), int(y[0]))) - params = {} - - # For each related item... - for _, rel_new_data in obj_list: - - # Keep track of which core=True fields were provided. - # If all core fields were given, the related object will be saved. - # If none of the core fields were given, the object will be deleted. - # If some, but not all, of the fields were given, the validator would - # have caught that. - all_cores_given, all_cores_blank = True, True - - # Get a reference to the old object. We'll use it to compare the - # old to the new, to see which fields have changed. - old_rel_obj = None - if change: - if rel_new_data[related.opts.pk.name][0]: - try: - old_rel_obj = getattr(self.original_object, 'get_%s' % related.get_method_name_part() )(**{'%s__exact' % related.opts.pk.name: rel_new_data[related.opts.pk.attname][0]}) - except ObjectDoesNotExist: - pass - - for f in related.opts.fields: - if f.core and not isinstance(f, FileField) and f.get_manipulator_new_data(rel_new_data, rel=True) in (None, ''): - all_cores_given = False - elif f.core and not isinstance(f, FileField) and f.get_manipulator_new_data(rel_new_data, rel=True) not in (None, ''): - all_cores_blank = False - # If this field isn't editable, give it the same value it had - # previously, according to the given ID. If the ID wasn't - # given, use a default value. FileFields are also a special - # case, because they'll be dealt with later. - - if f == related.field: - param = getattr(new_object, related.field.rel.field_name) - elif add and isinstance(f, AutoField): - param = None - elif change and (isinstance(f, FileField) or not child_follow.get(f.name, None)): - if old_rel_obj: - param = getattr(old_rel_obj, f.column) - else: - param = f.get_default() - else: - param = f.get_manipulator_new_data(rel_new_data, rel=True) - if param != None: - params[f.attname] = param - - # Create the related item. - new_rel_obj = related.opts.get_model_module().Klass(**params) - - # If all the core fields were provided (non-empty), save the item. - if all_cores_given: - new_rel_obj.save() - - # Save any uploaded files. - for f in related.opts.fields: - if child_follow.get(f.name, None): - if isinstance(f, FileField) and rel_new_data.get(f.name, False): - f.save_file(rel_new_data, new_rel_obj, change and old_rel_obj or None, old_rel_obj is not None, rel=True) - - # Calculate whether any fields have changed. - if change: - if not old_rel_obj: # This object didn't exist before. - self.fields_added.append('%s "%s"' % (related.opts.verbose_name, new_rel_obj)) - else: - for f in related.opts.fields: - if not f.primary_key and f != related.field and str(getattr(old_rel_obj, f.attname)) != str(getattr(new_rel_obj, f.attname)): - self.fields_changed.append('%s for %s "%s"' % (f.verbose_name, related.opts.verbose_name, new_rel_obj)) - - # Save many-to-many objects. - for f in related.opts.many_to_many: - if child_follow.get(f.name, None) and not f.rel.edit_inline: - was_changed = getattr(new_rel_obj, 'set_%s' % f.name)(rel_new_data[f.attname]) - if change and was_changed: - self.fields_changed.append('%s for %s "%s"' % (f.verbose_name, related.opts.verbose_name, new_rel_obj)) - - # If, in the change stage, all of the core fields were blank and - # the primary key (ID) was provided, delete the item. - if change and all_cores_blank and old_rel_obj: - new_rel_obj.delete() - self.fields_deleted.append('%s "%s"' % (related.opts.verbose_name, old_rel_obj)) - - # Save the order, if applicable. - if change and opts.get_ordered_objects(): - order = new_data['order_'] and map(int, new_data['order_'].split(',')) or [] - for rel_opts in opts.get_ordered_objects(): - getattr(new_object, 'set_%s_order' % rel_opts.object_name.lower())(order) - return new_object - - def get_related_objects(opts, klass, add, change, self): - return opts.get_followed_related_objects(self.follow) - - def flatten_data(opts, klass, add, change, self): - new_data = {} - for f in opts.get_data_holders(self.follow): - fol = self.follow.get(f.name) - new_data.update(f.flatten_data(fol, self.original_object)) - return new_data - -class ModelAddManipulator(AutomaticManipulator): - change = False - add = True - -class ModelChangeManipulator(AutomaticManipulator): - change = False - add = True - - def __init__(self, obj_key=None, follow=None): - assert obj_key is not None, "ChangeManipulator.__init__() must be passed obj_key parameter." - self.obj_key = obj_key - try: - original_object = self.manager.get_object(pk=obj_key) - except ObjectDoesNotExist: - # If the object doesn't exist, this might be a manipulator for a - # one-to-one related object that hasn't created its subobject yet. - # For example, this might be a Restaurant for a Place that doesn't - # yet have restaurant information. - if opts.one_to_one_field: - # Sanity check -- Make sure the "parent" object exists. - # For example, make sure the Place exists for the Restaurant. - # Let the ObjectDoesNotExist exception propogate up. - lookup_kwargs = opts.one_to_one_field.rel.limit_choices_to - lookup_kwargs['%s__exact' % opts.one_to_one_field.rel.field_name] = obj_key - _ = opts.one_to_one_field.rel.to._meta.get_model_module().get_object(**lookup_kwargs) - params = dict([(f.attname, f.get_default()) for f in opts.fields]) - params[opts.pk.attname] = obj_key - original_object = opts.get_model_module().Klass(**params) - else: - raise - - super(ModelSaveManipulator, self).__init__(self, follow=follow) - self.original_object = original_object - - if opts.get_ordered_objects(): - self.fields.append(formfields.CommaSeparatedIntegerField(field_name="order_")) - -#def manipulator_init(opts, add, change, self, obj_key=None, follow=None): -# self.follow = opts.get_follow(follow) -# -# if change: -# assert obj_key is not None, "ChangeManipulator.__init__() must be passed obj_key parameter." -# self.obj_key = obj_key -# try: -# self.original_object = opts.get_model_module().get_object(pk=obj_key) -# except ObjectDoesNotExist: -# # If the object doesn't exist, this might be a manipulator for a -# # one-to-one related object that hasn't created its subobject yet. -# # For example, this might be a Restaurant for a Place that doesn't -# # yet have restaurant information. -# if opts.one_to_one_field: -# # Sanity check -- Make sure the "parent" object exists. -# # For example, make sure the Place exists for the Restaurant. -# # Let the ObjectDoesNotExist exception propogate up. -# lookup_kwargs = opts.one_to_one_field.rel.limit_choices_to -# lookup_kwargs['%s__exact' % opts.one_to_one_field.rel.field_name] = obj_key -# _ = opts.one_to_one_field.rel.to._meta.get_model_module().get_object(**lookup_kwargs) -# params = dict([(f.attname, f.get_default()) for f in opts.fields]) -# params[opts.pk.attname] = obj_key -# self.original_object = opts.get_model_module().Klass(**params) -# else: -# raise -# self.fields = [] -# -# for f in opts.fields + opts.many_to_many: -# if self.follow.get(f.name, False): -# self.fields.extend(f.get_manipulator_fields(opts, self, change)) -# -# # Add fields for related objects. -# for f in opts.get_all_related_objects(): -# if self.follow.get(f.name, False): -# fol = self.follow[f.name] -# self.fields.extend(f.get_manipulator_fields(opts, self, change, fol)) -# -# # Add field for ordering. -# if change and opts.get_ordered_objects(): -# self.fields.append(formfields.CommaSeparatedIntegerField(field_name="order_")) - -#def manipulator_save(opts, klass, add, change, self, new_data): -# # TODO: big cleanup when core fields go -> use recursive manipulators. -# from django.utils.datastructures import DotExpandedDict -# params = {} -# for f in opts.fields: -# # Fields with auto_now_add should keep their original value in the change stage. -# auto_now_add = change and getattr(f, 'auto_now_add', False) -# if self.follow.get(f.name, None) and not auto_now_add: -# param = f.get_manipulator_new_data(new_data) -# else: -# if change: -# param = getattr(self.original_object, f.attname) -# else: -# param = f.get_default() -# params[f.attname] = param -# -# if change: -# params[opts.pk.attname] = self.obj_key -# -# # First, save the basic object itself. -# new_object = klass(**params) -# new_object.save() -# -# # Now that the object's been saved, save any uploaded files. -# for f in opts.fields: -# if isinstance(f, FileField): -# f.save_file(new_data, new_object, change and self.original_object or None, change, rel=False) -# -# # Calculate which primary fields have changed. -# if change: -# self.fields_added, self.fields_changed, self.fields_deleted = [], [], [] -# for f in opts.fields: -# if not f.primary_key and str(getattr(self.original_object, f.attname)) != str(getattr(new_object, f.attname)): -# self.fields_changed.append(f.verbose_name) -# -# # Save many-to-many objects. Example: Poll.set_sites() -# for f in opts.many_to_many: -# if self.follow.get(f.name, None): -# if not f.rel.edit_inline: -# if f.rel.raw_id_admin: -# new_vals = new_data.get(f.name, ()) -# else: -# new_vals = new_data.getlist(f.name) -# was_changed = getattr(new_object, 'set_%s' % f.name)(new_vals) -# if change and was_changed: -# self.fields_changed.append(f.verbose_name) -# -# expanded_data = DotExpandedDict(dict(new_data)) -# # Save many-to-one objects. Example: Add the Choice objects for a Poll. -# for related in opts.get_all_related_objects(): -# # Create obj_list, which is a DotExpandedDict such as this: -# # [('0', {'id': ['940'], 'choice': ['This is the first choice']}), -# # ('1', {'id': ['941'], 'choice': ['This is the second choice']}), -# # ('2', {'id': [''], 'choice': ['']})] -# child_follow = self.follow.get(related.name, None) -# -# if child_follow: -# obj_list = expanded_data[related.var_name].items() -# obj_list.sort(lambda x, y: cmp(int(x[0]), int(y[0]))) -# params = {} -# -# # For each related item... -# for _, rel_new_data in obj_list: -# -# # Keep track of which core=True fields were provided. -# # If all core fields were given, the related object will be saved. -# # If none of the core fields were given, the object will be deleted. -# # If some, but not all, of the fields were given, the validator would -# # have caught that. -# all_cores_given, all_cores_blank = True, True -# -# # Get a reference to the old object. We'll use it to compare the -# # old to the new, to see which fields have changed. -# old_rel_obj = None -# if change: -# if rel_new_data[related.opts.pk.name][0]: -# try: -# old_rel_obj = getattr(self.original_object, 'get_%s' % related.get_method_name_part() )(**{'%s__exact' % related.opts.pk.name: rel_new_data[related.opts.pk.attname][0]}) -# except ObjectDoesNotExist: -# pass -# -# for f in related.opts.fields: -# if f.core and not isinstance(f, FileField) and f.get_manipulator_new_data(rel_new_data, rel=True) in (None, ''): -# all_cores_given = False -# elif f.core and not isinstance(f, FileField) and f.get_manipulator_new_data(rel_new_data, rel=True) not in (None, ''): -# all_cores_blank = False -# # If this field isn't editable, give it the same value it had -# # previously, according to the given ID. If the ID wasn't -# # given, use a default value. FileFields are also a special -# # case, because they'll be dealt with later. -# -# if f == related.field: -# param = getattr(new_object, related.field.rel.field_name) -# elif add and isinstance(f, AutoField): -# param = None -# elif change and (isinstance(f, FileField) or not child_follow.get(f.name, None)): -# if old_rel_obj: -# param = getattr(old_rel_obj, f.column) -# else: -# param = f.get_default() -# else: -# param = f.get_manipulator_new_data(rel_new_data, rel=True) -# if param != None: -# params[f.attname] = param -# -# # Create the related item. -# new_rel_obj = related.opts.get_model_module().Klass(**params) -# -# # If all the core fields were provided (non-empty), save the item. -# if all_cores_given: -# new_rel_obj.save() -# -# # Save any uploaded files. -# for f in related.opts.fields: -# if child_follow.get(f.name, None): -# if isinstance(f, FileField) and rel_new_data.get(f.name, False): -# f.save_file(rel_new_data, new_rel_obj, change and old_rel_obj or None, old_rel_obj is not None, rel=True) -# -# # Calculate whether any fields have changed. -# if change: -# if not old_rel_obj: # This object didn't exist before. -# self.fields_added.append('%s "%s"' % (related.opts.verbose_name, new_rel_obj)) -# else: -# for f in related.opts.fields: -# if not f.primary_key and f != related.field and str(getattr(old_rel_obj, f.attname)) != str(getattr(new_rel_obj, f.attname)): -# self.fields_changed.append('%s for %s "%s"' % (f.verbose_name, related.opts.verbose_name, new_rel_obj)) -# -# # Save many-to-many objects. -# for f in related.opts.many_to_many: -# if child_follow.get(f.name, None) and not f.rel.edit_inline: -# was_changed = getattr(new_rel_obj, 'set_%s' % f.name)(rel_new_data[f.attname]) -# if change and was_changed: -# self.fields_changed.append('%s for %s "%s"' % (f.verbose_name, related.opts.verbose_name, new_rel_obj)) -# -# # If, in the change stage, all of the core fields were blank and -# # the primary key (ID) was provided, delete the item. -# if change and all_cores_blank and old_rel_obj: -# new_rel_obj.delete() -# self.fields_deleted.append('%s "%s"' % (related.opts.verbose_name, old_rel_obj)) -# -# # Save the order, if applicable. -# if change and opts.get_ordered_objects(): -# order = new_data['order_'] and map(int, new_data['order_'].split(',')) or [] -# for rel_opts in opts.get_ordered_objects(): -# getattr(new_object, 'set_%s_order' % rel_opts.object_name.lower())(order) -# return new_object - def manipulator_validator_unique_together(field_name_list, opts, self, field_data, all_data): diff --git a/django/db/models/fields.py b/django/db/models/fields.py index 3c1c21e0f3..75707390eb 100644 --- a/django/db/models/fields.py +++ b/django/db/models/fields.py @@ -311,9 +311,10 @@ class Field(object): first_choice = include_blank and blank_choice or [] if self.choices: return first_choice + list(self.choices) + rel_model = self.rel.to return first_choice + [(getattr(x, rel_model._meta.pk.attname), str(x)) - for x in rel_model._default_manager.get_list(**self.rel._meta.limit_choices_to)] + for x in rel_model._default_manager.get_list(**rel_model._meta.limit_choices_to)] def get_choices_default(self): if(self.radio_admin):