Raising an exception on updating a 'constant' attribute in python

前端 未结 5 1437
逝去的感伤
逝去的感伤 2021-01-14 15:29

As python does not have concept of constants, would it be possible to raise an exception if an \'constant\' attribute is updated? How?

class MyClass():
            


        
相关标签:
5条回答
  • 2021-01-14 16:03

    If you really want to have constant that can't be changed then look at this: http://code.activestate.com/recipes/65207/

    0 讨论(0)
  • 2021-01-14 16:05

    You can use a metaclass to achieve this:

    class ImmutableConstants(type):
        def __init__(cls, name, bases, dct):
            type.__init__(cls, name, bases, dct)
    
            old_setattr = cls.__setattr__
            def __setattr__(self, key, value):
                cls.assert_attribute_mutable(key)
                old_setattr(self, key, value)
            cls.__setattr__ = __setattr__
    
        def __setattr__(self, key, value):
            self.assert_attribute_mutable(key)
            type.__setattr__(self, key, value)
    
        def assert_attribute_mutable(self, name):
            if name.isupper():
                raise AttributeError('Attribute %s is constant' % name)
    
    class Foo(object):
        __metaclass__ = ImmutableConstants
        CONST = 5
        class_var = 'foobar'
    
    Foo.class_var = 'new value'
    Foo.CONST = 42 # raises
    

    But are you sure this is a real issue? Are you really accidentally setting constants all over the place? You can find most of these pretty easily with a grep -r '\.[A-Z][A-Z0-9_]*\s*=' src/.

    0 讨论(0)
  • 2021-01-14 16:08

    Customizing __setattr__ in every class (e.g. as exemplified in my old recipe that @ainab's answer is pointing to, and other answers), only works to stop assignment to INSTANCE attributes and not to CLASS attributes. So, none of the existing answers would actually satisfy your requirement as stated.

    If what you asked for IS actually exactly what you want, you could resort to some mix of custom metaclasses and descriptors, such as:

    class const(object):
      def __init__(self, val): self.val = val
      def __get__(self, *_): return self.val
      def __set__(self, *_): raise TypeError("Can't reset const!")
    
    class mcl(type):
      def __init__(cls, *a, **k):
        mkl = cls.__class__
        class spec(mkl): pass
        for n, v in vars(cls).items():
          if isinstance(v, const):
            setattr(spec, n, v)
        spec.__name__ = mkl.__name__
        cls.__class__ = spec
    
    class with_const:
      __metaclass__ = mcl
    
    class foo(with_const):
      CLASS_CONSTANT = const('this is a constant')
    
    print foo().CLASS_CONSTANT
    print foo.CLASS_CONSTANT
    foo.CLASS_CONSTANT = 'Oops!'
    print foo.CLASS_CONSTANT
    

    This is pretty advanced stuff, so you might prefer the simpler __setattr__ approach suggested in other answers, despite it NOT meeting your requirements as stated (i.e., you might reasonably choose to weaken your requirements in order to gain simplicity;-). But the techniques here might still be interesting: the custom descriptor type const is another way (IMHO far nicer than overriding __setattr__ in each and every class that needs some constants AND making all attributes constants rather than picking and choosing...) to block assignment to an instance attribute; the rest of the code is about a custom metaclass creating unique per-class sub-metaclasses of itself, in order to exploit said custom descriptor to the fullest and achieving the exact functionality you specifically asked for.

    0 讨论(0)
  • 2021-01-14 16:19

    You could do something like this: (from http://www.siafoo.net/snippet/108)

    class Constants:
      # A constant variable
      foo = 1337
    
      def __setattr__(self, attr, value):
        if hasattr(self, attr):
          raise ValueError, 'Attribute %s already has a value and so cannot be written to' % attr
        self.__dict__[attr] = value
    

    Then use it like this:

    >>> const = Constants()
    >>> const.test1 = 42
    >>> const.test1
    42
    >>> const.test1 = 43
    Traceback (most recent call last):
      File "<stdin>", line 1, in <module>
      File "<stdin>", line 4, in __setattr__
    ValueError: Attribute test1 already has a value and so cannot be written to
    >>> const.test1
    42
    
    0 讨论(0)
  • 2021-01-14 16:22

    Start reading this:

    http://docs.python.org/reference/datamodel.html#customizing-attribute-access

    You basically write your own version of __setattr__ that throws exceptions for some attributes, but not others.

    0 讨论(0)
提交回复
热议问题