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. YUI Loader as Django middleware by akaihola 5 years, 2 months ago
  2. Resource by zvoase 4 years, 9 months ago
  3. Dojo Helper by limscoder 3 years, 1 month ago
  4. MODPYTHON logging by adroffner 5 years, 3 months ago
  5. filter dates to user profile's timezone by Scanner 6 years, 2 months ago

Comments

(Forgotten your password?)