重写metaclass,实现类下标的调用方法

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]
>>> '二'
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值