Add NaturalSortField support.
This commit is contained in:
parent
113c93d27b
commit
d349b51923
1 changed files with 35 additions and 0 deletions
35
savepointradio/radio/fields.py
Normal file
35
savepointradio/radio/fields.py
Normal file
|
@ -0,0 +1,35 @@
|
||||||
|
import re
|
||||||
|
from unicodedata import normalize
|
||||||
|
|
||||||
|
from django.db import models
|
||||||
|
|
||||||
|
|
||||||
|
class NaturalSortField(models.CharField):
|
||||||
|
"""
|
||||||
|
A custom model field specifically for storing a normalized unicode string,
|
||||||
|
with removed starting articles, for use in natural sorting.
|
||||||
|
|
||||||
|
Code is a modified version of 'django-naturalsortfield' by Nathan Reynolds:
|
||||||
|
https://github.com/nathforge/django-naturalsortfield
|
||||||
|
"""
|
||||||
|
def __init__(self, for_field, **kwargs):
|
||||||
|
self.for_field = for_field
|
||||||
|
kwargs.setdefault('db_index', True)
|
||||||
|
kwargs.setdefault('editable', False)
|
||||||
|
kwargs.setdefault('max_length', 255)
|
||||||
|
super(NaturalSortField, self).__init__(**kwargs)
|
||||||
|
|
||||||
|
def pre_save(self, model_instance, add):
|
||||||
|
return self.naturalize(getattr(model_instance, self.for_field))
|
||||||
|
|
||||||
|
def naturalize(self, string):
|
||||||
|
def naturalize_int_match(match):
|
||||||
|
return '%08d' % (int(match.group(0)),)
|
||||||
|
|
||||||
|
string = normalize('NFKD', string).encode('ascii', 'ignore').decode('ascii')
|
||||||
|
string = string.lower()
|
||||||
|
string = string.strip()
|
||||||
|
string = re.sub(r'^(a|an|the)\s+', '', string)
|
||||||
|
string = re.sub(r'\d+', naturalize_int_match, string)
|
||||||
|
|
||||||
|
return string
|
Loading…
Reference in a new issue