1. 对象下标的的调用方法
直接用内置方法__getitem__即可实现,如下
class User:
def __getitem__(self, item):
return item * 2
u = User()
u[2]
>>> 4
User[2]
>>> Traceback (most recent call last):
>>> File "<input>", line 1, in <module>
>>> TypeError: 'type' object is not subscriptable
2.类直接调用下标方法,抛错
如上,类直接调用的下标方法的话,会报错,如上;那么要怎么实现呢。首先尝试给__getitem__()增加@classmethod装饰器。如下,结果还是会抛错TypeError: ‘type’ object is not subscriptable
class User:
@classmethod
def __getitem__(self, item):
return item * 2
User()[2]
>>> 4
User[2]
>>> Traceback (most recent call last):
>>> File "<input>", line 1, in <module>
>>> TypeError: 'type' object is not subscriptable
3.抛错问题分析
我们的需求是类调用下标方法,抛错是type对象不可下标调用,我们知道类实际也是一个对象,是又元类创建出来的对象,对应我们的User类就是一个type对象(即User类是由元类type创建出来的)。
那么我们只要重写元类,给元类增加__getitem__()方法,元类的对象就是可以下标调用的了。
4.重写元类,实现类下标调用
4.1. 重写元类的__getitem__()实现
class Mytype(type):
def __getitem__(self, item):
return item * 2
class User(metaclass=Mytype):
def __getitem__(self, item):
return item * 2
User()[2]
>>> 4
User[2]
>>> 4
4.2. 重写元类的__getitem__()调用类方法实现
元类的方法里self对应的是User类,调用self.getitem(item),相当于调用了User.getitem(item),所以User的__getitem__方法应该加@classmethod装饰器定义为类方法
class Mytype(type):
def __getitem__(self, item):
return self.__getitem__(item) # self对应User类,__getitem__应该是类方法
class User(metaclass=Mytype):
def __getitem__(self, item): # 写成实例方法是不对的,调用时会抛错少参数
return item * 2
User()[2]
>>> 4
User[2]
>>> Traceback (most recent call last):
>>> File "<input>", line 1, in <module>
>>> File "<input>", line 3, in __getitem__
>>> TypeError: __getitem__() missing 1 required positional argument: 'item'
class Mytype(type):
def __getitem__(self, item):
return self.__getitem__(item)
class User(metaclass=Mytype):
@classmethod
def __getitem__(cls, item): # 写成类方法是正确的
return item * 2
User()[2]
>>> 4
User[2]
>>> 4
5.Django中的model类下标方法调用,用元类方法实现
model的元类是ModelBase,继承他重写__getitem__(),然后model指定metaclass即可实现,如下
from django.db.models.base import ModelBase
from django.db import models
class Mytype(ModelBase):
def __getitem__(self, item):
return self.__getitem__(item)
class BaseMixin:
objects: None
@classmethod
def __getitem__(cls, item):
return cls.get(item)
@classmethod
def get(cls, name):
if isinstance(name, int)
return cls.objects.get(id=name)
return cls.objects.get(name=name)
class User(metaclass=Mytype):
name = models.Model('人名', max_length=255)
alias = models.Model('别名', max_length=255)
class Meta:
verbose_name = '人物表'
db_table = 't_user'
def __str__(self):
return f'{self.pk}<{self.name}>'
User.objects.get(id=1)
>>> <User: 1<张三>>
User.get(1)
>>> <User: 1<张三>>
User[1]
>>> <User: 1<张三>>
User['张三']
>>> <User: 1<张三>>
u2=User.objects.get(id=2)
u2[1] # 实例调用下标方法,结果也是一样的。
>>> <User: 1<张三>>
最后,如果希望类下标方法和实例下标方法,分别返回不同的结果。修改元类调用的类方法,两者分开即可
class Mytype(ModelBase):
def __getitem__(self, item):
return self.__clsitem__(item)
class BaseMixin:
objects: None
@classmethod
def __clsitem__(cls, item): # 类方法,供元类调用
return cls.get(item)
def __getitem__(self, item): # 实例方法,返回别名的对应下标结果
return self.alias[item]
@classmethod
def get(cls, name):
if isinstance(name, int)
return cls.objects.get(id=name)
return cls.objects.get(name=name)
User[1]
>>> <User: 1<张三>>
u1=User.objects.get(id=1)
u1.alias
>>> '零一二三'
u1[0]
>>> '零'
u1[2]
>>> '二'