Ticket #2135: urlify.diff

django/db/models/base.py

@@ -2 +2 @@
 import django.db.models.manager
 from django.core import validators
 from django.core.exceptions import ObjectDoesNotExist
-from django.db.models.fields import AutoField, ImageField, FieldDoesNotExist
+from django.db.models.fields import AutoField, ImageField, FieldDoesNotExist, URLify
 from django.db.models.fields.related import OneToOneRel, ManyToOneRel
 from django.db.models.related import RelatedObject
 from django.db.models.query import orderlist2sql, delete_objects
@@ -144 +144 @@
         dispatcher.send(signal=signals.class_prepared, sender=cls)
 
     _prepare = classmethod(_prepare)
+    
+    def prepopulate(self):
+        for f in self._meta.fields:
+            if not f.prepopulate_from: continue
+            if getattr(self, f.attname): continue
+            value = [getattr(self, p) for p in f.prepopulate_from]
+            value = URLify(' '.join(value), f.maxlength)
+            setattr(self, f.attname, value)
 
     def save(self):
         dispatcher.send(signal=signals.pre_save, sender=self.__class__, instance=self)

django/db/models/fields/__init__.py

@@ -7 +7 @@
 from django.utils.functional import curry, lazy
 from django.utils.text import capfirst
 from django.utils.translation import gettext, gettext_lazy, ngettext
-import datetime, os, time
+import datetime, os, time, re
 
 class NOT_PROVIDED:
     pass
@@ -702 +702 @@
 
     def get_manipulator_field_objs(self):
         return [forms.TextField]
+    
+def URLify(s, num_chars):
+    '''
+    Changes, e.g., "Petty theft" to "petty_theft".
+    This function is the Python equivalent of the javascript function
+    of the same name in django/contrib/admin/media/js/urlify.js.
+    It can get invoked for any field that has a prepopulate_from
+    attribute defined, although it only really makes sense for
+    SlugFields.
+    '''
+    # remove all these words from the string before urlifying
+    removelist = ["a", "an", "as", "at", "before", "but", "by", "for",
+                  "from", "is", "in", "into", "like", "of", "off", "on",
+                  "onto", "per", "since", "than", "the", "this", "that",
+                  "to", "up", "via", "with"]
+    ignore_words_pat = re.compile('|'.join([r'\b%s\b' % r for r in removelist]), re.I)
+    ignore_chars_pat = re.compile(r'[^-A-Z0-9\s]', re.I)
+    outside_space_pat = re.compile(r'^\s+|\s+$')
+    inside_space_pat = re.compile(r'[-\s]+')
+    
+    s = ignore_words_pat.sub('', s)  # remove unimportant words
+    s = ignore_chars_pat.sub('', s)  # remove unneeded chars
+    s = outside_space_pat.sub('', s) # trim leading/trailing spaces
+    s = inside_space_pat.sub('-', s) # convert spaces to hyphens
+    s = s.lower()                        # convert to lowercase
+    return s[:num_chars]                 # trim to first num_chars chars
 
 class SmallIntegerField(IntegerField):
     def get_manipulator_field_objs(self):