mirror of
https://github.com/dolphin-emu/www.git
synced 2026-01-31 01:05:23 +01:00
22 lines
772 B
Python
22 lines
772 B
Python
# Copyright (c) 2018 Dolphin Emulator Website Contributors
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
from django.db import models
|
|
|
|
class DefaultSelectOrPrefetchManager(models.Manager):
|
|
def __init__(self, *args, **kwargs):
|
|
self._select_related = kwargs.pop('select_related', None)
|
|
self._prefetch_related = kwargs.pop('prefetch_related', None)
|
|
|
|
super(DefaultSelectOrPrefetchManager, self).__init__(*args, **kwargs)
|
|
|
|
def get_queryset(self, *args, **kwargs):
|
|
qs = super(DefaultSelectOrPrefetchManager, self).get_queryset(*args, **kwargs)
|
|
|
|
if self._select_related:
|
|
qs = qs.select_related(*self._select_related)
|
|
if self._prefetch_related:
|
|
qs = qs.prefetch_related(*self._prefetch_related)
|
|
|
|
return qs
|