from django.conf import settings from django.core import formfields, validators from django.core import db from django.core.exceptions import ObjectDoesNotExist from django.core.meta.fields import * from django.utils.functional import curry from django.utils.text import capfirst import copy, datetime, os, re, sys, types # Admin stages. ADD, CHANGE, BOTH = 1, 2, 3 # Size of each "chunk" for get_iterator calls. # Larger values are slightly faster at the expense of more storage space. GET_ITERATOR_CHUNK_SIZE = 100 # Prefix (in Python path style) to location of models. MODEL_PREFIX = 'django.models' # Methods on models with the following prefix will be removed and # converted to module-level functions. MODEL_FUNCTIONS_PREFIX = '_module_' # Methods on models with the following prefix will be removed and # converted to manipulator methods. MANIPULATOR_FUNCTIONS_PREFIX = '_manipulator_' LOOKUP_SEPARATOR = '__' #################### # HELPER FUNCTIONS # #################### # Django currently supports two forms of ordering. # Form 1 (deprecated) example: # order_by=(('pub_date', 'DESC'), ('headline', 'ASC'), (None, 'RANDOM')) # Form 2 (new-style) example: # order_by=('-pub_date', 'headline', '?') # Form 1 is deprecated and will no longer be supported for Django's first # official release. The following code converts from Form 1 to Form 2. LEGACY_ORDERING_MAPPING = {'ASC': '_', 'DESC': '-_', 'RANDOM': '?'} def handle_legacy_orderlist(order_list): if not order_list or isinstance(order_list[0], basestring): return order_list else: import warnings new_order_list = [LEGACY_ORDERING_MAPPING[j.upper()].replace('_', str(i)) for i, j in order_list] warnings.warn("%r ordering syntax is deprecated. Use %r instead." % (order_list, new_order_list), DeprecationWarning) return new_order_list def orderfield2column(f, opts): try: return opts.get_field(f, False).column except FieldDoesNotExist: return f def orderlist2sql(order_list, opts, prefix=''): if prefix.endswith('.'): prefix = db.db.quote_name(prefix[:-1]) + '.' output = [] for f in handle_legacy_orderlist(order_list): if f.startswith('-'): output.append('%s%s DESC' % (prefix, db.db.quote_name(orderfield2column(f[1:], opts)))) elif f == '?': output.append(db.get_random_function_sql()) else: output.append('%s%s ASC' % (prefix, db.db.quote_name(orderfield2column(f, opts)))) return ', '.join(output) def get_module(app_label, module_name): return __import__('%s.%s.%s' % (MODEL_PREFIX, app_label, module_name), '', '', ['']) def get_app(app_label): return __import__('%s.%s' % (MODEL_PREFIX, app_label), '', '', ['']) _installed_models_cache = None def get_installed_models(): """ Returns a list of installed "models" packages, such as foo.models, ellington.news.models, etc. This does NOT include django.models. """ global _installed_models_cache if _installed_models_cache is not None: return _installed_models_cache _installed_models_cache = [] for a in settings.INSTALLED_APPS: try: _installed_models_cache.append(__import__(a + '.models', '', '', [''])) except ImportError: pass return _installed_models_cache _installed_modules_cache = None def get_installed_model_modules(core_models=None): """ Returns a list of installed models, such as django.models.core, ellington.news.models.news, foo.models.bar, etc. """ global _installed_modules_cache if _installed_modules_cache is not None: return _installed_modules_cache _installed_modules_cache = [] # django.models is a special case. for submodule in (core_models or []): _installed_modules_cache.append(__import__('django.models.%s' % submodule, '', '', [''])) for m in get_installed_models(): for submodule in getattr(m, '__all__', []): mod = __import__('django.models.%s' % submodule, '', '', ['']) try: mod._MODELS except AttributeError: pass # Skip model modules that don't actually have models in them. else: _installed_modules_cache.append(mod) return _installed_modules_cache class LazyDate: """ Use in limit_choices_to to compare the field to dates calculated at run time instead of when the model is loaded. For example:: ... limit_choices_to = {'date__gt' : meta.LazyDate(days=-3)} ... which will limit the choices to dates greater than three days ago. """ def __init__(self, **kwargs): self.delta = datetime.timedelta(**kwargs) def __str__(self): return str(self.__get_value__()) def __repr__(self): return "" % self.delta def __get_value__(self): return datetime.datetime.now() + self.delta ################ # MAIN CLASSES # ################ class FieldDoesNotExist(Exception): pass class BadKeywordArguments(Exception): pass class Options: def __init__(self, module_name='', verbose_name='', verbose_name_plural='', db_table='', fields=None, ordering=None, unique_together=None, admin=None, has_related_links=False, where_constraints=None, object_name=None, app_label=None, exceptions=None, permissions=None, get_latest_by=None, order_with_respect_to=None, module_constants=None): # Save the original function args, for use by copy(). Note that we're # NOT using copy.deepcopy(), because that would create a new copy of # everything in memory, and it's better to conserve memory. Of course, # this comes with the important gotcha that changing any attribute of # this object will change its value in self._orig_init_args, so we # need to be careful not to do that. In practice, we can pull this off # because Options are generally read-only objects, and __init__() is # the only place where its attributes are manipulated. # locals() is used purely for convenience, so we don't have to do # something verbose like this: # self._orig_init_args = { # 'module_name': module_name, # 'verbose_name': verbose_name, # ... # } self._orig_init_args = locals() del self._orig_init_args['self'] # because we don't care about it. # Move many-to-many related fields from self.fields into self.many_to_many. self.fields, self.many_to_many = [], [] for field in (fields or []): if field.rel and isinstance(field.rel, ManyToMany): self.many_to_many.append(field) else: self.fields.append(field) self.module_name, self.verbose_name = module_name, verbose_name self.verbose_name_plural = verbose_name_plural or verbose_name + 's' self.db_table, self.has_related_links = db_table, has_related_links self.ordering = ordering or [] self.unique_together = unique_together or [] self.where_constraints = where_constraints or [] self.exceptions = exceptions or [] self.permissions = permissions or [] self.object_name, self.app_label = object_name, app_label self.get_latest_by = get_latest_by if order_with_respect_to: self.order_with_respect_to = self.get_field(order_with_respect_to) self.ordering = ('_order',) else: self.order_with_respect_to = None self.module_constants = module_constants or {} self.admin = admin # 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: self.fields.insert(0, AutoField(name='id', verbose_name='ID', primary_key=True)) 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 def __repr__(self): return '' % self.module_name def copy(self, **kwargs): args = self._orig_init_args.copy() args.update(kwargs) return self.__class__(**args) def get_model_module(self): return get_module(self.app_label, self.module_name) 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 return self._content_type_id def get_field(self, name, many_to_many=True): """ Returns the requested field by name. Raises FieldDoesNotExist on error. """ to_search = many_to_many and (self.fields + self.many_to_many) or self.fields for f in to_search: if f.name == name: return f raise FieldDoesNotExist, "name=%s" % name def get_order_sql(self, table_prefix=''): "Returns the full 'ORDER BY' clause for this object, according to self.ordering." if not self.ordering: return '' pre = table_prefix and (table_prefix + '.') or '' return 'ORDER BY ' + orderlist2sql(self.ordering, self, pre) def get_add_permission(self): return 'add_%s' % self.object_name.lower() def get_change_permission(self): return 'change_%s' % self.object_name.lower() def get_delete_permission(self): return 'delete_%s' % self.object_name.lower() def get_rel_object_method_name(self, rel_opts, rel_field): # This method encapsulates the logic that decides what name to give a # method that retrieves related many-to-one objects. Usually it just # uses the lower-cased object_name, but if the related object is in # another app, its app_label is appended. # # Examples: # # # Normal case -- a related object in the same app. # # This method returns "choice". # Poll.get_choice_list() # # # A related object in a different app. # # This method returns "lcom_bestofaward". # Place.get_lcom_bestofaward_list() # "lcom_bestofaward" rel_obj_name = rel_field.rel.related_name or rel_opts.object_name.lower() if self.app_label != rel_opts.app_label: rel_obj_name = '%s_%s' % (rel_opts.app_label, rel_obj_name) return rel_obj_name def get_all_related_objects(self): try: # Try the cache first. return self._all_related_objects except AttributeError: module_list = get_installed_model_modules() rel_objs = [] for mod in module_list: for klass in mod._MODELS: for f in klass._meta.fields: if f.rel and self == f.rel.to: rel_objs.append((klass._meta, f)) if self.has_related_links: # Manually add RelatedLink objects, which are a special case. relatedlinks = get_module('relatedlinks', 'relatedlinks') # Note that the copy() is very important -- otherwise any # subsequently loaded object with related links will override this # relationship we're adding. link_field = copy.copy(relatedlinks.RelatedLink._meta.get_field('object_id')) link_field.rel = ManyToOne(self.get_model_module().Klass, 'id', num_in_admin=3, min_num_in_admin=3, edit_inline=TABULAR, lookup_overrides={ 'content_type__package__label__exact': self.app_label, 'content_type__python_module_name__exact': self.module_name, }) rel_objs.append((relatedlinks.RelatedLink._meta, link_field)) self._all_related_objects = rel_objs return rel_objs def get_inline_related_objects(self): return [(a, b) for a, b in self.get_all_related_objects() if b.rel.edit_inline] def get_all_related_many_to_many_objects(self): module_list = get_installed_model_modules() rel_objs = [] for mod in module_list: for klass in mod._MODELS: try: for f in klass._meta.many_to_many: if f.rel and self == f.rel.to: rel_objs.append((klass._meta, f)) raise StopIteration except StopIteration: continue return rel_objs def get_ordered_objects(self): "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: objects.append(opts) self._ordered_objects = objects return self._ordered_objects def has_field_type(self, field_type): """ Returns True if this object's admin form has at least one of the given field_type (e.g. FileField). """ if not hasattr(self, '_field_types'): self._field_types = {} if not self._field_types.has_key(field_type): try: # First check self.fields. for f in self.fields: if isinstance(f, field_type): raise StopIteration # Failing that, check related fields. for rel_obj, rel_field in self.get_inline_related_objects(): for f in rel_obj.fields: if isinstance(f, field_type): raise StopIteration except StopIteration: self._field_types[field_type] = True else: self._field_types[field_type] = False return self._field_types[field_type] def _reassign_globals(function_dict, extra_globals, namespace): new_functions = {} for k, v in function_dict.items(): # Get the code object. code = v.func_code # Recreate the function, but give it access to extra_globals and the # given namespace's globals, too. new_globals = {'__builtins__': __builtins__, 'db': db.db, 'datetime': datetime} new_globals.update(extra_globals.__dict__) func = types.FunctionType(code, globals=new_globals, name=k, argdefs=v.func_defaults) func.__dict__.update(v.__dict__) setattr(namespace, k, func) # For all of the custom functions that have been added so far, give # them access to the new function we've just created. for new_k, new_v in new_functions.items(): new_v.func_globals[k] = func new_functions[k] = func # Calculate the module_name using a poor-man's pluralization. get_module_name = lambda class_name: class_name.lower() + 's' # Calculate the verbose_name by converting from InitialCaps to "lowercase with spaces". get_verbose_name = lambda class_name: re.sub('([A-Z])', ' \\1', class_name).lower().strip() class ModelBase(type): "Metaclass for all models" def __new__(cls, name, bases, attrs): # If this isn't a subclass of Model, don't do anything special. if not bases: return type.__new__(cls, name, bases, attrs) try: meta_attrs = attrs.pop('META').__dict__ del meta_attrs['__module__'] del meta_attrs['__doc__'] except KeyError: meta_attrs = {} # Gather all attributes that are Field instances. fields = [] for obj_name, obj in attrs.items(): if isinstance(obj, Field): obj.set_name(obj_name) fields.append(obj) del attrs[obj_name] # Sort the fields in the order that they were created. The # "creation_counter" is needed because metaclasses don't preserve the # attribute order. fields.sort(lambda x, y: x.creation_counter - y.creation_counter) # If this model is a subclass of another model, create an Options # object by first copying the base class's _meta and then updating it # with the overrides from this class. replaces_module = None if bases[0] != Model: field_names = [f.name for f in fields] remove_fields = meta_attrs.pop('remove_fields', []) for f in bases[0]._meta._orig_init_args['fields']: if f.name not in field_names and f.name not in remove_fields: fields.insert(0, f) if meta_attrs.has_key('replaces_module'): # Set the replaces_module variable for now. We can't actually # do anything with it yet, because the module hasn't yet been # created. replaces_module = meta_attrs.pop('replaces_module').split('.') # Pass any Options overrides to the base's Options instance, and # simultaneously remove them from attrs. When this is done, attrs # will be a dictionary of custom methods, plus __module__. meta_overrides = {'fields': fields, 'module_name': get_module_name(name), 'verbose_name': get_verbose_name(name)} for k, v in meta_attrs.items(): if not callable(v) and k != '__module__': meta_overrides[k] = meta_attrs.pop(k) opts = bases[0]._meta.copy(**meta_overrides) opts.object_name = name del meta_overrides else: opts = Options( module_name = meta_attrs.pop('module_name', get_module_name(name)), # If the verbose_name wasn't given, use the class name, # converted from InitialCaps to "lowercase with spaces". verbose_name = meta_attrs.pop('verbose_name', get_verbose_name(name)), verbose_name_plural = meta_attrs.pop('verbose_name_plural', ''), db_table = meta_attrs.pop('db_table', ''), fields = fields, ordering = meta_attrs.pop('ordering', None), unique_together = meta_attrs.pop('unique_together', None), admin = meta_attrs.pop('admin', None), has_related_links = meta_attrs.pop('has_related_links', False), where_constraints = meta_attrs.pop('where_constraints', None), object_name = name, app_label = meta_attrs.pop('app_label', None), exceptions = meta_attrs.pop('exceptions', None), permissions = meta_attrs.pop('permissions', None), get_latest_by = meta_attrs.pop('get_latest_by', None), order_with_respect_to = meta_attrs.pop('order_with_respect_to', None), module_constants = meta_attrs.pop('module_constants', None), ) if meta_attrs != {}: raise TypeError, "'class META' got invalid attribute(s): %s" % ','.join(meta_attrs.keys()) # Dynamically create the module that will contain this class and its # associated helper functions. if replaces_module is not None: new_mod = get_module(*replaces_module) else: new_mod = types.ModuleType(opts.module_name) # Collect any/all custom class methods and module functions, and move # them to a temporary holding variable. We'll deal with them later. if replaces_module is not None: # Initialize these values to the base class' custom_methods and # custom_functions. custom_methods = dict([(k, v) for k, v in new_mod.Klass.__dict__.items() if hasattr(v, 'custom')]) custom_functions = dict([(k, v) for k, v in new_mod.__dict__.items() if hasattr(v, 'custom')]) else: custom_methods, custom_functions = {}, {} manipulator_methods = {} for k, v in attrs.items(): if k in ('__module__', '__init__', '_overrides', '__doc__'): continue # Skip the important stuff. assert callable(v), "%r is an invalid model parameter." % k # Give the function a function attribute "custom" to designate that # it's a custom function/method. v.custom = True if k.startswith(MODEL_FUNCTIONS_PREFIX): custom_functions[k[len(MODEL_FUNCTIONS_PREFIX):]] = v elif k.startswith(MANIPULATOR_FUNCTIONS_PREFIX): manipulator_methods[k[len(MANIPULATOR_FUNCTIONS_PREFIX):]] = v else: custom_methods[k] = v del attrs[k] # Create the module-level ObjectDoesNotExist exception. dne_exc_name = '%sDoesNotExist' % name does_not_exist_exception = types.ClassType(dne_exc_name, (ObjectDoesNotExist,), {}) # Explicitly set its __module__ because it will initially (incorrectly) # be set to the module the code is being executed in. does_not_exist_exception.__module__ = MODEL_PREFIX + '.' + opts.module_name setattr(new_mod, dne_exc_name, does_not_exist_exception) # Create other exceptions. for exception_name in opts.exceptions: exc = types.ClassType(exception_name, (Exception,), {}) exc.__module__ = MODEL_PREFIX + '.' + opts.module_name # Set this explicitly, as above. setattr(new_mod, exception_name, exc) # Create any module-level constants, if applicable. for k, v in opts.module_constants.items(): setattr(new_mod, k, v) # Create the default class methods. attrs['__init__'] = curry(method_init, opts) attrs['__eq__'] = curry(method_eq, opts) attrs['save'] = curry(method_save, opts) attrs['save'].alters_data = True attrs['delete'] = curry(method_delete, opts) attrs['delete'].alters_data = True if opts.order_with_respect_to: attrs['get_next_in_order'] = curry(method_get_next_in_order, opts, opts.order_with_respect_to) attrs['get_previous_in_order'] = curry(method_get_previous_in_order, opts, opts.order_with_respect_to) for f in opts.fields: # If the object has a relationship to itself, as designated by # RECURSIVE_RELATIONSHIP_CONSTANT, create that relationship formally. if f.rel and f.rel.to == RECURSIVE_RELATIONSHIP_CONSTANT: f.rel.to = opts f.name = f.name or (f.rel.to.object_name.lower() + '_' + f.rel.to.pk.name) f.verbose_name = f.verbose_name or f.rel.to.verbose_name f.rel.field_name = f.rel.field_name or f.rel.to.pk.name # Add "get_thingie" methods for many-to-one related objects. # EXAMPLES: Choice.get_poll(), Story.get_dateline() if isinstance(f.rel, ManyToOne): func = curry(method_get_many_to_one, f) func.__doc__ = "Returns the associated `%s.%s` object." % (f.rel.to.app_label, f.rel.to.module_name) attrs['get_%s' % f.name] = func for f in opts.many_to_many: # Add "get_thingie" methods for many-to-many related objects. # EXAMPLES: Poll.get_site_list(), Story.get_byline_list() func = curry(method_get_many_to_many, f) func.__doc__ = "Returns a list of associated `%s.%s` objects." % (f.rel.to.app_label, f.rel.to.module_name) attrs['get_%s_list' % f.rel.singular] = func # Add "set_thingie" methods for many-to-many related objects. # EXAMPLES: Poll.set_sites(), Story.set_bylines() func = curry(method_set_many_to_many, f) func.__doc__ = "Resets this object's `%s.%s` list to the given list of IDs. Note that it doesn't check whether the given IDs are valid." % (f.rel.to.app_label, f.rel.to.module_name) func.alters_data = True attrs['set_%s' % f.name] = func # Create the class, because we need it to use in currying. new_class = type.__new__(cls, name, bases, attrs) # Give the class a docstring -- its definition. if new_class.__doc__ is None: new_class.__doc__ = "%s.%s(%s)" % (opts.module_name, name, ", ".join([f.name for f in opts.fields])) # Create the standard, module-level API helper functions such # as get_object() and get_list(). new_mod.get_object = curry(function_get_object, opts, new_class, does_not_exist_exception) new_mod.get_object.__doc__ = "Returns the %s object matching the given parameters." % name new_mod.get_list = curry(function_get_list, opts, new_class) new_mod.get_list.__doc__ = "Returns a list of %s objects matching the given parameters." % name new_mod.get_iterator = curry(function_get_iterator, opts, new_class) new_mod.get_iterator.__doc__ = "Returns an iterator of %s objects matching the given parameters." % name new_mod.get_values = curry(function_get_values, opts, new_class) new_mod.get_values.__doc__ = "Returns a list of dictionaries matching the given parameters." new_mod.get_values_iterator = curry(function_get_values_iterator, opts, new_class) new_mod.get_values_iterator.__doc__ = "Returns an iterator of dictionaries matching the given parameters." new_mod.get_count = curry(function_get_count, opts) new_mod.get_count.__doc__ = "Returns the number of %s objects matching the given parameters." % name new_mod._get_sql_clause = curry(function_get_sql_clause, opts) new_mod.get_in_bulk = curry(function_get_in_bulk, opts, new_class) new_mod.get_in_bulk.__doc__ = "Returns a dictionary of ID -> %s for the %s objects with IDs in the given id_list." % (name, name) if opts.get_latest_by: new_mod.get_latest = curry(function_get_latest, opts, new_class, does_not_exist_exception) for f in opts.fields: if f.choices: # Add "get_thingie_display" method to get human-readable value. func = curry(method_get_display_value, f) setattr(new_class, 'get_%s_display' % f.name, func) if isinstance(f, DateField) or isinstance(f, DateTimeField): # Add "get_next_by_thingie" and "get_previous_by_thingie" methods # for all DateFields and DateTimeFields that cannot be null. # EXAMPLES: Poll.get_next_by_pub_date(), Poll.get_previous_by_pub_date() if not f.null: setattr(new_class, 'get_next_by_%s' % f.name, curry(method_get_next_or_previous, new_mod.get_object, opts, f, True)) setattr(new_class, 'get_previous_by_%s' % f.name, curry(method_get_next_or_previous, new_mod.get_object, opts, f, False)) # Add "get_thingie_list" for all DateFields and DateTimeFields. # EXAMPLE: polls.get_pub_date_list() func = curry(function_get_date_list, opts, f) func.__doc__ = "Returns a list of days, months or years (as datetime.datetime objects) in which %s objects are available. The first parameter ('kind') must be one of 'year', 'month' or 'day'." % name setattr(new_mod, 'get_%s_list' % f.name, func) elif isinstance(f, FileField): setattr(new_class, 'get_%s_filename' % f.name, curry(method_get_file_filename, f)) setattr(new_class, 'get_%s_url' % f.name, curry(method_get_file_url, f)) setattr(new_class, 'get_%s_size' % f.name, curry(method_get_file_size, f)) func = curry(method_save_file, f) func.alters_data = True setattr(new_class, 'save_%s_file' % f.name, func) if isinstance(f, ImageField): # Add get_BLAH_width and get_BLAH_height methods, but only # if the image field doesn't have width and height cache # fields. if not f.width_field: setattr(new_class, 'get_%s_width' % f.name, curry(method_get_image_width, f)) if not f.height_field: setattr(new_class, 'get_%s_height' % f.name, curry(method_get_image_height, f)) # Add the class itself to the new module we've created. new_mod.__dict__[name] = new_class # Add "Klass" -- a shortcut reference to the class. new_mod.__dict__['Klass'] = new_class # Add the Manipulators. new_mod.__dict__['AddManipulator'] = get_manipulator(opts, new_class, manipulator_methods, add=True) new_mod.__dict__['ChangeManipulator'] = get_manipulator(opts, new_class, manipulator_methods, change=True) # Now that we have references to new_mod and new_class, we can add # any/all extra class methods to the new class. Note that we could # have just left the extra methods in attrs (above), but that would # have meant that any code within the extra methods would *not* have # access to module-level globals, such as get_list(), db, etc. # In order to give these methods access to those globals, we have to # deconstruct the method getting its raw "code" object, then recreating # the function with a new "globals" dictionary. # # To complicate matters more, because each method is manually assigned # a "globals" value, that "globals" value does NOT include the methods # that haven't been created yet. For instance, if there are two custom # methods, foo() and bar(), and foo() is created first, it won't have # bar() within its globals(). This is a problem because sometimes # custom methods/functions refer to other custom methods/functions. To # solve this problem, we keep track of the new functions created (in # the new_functions variable) and manually append each new function to # the func_globals() of all previously-created functions. So, by the # end of the loop, all functions will "know" about all the other # functions. _reassign_globals(custom_methods, new_mod, new_class) _reassign_globals(custom_functions, new_mod, new_mod) _reassign_globals(manipulator_methods, new_mod, new_mod.__dict__['AddManipulator']) _reassign_globals(manipulator_methods, new_mod, new_mod.__dict__['ChangeManipulator']) if hasattr(new_class, 'get_absolute_url'): new_class.get_absolute_url = curry(get_absolute_url, opts, new_class.get_absolute_url) # Get a reference to the module the class is in, and dynamically add # the new module to it. app_package = sys.modules.get(new_class.__module__) if replaces_module is not None: app_label = replaces_module[0] else: app_package.__dict__[opts.module_name] = new_mod app_label = app_package.__name__[app_package.__name__.rfind('.')+1:] # Populate the _MODELS member on the module the class is in. # Example: django.models.polls will have a _MODELS member that will # contain this list: # [, ] # Don't do this if replaces_module is set. app_package.__dict__.setdefault('_MODELS', []).append(new_class) # Cache the app label. opts.app_label = app_label # If the db_table wasn't provided, use the app_label + module_name. if not opts.db_table: opts.db_table = "%s_%s" % (app_label, opts.module_name) new_class._meta = opts # Set the __file__ attribute to the __file__ attribute of its package, # because they're technically from the same file. Note: if we didn't # set this, sys.modules would think this module was built-in. try: new_mod.__file__ = app_package.__file__ except AttributeError: # 'module' object has no attribute '__file__', which means the # class was probably being entered via the interactive interpreter. pass # Add the module's entry to sys.modules -- for instance, # "django.models.polls.polls". Note that "django.models.polls" has already # been added automatically. sys.modules.setdefault('%s.%s.%s' % (MODEL_PREFIX, app_label, opts.module_name), new_mod) # If this module replaces another one, get a reference to the other # module's parent, and replace the other module with the one we've just # created. if replaces_module is not None: old_app = get_app(replaces_module[0]) setattr(old_app, replaces_module[1], new_mod) for i, model in enumerate(old_app._MODELS): if model._meta.module_name == replaces_module[1]: # Replace the appropriate member of the old app's _MODELS # data structure. old_app._MODELS[i] = new_class # Replace all relationships to the old class with # relationships to the new one. for rel_opts, rel_field in model._meta.get_all_related_objects(): rel_field.rel.to = opts for rel_opts, rel_field in model._meta.get_all_related_many_to_many_objects(): rel_field.rel.to = opts break return new_class class Model: __metaclass__ = ModelBase def __repr__(self): return '<%s object>' % self.__class__.__name__ ############################################ # HELPER FUNCTIONS (CURRIED MODEL METHODS) # ############################################ # CORE METHODS ############################# def method_init(opts, self, *args, **kwargs): if kwargs: for f in opts.fields: if isinstance(f.rel, ManyToOne): try: # Assume object instance was passed in. rel_obj = kwargs.pop(f.name) except KeyError: try: # Object instance wasn't passed in -- must be an ID. val = kwargs.pop(f.attname) except KeyError: val = f.get_default() else: # Object instance was passed in. # Special case: You can pass in "None" for related objects if it's allowed. if rel_obj is None and f.null: val = None else: try: val = getattr(rel_obj, f.rel.get_related_field().attname) except AttributeError: raise TypeError, "Invalid value: %r should be a %s instance, not a %s" % (f.name, f.rel.to, type(rel_obj)) setattr(self, f.attname, val) else: val = kwargs.pop(f.attname, f.get_default()) setattr(self, f.attname, val) if kwargs: raise TypeError, "'%s' is an invalid keyword argument for this function" % kwargs.keys()[0] for i, arg in enumerate(args): setattr(self, opts.fields[i].attname, arg) def method_eq(opts, self, other): return isinstance(other, self.__class__) and getattr(self, opts.pk.attname) == getattr(other, opts.pk.attname) def method_save(opts, self): # Run any pre-save hooks. if hasattr(self, '_pre_save'): self._pre_save() non_pks = [f for f in opts.fields if not f.primary_key] cursor = db.db.cursor() # First, try an UPDATE. If that doesn't update anything, do an INSERT. pk_val = getattr(self, opts.pk.attname) pk_set = bool(pk_val) record_exists = True if pk_set: # Determine whether a record with the primary key already exists. cursor.execute("SELECT 1 FROM %s WHERE %s=%%s LIMIT 1" % \ (db.db.quote_name(opts.db_table), db.db.quote_name(opts.pk.column)), [pk_val]) # If it does already exist, do an UPDATE. if cursor.fetchone(): db_values = [f.get_db_prep_save(f.pre_save(getattr(self, f.attname), False)) for f in non_pks] cursor.execute("UPDATE %s SET %s WHERE %s=%%s" % \ (db.db.quote_name(opts.db_table), ','.join(['%s=%%s' % db.db.quote_name(f.column) for f in non_pks]), db.db.quote_name(opts.pk.attname)), db_values + [pk_val]) else: record_exists = False if not pk_set or not record_exists: field_names = [db.db.quote_name(f.column) for f in opts.fields if not isinstance(f, AutoField)] placeholders = ['%s'] * len(field_names) db_values = [f.get_db_prep_save(f.pre_save(getattr(self, f.attname), True)) for f in opts.fields if not isinstance(f, AutoField)] if opts.order_with_respect_to: field_names.append(db.db.quote_name('_order')) # TODO: This assumes the database supports subqueries. placeholders.append('(SELECT COUNT(*) FROM %s WHERE %s = %%s)' % \ (db.db.quote_name(opts.db_table), db.db.quote_name(opts.order_with_respect_to.column))) db_values.append(getattr(self, opts.order_with_respect_to.attname)) cursor.execute("INSERT INTO %s (%s) VALUES (%s)" % \ (db.db.quote_name(opts.db_table), ','.join(field_names), ','.join(placeholders)), db_values) if opts.has_auto_field: setattr(self, opts.pk.attname, db.get_last_insert_id(cursor, opts.db_table, opts.pk.column)) db.db.commit() # Run any post-save hooks. if hasattr(self, '_post_save'): self._post_save() def method_delete(opts, self): assert getattr(self, opts.pk.attname) is not None, "%r can't be deleted because it doesn't have an ID." # Run any pre-delete hooks. if hasattr(self, '_pre_delete'): self._pre_delete() cursor = db.db.cursor() for rel_opts, rel_field in opts.get_all_related_objects(): rel_opts_name = opts.get_rel_object_method_name(rel_opts, rel_field) if isinstance(rel_field.rel, OneToOne): try: sub_obj = getattr(self, 'get_%s' % rel_opts_name)() except ObjectDoesNotExist: pass else: sub_obj.delete() else: for sub_obj in getattr(self, 'get_%s_list' % rel_opts_name)(): sub_obj.delete() for rel_opts, rel_field in opts.get_all_related_many_to_many_objects(): cursor.execute("DELETE FROM %s WHERE %s=%%s" % \ (db.db.quote_name(rel_field.get_m2m_db_table(rel_opts)), db.db.quote_name(self._meta.object_name.lower() + '_id')), [getattr(self, opts.pk.attname)]) for f in opts.many_to_many: cursor.execute("DELETE FROM %s WHERE %s=%%s" % \ (db.db.quote_name(f.get_m2m_db_table(opts)), db.db.quote_name(self._meta.object_name.lower() + '_id')), [getattr(self, opts.pk.attname)]) cursor.execute("DELETE FROM %s WHERE %s=%%s" % \ (db.db.quote_name(opts.db_table), db.db.quote_name(opts.pk.column)), [getattr(self, opts.pk.attname)]) db.db.commit() setattr(self, opts.pk.attname, None) for f in opts.fields: if isinstance(f, FileField) and getattr(self, f.attname): file_name = getattr(self, 'get_%s_filename' % f.name)() # If the file exists and no other object of this type references it, # delete it from the filesystem. if os.path.exists(file_name) and not opts.get_model_module().get_list(**{'%s__exact' % f.name: getattr(self, f.name)}): os.remove(file_name) # Run any post-delete hooks. if hasattr(self, '_post_delete'): self._post_delete() def method_get_next_in_order(opts, order_field, self): if not hasattr(self, '_next_in_order_cache'): self._next_in_order_cache = opts.get_model_module().get_object(order_by=('_order',), where=['%s > (SELECT %s FROM %s WHERE %s=%%s)' % \ (db.db.quote_name('_order'), db.db.quote_name('_order'), db.db.quote_name(opts.db_table), db.db.quote_name(opts.pk.column)), '%s=%%s' % db.db.quote_name(order_field.column)], limit=1, params=[getattr(self, opts.pk.attname), getattr(self, order_field.attname)]) return self._next_in_order_cache def method_get_previous_in_order(opts, order_field, self): if not hasattr(self, '_previous_in_order_cache'): self._previous_in_order_cache = opts.get_model_module().get_object(order_by=('-_order',), where=['%s < (SELECT %s FROM %s WHERE %s=%%s)' % \ (db.db.quote_name('_order'), db.db.quote_name('_order'), db.db.quote_name(opts.db_table), db.db.quote_name(opts.pk.column)), '%s=%%s' % db.db.quote_name(order_field.column)], limit=1, params=[getattr(self, opts.pk.attname), getattr(self, order_field.attname)]) return self._previous_in_order_cache # RELATIONSHIP METHODS ##################### # Example: Story.get_dateline() def method_get_many_to_one(field_with_rel, self): cache_var = field_with_rel.get_cache_name() if not hasattr(self, cache_var): val = getattr(self, field_with_rel.attname) mod = field_with_rel.rel.to.get_model_module() if val is None: raise getattr(mod, '%sDoesNotExist' % field_with_rel.rel.to.object_name) other_field = field_with_rel.rel.get_related_field() if other_field.rel: params = {'%s__%s__exact' % (field_with_rel.rel.field_name, other_field.rel.field_name): val} else: params = {'%s__exact'% field_with_rel.rel.field_name: val} retrieved_obj = mod.get_object(**params) setattr(self, cache_var, retrieved_obj) return getattr(self, cache_var) # Handles getting many-to-many related objects. # Example: Poll.get_site_list() def method_get_many_to_many(field_with_rel, self): rel = field_with_rel.rel.to cache_var = '_%s_cache' % field_with_rel.name if not hasattr(self, cache_var): mod = rel.get_model_module() sql = "SELECT %s FROM %s a, %s b WHERE a.%s = b.%s AND b.%s = %%s %s" % \ (','.join(['a.%s' % db.db.quote_name(f.column) for f in rel.fields]), db.db.quote_name(rel.db_table), db.db.quote_name(field_with_rel.get_m2m_db_table(self._meta)), db.db.quote_name(rel.pk.column), db.db.quote_name(rel.object_name.lower() + '_id'), db.db.quote_name(self._meta.object_name.lower() + '_id'), rel.get_order_sql('a')) cursor = db.db.cursor() cursor.execute(sql, [getattr(self, self._meta.pk.attname)]) setattr(self, cache_var, [getattr(mod, rel.object_name)(*row) for row in cursor.fetchall()]) return getattr(self, cache_var) # Handles setting many-to-many relationships. # Example: Poll.set_sites() def method_set_many_to_many(rel_field, self, id_list): current_ids = [obj.id for obj in method_get_many_to_many(rel_field, self)] ids_to_add, ids_to_delete = dict([(i, 1) for i in id_list]), [] for current_id in current_ids: if current_id in id_list: del ids_to_add[current_id] else: ids_to_delete.append(current_id) ids_to_add = ids_to_add.keys() # Now ids_to_add is a list of IDs to add, and ids_to_delete is a list of IDs to delete. if not ids_to_delete and not ids_to_add: return False # No change rel = rel_field.rel.to m2m_table = rel_field.get_m2m_db_table(self._meta) cursor = db.db.cursor() this_id = getattr(self, self._meta.pk.attname) if ids_to_delete: sql = "DELETE FROM %s WHERE %s = %%s AND %s IN (%s)" % \ (db.db.quote_name(m2m_table), db.db.quote_name(self._meta.object_name.lower() + '_id'), db.db.quote_name(rel.object_name.lower() + '_id'), ','.join(map(str, ids_to_delete))) cursor.execute(sql, [this_id]) if ids_to_add: sql = "INSERT INTO %s (%s, %s) VALUES (%%s, %%s)" % \ (db.db.quote_name(m2m_table), db.db.quote_name(self._meta.object_name.lower() + '_id'), db.db.quote_name(rel.object_name.lower() + '_id')) cursor.executemany(sql, [(this_id, i) for i in ids_to_add]) db.db.commit() try: delattr(self, '_%s_cache' % rel_field.name) # clear cache, if it exists except AttributeError: pass return True # Handles related-object retrieval. # Examples: Poll.get_choice(), Poll.get_choice_list(), Poll.get_choice_count() def method_get_related(method_name, rel_mod, rel_field, self, **kwargs): if self._meta.has_related_links and rel_mod.Klass._meta.module_name == 'relatedlinks': kwargs['object_id__exact'] = getattr(self, rel_field.rel.field_name) else: kwargs['%s__%s__exact' % (rel_field.name, rel_field.rel.to.pk.name)] = getattr(self, rel_field.rel.get_related_field().attname) kwargs.update(rel_field.rel.lookup_overrides) return getattr(rel_mod, method_name)(**kwargs) # Handles adding related objects. # Example: Poll.add_choice() def method_add_related(rel_obj, rel_mod, rel_field, self, *args, **kwargs): init_kwargs = dict(zip([f.attname for f in rel_obj.fields if f != rel_field and not isinstance(f, AutoField)], args)) init_kwargs.update(kwargs) for f in rel_obj.fields: if isinstance(f, AutoField): init_kwargs[f.attname] = None init_kwargs[rel_field.name] = self obj = rel_mod.Klass(**init_kwargs) obj.save() return obj # Handles related many-to-many object retrieval. # Examples: Album.get_song(), Album.get_song_list(), Album.get_song_count() def method_get_related_many_to_many(method_name, opts, rel_mod, rel_field, self, **kwargs): kwargs['%s__%s__exact' % (rel_field.name, opts.pk.name)] = getattr(self, opts.pk.attname) return getattr(rel_mod, method_name)(**kwargs) # Handles setting many-to-many related objects. # Example: Album.set_songs() def method_set_related_many_to_many(rel_opts, rel_field, self, id_list): id_list = map(int, id_list) # normalize to integers rel = rel_field.rel.to m2m_table = rel_field.get_m2m_db_table(rel_opts) this_id = getattr(self, self._meta.pk.attname) cursor = db.db.cursor() cursor.execute("DELETE FROM %s WHERE %s = %%s" % \ (db.db.quote_name(m2m_table), db.db.quote_name(rel.object_name.lower() + '_id')), [this_id]) sql = "INSERT INTO %s (%s, %s) VALUES (%%s, %%s)" % \ (db.db.quote_name(m2m_table), db.db.quote_name(rel.object_name.lower() + '_id'), db.db.quote_name(rel_opts.object_name.lower() + '_id')) cursor.executemany(sql, [(this_id, i) for i in id_list]) db.db.commit() # ORDERING METHODS ######################### def method_set_order(ordered_obj, self, id_list): cursor = db.db.cursor() # Example: "UPDATE poll_choices SET _order = %s WHERE poll_id = %s AND id = %s" sql = "UPDATE %s SET %s = %%s WHERE %s = %%s AND %s = %%s" % \ (db.db.quote_name(ordered_obj.db_table), db.db.quote_name('_order'), db.db.quote_name(ordered_obj.order_with_respect_to.column), db.db.quote_name(ordered_obj.pk.column)) rel_val = getattr(self, ordered_obj.order_with_respect_to.rel.field_name) cursor.executemany(sql, [(i, rel_val, j) for i, j in enumerate(id_list)]) db.db.commit() def method_get_order(ordered_obj, self): cursor = db.db.cursor() # Example: "SELECT id FROM poll_choices WHERE poll_id = %s ORDER BY _order" sql = "SELECT %s FROM %s WHERE %s = %%s ORDER BY %s" % \ (db.db.quote_name(ordered_obj.pk.column), db.db.quote_name(ordered_obj.db_table), db.db.quote_name(ordered_obj.order_with_respect_to.column), db.db.quote_name('_order')) rel_val = getattr(self, ordered_obj.order_with_respect_to.rel.field_name) cursor.execute(sql, [rel_val]) return [r[0] for r in cursor.fetchall()] # DATE-RELATED METHODS ##################### def method_get_next_or_previous(get_object_func, opts, field, is_next, self, **kwargs): op = is_next and '>' or '<' kwargs.setdefault('where', []).append('(%s %s %%s OR (%s = %%s AND %s %s %%s))' % \ (db.db.quote_name(field.column), op, db.db.quote_name(field.column), db.db.quote_name(opts.pk.column), op)) param = str(getattr(self, field.attname)) kwargs.setdefault('params', []).extend([param, param, getattr(self, opts.pk.attname)]) kwargs['order_by'] = [(not is_next and '-' or '') + field.name, (not is_next and '-' or '') + opts.pk.name] kwargs['limit'] = 1 return get_object_func(**kwargs) # CHOICE-RELATED METHODS ################### def method_get_display_value(field, self): value = getattr(self, field.attname) return dict(field.choices).get(value, value) # FILE-RELATED METHODS ##################### def method_get_file_filename(field, self): return os.path.join(settings.MEDIA_ROOT, getattr(self, field.attname)) def method_get_file_url(field, self): if getattr(self, field.attname): # value is not blank import urlparse return urlparse.urljoin(settings.MEDIA_URL, getattr(self, field.attname)).replace('\\', '/') return '' def method_get_file_size(field, self): return os.path.getsize(method_get_file_filename(field, self)) def method_save_file(field, self, filename, raw_contents): directory = field.get_directory_name() try: # Create the date-based directory if it doesn't exist. os.makedirs(os.path.join(settings.MEDIA_ROOT, directory)) except OSError: # Directory probably already exists. pass filename = field.get_filename(filename) # If the filename already exists, keep adding an underscore to the name of # the file until the filename doesn't exist. while os.path.exists(os.path.join(settings.MEDIA_ROOT, filename)): try: dot_index = filename.rindex('.') except ValueError: # filename has no dot filename += '_' else: filename = filename[:dot_index] + '_' + filename[dot_index:] # Write the file to disk. setattr(self, field.attname, filename) fp = open(getattr(self, 'get_%s_filename' % field.name)(), 'wb') fp.write(raw_contents) fp.close() # Save the width and/or height, if applicable. if isinstance(field, ImageField) and (field.width_field or field.height_field): from django.utils.images import get_image_dimensions width, height = get_image_dimensions(getattr(self, 'get_%s_filename' % field.name)()) if field.width_field: setattr(self, field.width_field, width) if field.height_field: setattr(self, field.height_field, height) # Save the object, because it has changed. self.save() # IMAGE FIELD METHODS ###################### def method_get_image_width(field, self): return _get_image_dimensions(field, self)[0] def method_get_image_height(field, self): return _get_image_dimensions(field, self)[1] def _get_image_dimensions(field, self): cachename = "__%s_dimensions_cache" % field.name if not hasattr(self, cachename): from django.utils.images import get_image_dimensions fname = getattr(self, "get_%s_filename" % field.name)() setattr(self, cachename, get_image_dimensions(fname)) return getattr(self, cachename) ############################################## # HELPER FUNCTIONS (CURRIED MODEL FUNCTIONS) # ############################################## def get_absolute_url(opts, func, self): return settings.ABSOLUTE_URL_OVERRIDES.get('%s.%s' % (opts.app_label, opts.module_name), func)(self) def _get_where_clause(lookup_type, table_prefix, field_name, value): if table_prefix.endswith('.'): table_prefix = db.db.quote_name(table_prefix[:-1])+'.' field_name = db.db.quote_name(field_name) try: return '%s%s %s' % (table_prefix, field_name, (db.OPERATOR_MAPPING[lookup_type] % '%s')) except KeyError: pass if lookup_type == 'in': return '%s%s IN (%s)' % (table_prefix, field_name, ','.join(['%s' for v in value])) elif lookup_type in ('range', 'year'): return '%s%s BETWEEN %%s AND %%s' % (table_prefix, field_name) elif lookup_type in ('month', 'day'): return "%s = %%s" % db.get_date_extract_sql(lookup_type, table_prefix + field_name) elif lookup_type == 'isnull': return "%s%s IS %sNULL" % (table_prefix, field_name, (not value and 'NOT ' or '')) raise TypeError, "Got invalid lookup_type: %s" % repr(lookup_type) def function_get_object(opts, klass, does_not_exist_exception, **kwargs): obj_list = function_get_list(opts, klass, **kwargs) if len(obj_list) < 1: raise does_not_exist_exception, "%s does not exist for %s" % (opts.object_name, kwargs) assert len(obj_list) == 1, "get_object() returned more than one %s -- it returned %s! Lookup parameters were %s" % (opts.object_name, len(obj_list), kwargs) return obj_list[0] def _get_cached_row(opts, row, index_start): "Helper function that recursively returns an object with cache filled" index_end = index_start + len(opts.fields) obj = opts.get_model_module().Klass(*row[index_start:index_end]) for f in opts.fields: if f.rel and not f.null: rel_obj, index_end = _get_cached_row(f.rel.to, row, index_end) setattr(obj, f.get_cache_name(), rel_obj) return obj, index_end def function_get_iterator(opts, klass, **kwargs): # kwargs['select'] is a dictionary, and dictionaries' key order is # undefined, so we convert it to a list of tuples internally. kwargs['select'] = kwargs.get('select', {}).items() cursor = db.db.cursor() select, sql, params = function_get_sql_clause(opts, **kwargs) cursor.execute("SELECT " + (kwargs.get('distinct') and "DISTINCT " or "") + ",".join(select) + sql, params) fill_cache = kwargs.get('select_related') index_end = len(opts.fields) while 1: rows = cursor.fetchmany(GET_ITERATOR_CHUNK_SIZE) if not rows: raise StopIteration for row in rows: if fill_cache: obj, index_end = _get_cached_row(opts, row, 0) else: obj = klass(*row[:index_end]) for i, k in enumerate(kwargs['select']): setattr(obj, k[0], row[index_end+i]) yield obj def function_get_list(opts, klass, **kwargs): return list(function_get_iterator(opts, klass, **kwargs)) def function_get_count(opts, **kwargs): kwargs['order_by'] = [] kwargs['offset'] = None kwargs['limit'] = None kwargs['select_related'] = False _, sql, params = function_get_sql_clause(opts, **kwargs) cursor = db.db.cursor() cursor.execute("SELECT COUNT(*)" + sql, params) return cursor.fetchone()[0] def function_get_values_iterator(opts, klass, **kwargs): # select_related and select aren't supported in get_values(). kwargs['select_related'] = False kwargs['select'] = {} # 'fields' is a list of field names to fetch. try: fields = [opts.get_field(f).column for f in kwargs.pop('fields')] except KeyError: # Default to all fields. fields = [f.column for f in opts.fields] cursor = db.db.cursor() _, sql, params = function_get_sql_clause(opts, **kwargs) select = ['%s.%s' % (db.db.quote_name(opts.db_table), db.db.quote_name(f)) for f in fields] cursor.execute("SELECT " + (kwargs.get('distinct') and "DISTINCT " or "") + ",".join(select) + sql, params) while 1: rows = cursor.fetchmany(GET_ITERATOR_CHUNK_SIZE) if not rows: raise StopIteration for row in rows: yield dict(zip(fields, row)) def function_get_values(opts, klass, **kwargs): return list(function_get_values_iterator(opts, klass, **kwargs)) def _fill_table_cache(opts, select, tables, where, old_prefix, cache_tables_seen): """ Helper function that recursively populates the select, tables and where (in place) for fill-cache queries. """ for f in opts.fields: if f.rel and not f.null: db_table = f.rel.to.db_table if db_table not in cache_tables_seen: tables.append(db.db.quote_name(db_table)) else: # The table was already seen, so give it a table alias. new_prefix = '%s%s' % (db_table, len(cache_tables_seen)) tables.append('%s %s' % (db.db.quote_name(db_table), db.db.quote_name(new_prefix))) db_table = new_prefix cache_tables_seen.append(db_table) where.append('%s.%s = %s.%s' % \ (db.db.quote_name(old_prefix), db.db.quote_name(f.column), db.db.quote_name(db_table), db.db.quote_name(f.rel.get_related_field().column))) select.extend(['%s.%s' % (db.db.quote_name(db_table), db.db.quote_name(f2.column)) for f2 in f.rel.to.fields]) _fill_table_cache(f.rel.to, select, tables, where, db_table, cache_tables_seen) def _throw_bad_kwarg_error(kwarg): # Helper function to remove redundancy. raise TypeError, "got unexpected keyword argument '%s'" % kwarg def _parse_lookup(kwarg_items, opts, table_count=0): # Helper function that handles converting API kwargs (e.g. # "name__exact": "tom") to SQL. # Note that there is a distinction between where and join_where. The latter # is specifically a list of where clauses to use for JOINs. This # distinction is necessary because of support for "_or". # table_count is used to ensure table aliases are unique. tables, join_where, where, params = [], [], [], [] for kwarg, kwarg_value in kwarg_items: if kwarg in ('order_by', 'limit', 'offset', 'select_related', 'distinct', 'select', 'tables', 'where', 'params'): continue if kwarg_value is None: continue if kwarg == '_or': for val in kwarg_value: tables2, join_where2, where2, params2, table_count = _parse_lookup(val, opts, table_count) tables.extend(tables2) join_where.extend(join_where2) where.append('(%s)' % ' OR '.join(where2)) params.extend(params2) continue lookup_list = kwarg.split(LOOKUP_SEPARATOR) # pk="value" is shorthand for (primary key)__exact="value" if lookup_list[-1] == 'pk': if opts.pk.rel: lookup_list = lookup_list[:-1] + [opts.pk.name, opts.pk.rel.field_name, 'exact'] else: lookup_list = lookup_list[:-1] + [opts.pk.name, 'exact'] if len(lookup_list) == 1: _throw_bad_kwarg_error(kwarg) lookup_type = lookup_list.pop() current_opts = opts # We'll be overwriting this, so keep a reference to the original opts. current_table_alias = current_opts.db_table param_required = False while lookup_list or param_required: table_count += 1 try: # "current" is a piece of the lookup list. For example, in # choices.get_list(poll__sites__id__exact=5), lookup_list is # ["polls", "sites", "id"], and the first current is "polls". try: current = lookup_list.pop(0) except IndexError: # If we're here, lookup_list is empty but param_required # is set to True, which means the kwarg was bad. # Example: choices.get_list(poll__exact='foo') _throw_bad_kwarg_error(kwarg) # Try many-to-many relationships first... for f in current_opts.many_to_many: if f.name == current: rel_table_alias = db.db.quote_name('t%s' % table_count) table_count += 1 tables.append('%s %s' % \ (db.db.quote_name(f.get_m2m_db_table(current_opts)), rel_table_alias)) join_where.append('%s.%s = %s.%s' % \ (db.db.quote_name(current_table_alias), db.db.quote_name(current_opts.pk.column), rel_table_alias, db.db.quote_name(current_opts.object_name.lower() + '_id'))) # Optimization: In the case of primary-key lookups, we # don't have to do an extra join. if lookup_list and lookup_list[0] == f.rel.to.pk.name and lookup_type == 'exact': where.append(_get_where_clause(lookup_type, rel_table_alias+'.', f.rel.to.object_name.lower()+'_id', kwarg_value)) params.extend(f.get_db_prep_lookup(lookup_type, kwarg_value)) lookup_list.pop() param_required = False else: new_table_alias = 't%s' % table_count tables.append('%s %s' % (db.db.quote_name(f.rel.to.db_table), db.db.quote_name(new_table_alias))) join_where.append('%s.%s = %s.%s' % \ (db.db.quote_name(rel_table_alias), db.db.quote_name(f.rel.to.object_name.lower() + '_id'), db.db.quote_name(new_table_alias), db.db.quote_name(f.rel.to.pk.column))) current_table_alias = new_table_alias param_required = True current_opts = f.rel.to raise StopIteration for f in current_opts.fields: # Try many-to-one relationships... if f.rel and f.name == current: # Optimization: In the case of primary-key lookups, we # don't have to do an extra join. if lookup_list and lookup_list[0] == f.rel.to.pk.name and lookup_type == 'exact': where.append(_get_where_clause(lookup_type, current_table_alias+'.', f.column, kwarg_value)) params.extend(f.get_db_prep_lookup(lookup_type, kwarg_value)) lookup_list.pop() param_required = False # 'isnull' lookups in many-to-one relationships are a special case, # because we don't want to do a join. We just want to find out # whether the foreign key field is NULL. elif lookup_type == 'isnull' and not lookup_list: where.append(_get_where_clause(lookup_type, current_table_alias+'.', f.column, kwarg_value)) params.extend(f.get_db_prep_lookup(lookup_type, kwarg_value)) else: new_table_alias = 't%s' % table_count tables.append('%s %s' % \ (db.db.quote_name(f.rel.to.db_table), db.db.quote_name(new_table_alias))) join_where.append('%s.%s = %s.%s' % \ (db.db.quote_name(current_table_alias), db.db.quote_name(f.column), db.db.quote_name(new_table_alias), db.db.quote_name(f.rel.to.pk.column))) current_table_alias = new_table_alias param_required = True current_opts = f.rel.to raise StopIteration # Try direct field-name lookups... if f.name == current: where.append(_get_where_clause(lookup_type, current_table_alias+'.', f.column, kwarg_value)) params.extend(f.get_db_prep_lookup(lookup_type, kwarg_value)) param_required = False raise StopIteration # If we haven't hit StopIteration at this point, "current" must be # an invalid lookup, so raise an exception. _throw_bad_kwarg_error(kwarg) except StopIteration: continue return tables, join_where, where, params, table_count def function_get_sql_clause(opts, **kwargs): select = ["%s.%s" % (db.db.quote_name(opts.db_table), db.db.quote_name(f.column)) for f in opts.fields] tables = [opts.db_table] + (kwargs.get('tables') and kwargs['tables'][:] or []) tables = [db.db.quote_name(t) for t in tables] where = kwargs.get('where') and kwargs['where'][:] or [] params = kwargs.get('params') and kwargs['params'][:] or [] # Convert the kwargs into SQL. tables2, join_where2, where2, params2, _ = _parse_lookup(kwargs.items(), opts) tables.extend(tables2) where.extend(join_where2 + where2) params.extend(params2) # Add any additional constraints from the "where_constraints" parameter. where.extend(opts.where_constraints) # Add additional tables and WHERE clauses based on select_related. if kwargs.get('select_related') is True: _fill_table_cache(opts, select, tables, where, opts.db_table, [opts.db_table]) # Add any additional SELECTs passed in via kwargs. def quote_only_if_word(word): if word.find(' ')>=0: return word else: return db.db.quote_name(word) if kwargs.get('select'): select.extend(['(%s) AS %s' % (quote_only_if_word(s[1]), db.db.quote_name(s[0])) for s in kwargs['select']]) # ORDER BY clause order_by = [] for f in handle_legacy_orderlist(kwargs.get('order_by', opts.ordering)): if f == '?': # Special case. order_by.append(db.get_random_function_sql()) else: if f.startswith('-'): col_name = f[1:] order = "DESC" else: col_name = f order = "ASC" if "." in col_name: table_prefix, col_name = col_name.split('.', 1) table_prefix = db.db.quote_name(table_prefix) + '.' else: # Use the database table as a column prefix if it wasn't given, # and if the requested column isn't a custom SELECT. if "." not in col_name and col_name not in [k[0] for k in kwargs.get('select', [])]: table_prefix = db.db.quote_name(opts.db_table) + '.' else: table_prefix = '' order_by.append('%s%s %s' % (table_prefix, db.db.quote_name(orderfield2column(col_name, opts)), order)) order_by = ", ".join(order_by) # LIMIT and OFFSET clauses if kwargs.get('limit') is not None: limit_sql = " %s " % db.get_limit_offset_sql(kwargs['limit'], kwargs.get('offset')) else: assert kwargs.get('offset') is None, "'offset' is not allowed without 'limit'" limit_sql = "" return select, " FROM " + ",".join(tables) + (where and " WHERE " + " AND ".join(where) or "") + (order_by and " ORDER BY " + order_by or "") + limit_sql, params def function_get_in_bulk(opts, klass, *args, **kwargs): id_list = args and args[0] or kwargs['id_list'] assert id_list != [], "get_in_bulk() cannot be passed an empty list." kwargs['where'] = ["%s.%s IN (%s)" % (db.db.quote_name(opts.db_table), db.db.quote_name(opts.pk.column), ",".join(['%s'] * len(id_list)))] kwargs['params'] = id_list obj_list = function_get_list(opts, klass, **kwargs) return dict([(getattr(o, opts.pk.attname), o) for o in obj_list]) def function_get_latest(opts, klass, does_not_exist_exception, **kwargs): kwargs['order_by'] = ('-' + opts.get_latest_by,) kwargs['limit'] = 1 return function_get_object(opts, klass, does_not_exist_exception, **kwargs) def function_get_date_list(opts, field, *args, **kwargs): from django.core.db.typecasts import typecast_timestamp kind = args and args[0] or kwargs['kind'] assert kind in ("month", "year", "day"), "'kind' must be one of 'year', 'month' or 'day'." order = 'ASC' if kwargs.has_key('_order'): order = kwargs['_order'] del kwargs['_order'] assert order in ('ASC', 'DESC'), "'order' must be either 'ASC' or 'DESC'" kwargs['order_by'] = [] # Clear this because it'll mess things up otherwise. if field.null: kwargs.setdefault('where', []).append('%s.%s IS NOT NULL' % \ (db.db.quote_name(opts.db_table), db.db.quote_name(field.column))) select, sql, params = function_get_sql_clause(opts, **kwargs) sql = 'SELECT %s %s GROUP BY 1 ORDER BY 1' % (db.get_date_trunc_sql(kind, '%s.%s' % (db.db.quote_name(opts.db_table), db.db.quote_name(field.column))), sql) cursor = db.db.cursor() cursor.execute(sql, params) # We have to manually run typecast_timestamp(str()) on the results, because # MySQL doesn't automatically cast the result of date functions as datetime # objects -- MySQL returns the values as strings, instead. return [typecast_timestamp(str(row[0])) for row in cursor.fetchall()] ################################### # 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" man = types.ClassType('%sManipulator%s' % (opts.object_name, add and 'Add' or 'Change'), (formfields.Manipulator,), {}) man.__module__ = MODEL_PREFIX + '.' + opts.module_name # Set this explicitly, as above. man.__init__ = curry(manipulator_init, opts, add, change) man.save = curry(manipulator_save, opts, klass, add, change) for field_name_list in opts.unique_together: setattr(man, 'isUnique%s' % '_'.join(field_name_list), curry(manipulator_validator_unique_together, field_name_list, opts)) for f in 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')) for k, v in extra_methods.items(): setattr(man, k, v) return man def manipulator_init(opts, add, change, self, obj_key=None): 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.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 f.editable and not (f.primary_key and change) and (not f.rel or not f.rel.edit_inline): self.fields.extend(f.get_manipulator_fields(opts, self, change)) # Add fields for related objects. for rel_opts, rel_field in opts.get_inline_related_objects(): if change: count = getattr(self.original_object, 'get_%s_count' % opts.get_rel_object_method_name(rel_opts, rel_field))() count += rel_field.rel.num_extra_on_change if rel_field.rel.min_num_in_admin: count = max(count, rel_field.rel.min_num_in_admin) if rel_field.rel.max_num_in_admin: count = min(count, rel_field.rel.max_num_in_admin) else: count = rel_field.rel.num_in_admin for f in rel_opts.fields + rel_opts.many_to_many: if f.editable and f != rel_field and (not f.primary_key or (f.primary_key and change)): for i in range(count): self.fields.extend(f.get_manipulator_fields(rel_opts, self, change, name_prefix='%s.%d.' % (rel_opts.object_name.lower(), i), rel=True)) # 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): from django.utils.datastructures import DotExpandedDict params = {} for f in opts.fields: # Fields with auto_now_add are another special case; they should keep # their original value in the change stage. if change and getattr(f, 'auto_now_add', False): params[f.attname] = getattr(self.original_object, f.attname) else: params[f.attname] = f.get_manipulator_new_data(new_data) 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 not f.rel.edit_inline: was_changed = getattr(new_object, 'set_%s' % f.name)(new_data.getlist(f.name)) if change and was_changed: self.fields_changed.append(f.verbose_name) # Save many-to-one objects. Example: Add the Choice objects for a Poll. for rel_opts, rel_field in opts.get_inline_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': ['']})] obj_list = DotExpandedDict(new_data.data)[rel_opts.object_name.lower()].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. if change: old_rel_obj = None if rel_new_data[rel_opts.pk.name][0]: try: old_rel_obj = getattr(self.original_object, 'get_%s' % opts.get_rel_object_method_name(rel_opts, rel_field))(**{'%s__exact' % rel_opts.pk.name: rel_new_data[rel_opts.pk.attname][0]}) except ObjectDoesNotExist: pass for f in rel_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 change and (isinstance(f, FileField) or not f.editable): if rel_new_data.get(rel_opts.pk.attname, False) and rel_new_data[rel_opts.pk.attname][0]: params[f.attname] = getattr(old_rel_obj, f.attname) else: params[f.attname] = f.get_default() elif f == rel_field: params[f.attname] = getattr(new_object, rel_field.rel.field_name) elif add and isinstance(f, AutoField): params[f.attname] = None else: params[f.attname] = f.get_manipulator_new_data(rel_new_data, rel=True) # Related links are a special case, because we have to # manually set the "content_type_id" and "object_id" fields. if opts.has_related_links and rel_opts.module_name == 'relatedlinks': contenttypes_mod = get_module('core', 'contenttypes') params['content_type_id'] = contenttypes_mod.get_object(package__label__exact=opts.app_label, python_module_name__exact=opts.module_name).id params['object_id'] = new_object.id # Create the related item. new_rel_obj = rel_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 rel_opts.fields: if isinstance(f, FileField) and rel_new_data.get(f.attname, 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 "%r"' % (rel_opts.verbose_name, new_rel_obj)) else: for f in rel_opts.fields: if not f.primary_key and f != rel_field and str(getattr(old_rel_obj, f.attname)) != str(getattr(new_rel_obj, f.attname)): self.fields_changed.append('%s for %s "%r"' % (f.verbose_name, rel_opts.verbose_name, new_rel_obj)) # Save many-to-many objects. for f in rel_opts.many_to_many: if 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, rel_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 rel_new_data.has_key(rel_opts.pk.attname) and rel_new_data[rel_opts.pk.attname][0]: new_rel_obj.delete() self.fields_deleted.append('%s "%r"' % (rel_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): from django.utils.text import get_text_list field_list = [opts.get_field(field_name) for field_name in field_name_list] if isinstance(field_list[0].rel, ManyToOne): kwargs = {'%s__%s__iexact' % (field_name_list[0], field_list[0].rel.field_name): field_data} else: kwargs = {'%s__iexact' % field_name_list[0]: field_data} for f in field_list[1:]: field_val = all_data.get(f.attname, None) if field_val is None: # This will be caught by another validator, assuming the field # doesn't have blank=True. return if isinstance(f.rel, ManyToOne): kwargs['%s__pk' % f.name] = field_val else: kwargs['%s__iexact' % f.name] = field_val mod = opts.get_model_module() try: old_obj = mod.get_object(**kwargs) except ObjectDoesNotExist: return if hasattr(self, 'original_object') and getattr(self.original_object, opts.pk.attname) == getattr(old_obj, opts.pk.attname): pass else: raise validators.ValidationError, "%s with this %s already exists for the given %s." % \ (capfirst(opts.verbose_name), field_list[0].verbose_name, get_text_list(field_name_list[1:], 'and')) def manipulator_validator_unique_for_date(from_field, date_field, opts, lookup_type, self, field_data, all_data): date_str = all_data.get(date_field.get_manipulator_field_names('')[0], None) mod = opts.get_model_module() date_val = formfields.DateField.html2python(date_str) if date_val is None: return # Date was invalid. This will be caught by another validator. lookup_kwargs = {'%s__year' % date_field.name: date_val.year} if isinstance(from_field.rel, ManyToOne): lookup_kwargs['%s__pk' % from_field.name] = field_data else: lookup_kwargs['%s__iexact' % from_field.name] = field_data if lookup_type in ('month', 'date'): lookup_kwargs['%s__month' % date_field.name] = date_val.month if lookup_type == 'date': lookup_kwargs['%s__day' % date_field.name] = date_val.day try: old_obj = mod.get_object(**lookup_kwargs) except ObjectDoesNotExist: return else: if hasattr(self, 'original_object') and getattr(self.original_object, opts.pk.attname) == getattr(old_obj, opts.pk.attname): pass else: format_string = (lookup_type == 'date') and '%B %d, %Y' or '%B %Y' raise validators.ValidationError, "Please enter a different %s. The one you entered is already being used for %s." % \ (from_field.verbose_name, date_val.strftime(format_string))