Retrieve similar objects from a model

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
from Levenshtein import ratio
from django.contrib.contenttypes.models import ContentType


def get_similar_items(mod, field, string, similarity_ratio=0.6, limit=None):
    """ 
        Get similar items for a specific model
        Example usage:
        similar = get_similar_items(Entry, 'title', 'Espresso is good for you', 0.5, 10)
        var = Entry.objects.filter(id__in=similar)
    """
    similar_list_id = []
    v = ContentType.objects.get_for_model(mod)
    c = v.model_class()
    for i in c.objects.all():
        if ratio(string, getattr(i,field)) > similarity_ratio:
            similar_list_id.append(i.id)
    return similar_list_id[:limit]

More like this

  1. truncate letters by trbs 4 years, 10 months ago
  2. Require Login Middleware by mattgrayson 3 years, 2 months ago
  3. Profiling middleware using cProfile by sgb 3 years, 9 months ago
  4. MetaOptions by haagenti 4 years, 7 months ago
  5. Add a CSS class to every field indicating what kind of field it is by ramen 1 year, 7 months ago

Comments

(Forgotten your password?)