Kaydet (Commit) 6b03179e authored tarafından Simon Charette's avatar Simon Charette Kaydeden (comit) Aymeric Augustin

Fixed #19688 -- Allow model subclassing with a custom metaclass using six.with_metaclass

üst e369dc28
......@@ -58,12 +58,21 @@ class ModelBase(type):
"""
def __new__(cls, name, bases, attrs):
super_new = super(ModelBase, cls).__new__
# six.with_metaclass() inserts an extra class called 'NewBase' in the
# inheritance tree: Model -> NewBase -> object. Ignore this class.
# inheritance tree: Model -> NewBase -> object. But the initialization
# should be executed only once for a given model class.
# attrs will never be empty for classes declared in the standard way
# (ie. with the `class` keyword). This is quite robust.
if name == 'NewBase' and attrs == {}:
return super_new(cls, name, bases, attrs)
# Also ensure initialization is only performed for subclasses of Model
# (excluding Model class itself).
parents = [b for b in bases if isinstance(b, ModelBase) and
not (b.__name__ == 'NewBase' and b.__mro__ == (b, object))]
if not parents:
# If this isn't a subclass of Model, don't do anything special.
return super_new(cls, name, bases, attrs)
# Create the class.
......
from django.db.models.base import ModelBase
class CustomBaseModel(ModelBase):
pass
from __future__ import unicode_literals
from django.db import models
from django.test.testcases import SimpleTestCase
from django.utils import six
from django.utils.unittest import skipIf
from .models import CustomBaseModel
class CustomBaseTest(SimpleTestCase):
@skipIf(six.PY3, 'test metaclass definition under Python 2')
def test_py2_custom_base(self):
"""
Make sure models.Model can be subclassed with a valid custom base
using __metaclass__
"""
try:
class MyModel(models.Model):
__metaclass__ = CustomBaseModel
except Exception:
self.fail("models.Model couldn't be subclassed with a valid "
"custom base using __metaclass__.")
def test_six_custom_base(self):
"""
Make sure models.Model can be subclassed with a valid custom base
using `six.with_metaclass`.
"""
try:
class MyModel(six.with_metaclass(CustomBaseModel, models.Model)):
pass
except Exception:
self.fail("models.Model couldn't be subclassed with a valid "
"custom base using `six.with_metaclass`.")
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment