Hopiu/django-modeltranslation
1
0
Fork 0
mirror of https://github.com/Hopiu/django-modeltranslation.git synced 2026-05-10 06:14:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Correctly patch all managers in Django 1.10+

Browse source 
Fixes #389 and #413.
This commit is contained in:
Thomas Jost 2017-09-26 16:22:22 +02:00
parent 019d116434
commit 22423047e9
No known key found for this signature in database
GPG key ID: 3F3B990246141762
1 changed files with 18 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
modeltranslation/translator.py
View file

@ -214,9 +214,24 @@ def add_manager(model):
manager.__class__ = NewMultilingualManager
managers = (model._meta.local_managers if NEW_MANAGER_API else
(getattr(model, x[1]) for x in
model._meta.concrete_managers + model._meta.abstract_managers))
if NEW_MANAGER_API:
# Inspired by django.db.models.options.Options.managers (find all
# managers by following the normal Python MRO rules), but keeps the
# original managers instead of making copies.
managers = []
seen = set()
bases = (b for b in model.mro() if hasattr(b, '_meta'))
for base in bases:
for manager in base._meta.local_managers:
if manager.name in seen:
continue
managers.append(manager)
seen.add(manager.name)
else:
managers = ((getattr(model, x[1]) for x in
model._meta.concrete_managers + model._meta.abstract_managers))
for current_manager in managers:
prev_class = current_manager.__class__
patch_manager_class(current_manager)