Comment extraire un enregistrement aléatoire en utilisant l'ORM de Django ?

Question

J'ai un modèle qui représente les peintures que je présente sur mon site. Sur la page principale, j'aimerais en montrer quelques-unes : la plus récente, celle qui n'a pas été visitée depuis le plus longtemps, la plus populaire et une autre au hasard.

J'utilise Django 1.0.2.

Alors que les 3 premiers sont faciles à réaliser en utilisant les modèles de Django, le dernier (aléatoire) me pose quelques problèmes. Je peux bien sûr le coder dans ma vue, pour quelque chose comme ceci :

number_of_records = models.Painting.objects.count()
random_index = int(random.random()*number_of_records)+1
random_paint = models.Painting.get(pk = random_index)

Cela ne ressemble pas à quelque chose que j'aimerais avoir dans ma vue - cela fait entièrement partie de l'abstraction de la base de données et devrait être dans le modèle. De plus, ici je dois m'occuper des enregistrements supprimés (alors le nombre de tous les enregistrements ne me couvrira pas toutes les valeurs clés possibles) et probablement beaucoup d'autres choses.

Y a-t-il d'autres options pour le faire, de préférence dans l'abstraction du modèle ?

Answer 1

Il suffit d'utiliser :

MyModel.objects.order_by('?').first()

Il est documenté dans API QuerySet .

Answer 2

Utilisation de order_by('?') tuera le serveur de données le deuxième jour de la production. Une meilleure solution est quelque chose comme ce qui est décrit dans Obtenir une ligne aléatoire d'une base de données relationnelle .

from django.db.models.aggregates import Count
from random import randint

class PaintingManager(models.Manager):
    def random(self):
        count = self.aggregate(count=Count('id'))['count']
        random_index = randint(0, count - 1)
        return self.all()[random_index]

Answer 3

Les solutions avec order_by('?')[:N] sont extrêmement lentes même pour les tables de taille moyenne si vous utilisez MySQL (je ne sais pas pour les autres bases de données).

order_by('?')[:N] sera traduit en SELECT ... FROM ... WHERE ... ORDER BY RAND() LIMIT N requête.

Cela signifie que pour chaque ligne du tableau, la fonction RAND() sera exécutée, puis le tableau entier sera trié selon la valeur de cette fonction et les N premiers enregistrements seront retournés. Si vos tables sont petites, c'est parfait. Mais dans la plupart des cas, cette requête est très lente.

J'ai écrit une fonction simple qui fonctionne même si les identifiants ont des trous (certaines lignes ont été supprimées) :

def get_random_item(model, max_id=None):
    if max_id is None:
        max_id = model.objects.aggregate(Max('id')).values()[0]
    min_id = math.ceil(max_id*random.random())
    return model.objects.filter(id__gte=min_id)[0]

Elle est plus rapide que order_by('?') dans presque tous les cas.

Answer 4

Vous pourriez créer un manager sur votre modèle pour faire ce genre de choses. Pour comprendre d'abord ce qu'est un gestionnaire, le Painting.objects est un gestionnaire qui contient all() , filter() , get() etc. La création de votre propre gestionnaire vous permet de pré-filtrer les résultats et de faire travailler toutes ces mêmes méthodes, ainsi que vos propres méthodes personnalisées, sur les résultats.

EDIT : J'ai modifié mon code pour refléter la order_by['?'] méthode. Notez que le gestionnaire renvoie un nombre illimité de modèles aléatoires. Pour cette raison, j'ai inclus un peu de code d'utilisation pour montrer comment obtenir un seul modèle.

from django.db import models

class RandomManager(models.Manager):
    def get_query_set(self):
        return super(RandomManager, self).get_query_set().order_by('?')

class Painting(models.Model):
    title = models.CharField(max_length=100)
    author = models.CharField(max_length=50)

    objects = models.Manager() # The default manager.
    randoms = RandomManager() # The random-specific manager.

Utilisation

random_painting = Painting.randoms.all()[0]

Enfin, vous pouvez avoir de nombreux gestionnaires sur vos modèles, alors n'hésitez pas à créer des LeastViewsManager() o MostPopularManager() .

Answer 5

Vous pourriez simplement le faire :

models.Painting.objects.all().order_by('?')[:1].get()

ce qui est beaucoup plus efficace.