Python __slots__限制动态添加变量

枫铃3年前 (2021-09-30)Python221

Python是一种非常灵活的动态语言,有时感觉太灵活以至于不知道遵循什么样的规则去驾驭。不过Python已经是非常完备的语言,想实现什么样的功能都是有方法的,而且也很容易,比如限制一个类动态添加成员变量。

一般情况下,我们定义完一个类,如果不加任何限制,还可以动态地为该类的对象或该类添加成员变量。

class Employee:
    def __init__(self,name=''):
        self.name = name

if __name__ == "__main__":
    try:
        employee1 = Employee('Bob')
        #动态为一个对象添加成员变量
        employee1.tel = '11111111'
        print(employee1.name,employee1.tel)
        employee2 = Employee('Tom')
        #employee2对象没有tel成员变量
        print(employee2.name,employee2.tel)
    except AttributeError as e:
        print(e)
    #动态为一个类添加成员变量
    Employee.tel = []
    print(employee2.name,employee2.tel)

#Bob 11111111
#'Employee' object has no attribute 'tel'
#Tom []

看起来很方便,不过如果我们如果不想使用者破坏类的结构、随便添加成员变量,要怎么做呢?

答案是,可以使用__slots__对象。

在类中,__slots__是一个以元组形式被定义的,只有在元组里的属性,才可以被赋值,不在元组里的属性赋值时会报错。

class Employee:
    __slots__ = ('name')
    def __init__(self,name=''):
        self.name = name

if __name__ == "__main__":
    employee1 = Employee('Bob')
    #动态为一个对象添加成员变量,但不在__slots__定义的元组内,会报异常
    employee1.tel = '11111111'
    print(employee1.name,employee1.tel)

#AttributeError: 'Employee' object has no attribute 'tel'

__solts__不能被子类继续,如果想限制子类的成员变量,子类也要定义__slots__变量,同时会继承父类的__slots__

'''
遇到问题没人解答?小编创建了一个Python学习交流QQ群:778463939
寻找有志同道合的小伙伴,互帮互助,群里还有不错的视频学习教程和PDF电子书!
'''
class Employee:
    __slots__ = ('name')
    def __init__(self,name=''):
        self.name = name

class Manager1(Employee):
    pass
class Manager2(Employee):
    __slots__ = ('addr')

if __name__ == "__main__":
    manager1 = Manager1('Bill')
    #动态为一个对象添加成员变量
    manager1.tel = '22222222'
    print(manager1.name,manager1.tel)
    manager2 = Manager2()
    manager2.name = 'David'
    manager2.addr = 'BJ'
    print(manager2.name,manager2.addr)
    #动态为一个对象添加成员变量,不在__slots__里,会报异常
    manager2.tel = '33333333'
    print(manager2.tel)

#Bill 22222222
#David BJ
#AttributeError: 'Manager2' object has no attribute 'tel'

如果想进一步限制对成员变量的使用,可以重载__setattr__, __getattr__,__getattribute__函数,__setattr__函数可以在赋值前拦截;__getattr__在找不到属性时被调用;__getattribute__则在获取属性值时无条件被调用,同时__getattr__不再被调用。注意不要在__getattribute__里使用self.attr来访问变量,这会导致无限递归循环。

class Employee:
    __slots__ = ('name')
    def __init__(self,name=''):
        self.name = name

class Manager2(Employee):
    __slots__ = ('addr')
    def __setattr__(self,name,value):
        if name == 'tel':
            raise AttributeError('manager has no tel')
        object.__setattr__(self, name, value)
    def __getattr__(self,name):
        if name == 'tel':
            return 0
        object.__getattribute__(self, name)
if __name__ == "__main__":
    manager2 = Manager2('Jone')
    manager2.name = 'David'
    manager2.addr = 'BJ'
    try:
        manager2.tel = '11111111'
    except Exception as e:
        print(e)
    print(manager2.name,manager2.addr,manager2.tel)

#manager has no tel
#David BJ 0

相关文章

利用python同步windows和linux文件

写python脚本的初衷,每次在windows编辑完文件后,想同步到linux上去,只能够登录服务器,...

爬虫基本原理

爬虫基本原理 一、爬虫是什么? 百度百科和维基百科对网络爬虫的定义:简单来说爬虫就是抓取目标网站内容的工具,一般是根据定义的行...

Django 函数和方法的区别

函数和方法的区别 1、函数要手动传self,方法不用传 2、如果是一个函数,用类名去调用,如果是一个方法...

Django 知识补漏单例模式

单例模式:(说白了就是)创建一个类的实例。在 Python 中,我们可以用多种方法来实现单例模式&#x...

Django基础知识MTV

Django简介 Django是使用Python编写的一个开源Web框架。可以用它来快速搭建一个高性能的网站。 Django也是一个MVC框架。但是在Dj...

Python mysql 索引原理与慢查询优化

一 介绍 为何要有索引? 一般的应用系统,读写比例在10:1左右,而且插入操作和一般的更新操作很少出现性能问题,...

发表评论

访客

看不清,换一张

◎欢迎参与讨论,请在这里发表您的看法和观点。