Last active
September 29, 2015 16:34
-
-
Save Vectorized/2e7d24bce2eff0fac153 to your computer and use it in GitHub Desktop.
Python add/set attributes to list (or any built-in classes)
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
class L(list): | |
""" | |
A subclass of list that can accept additional attributes. | |
Should be able to be used just like a regular list. | |
The problem: | |
a = [1, 2, 4, 8] | |
a.x = "Hey!" # AttributeError: 'list' object has no attribute 'x' | |
The solution: | |
a = L(1, 2, 4, 8) | |
a.x = "Hey!" | |
print a # [1, 2, 4, 8] | |
print a.x # "Hey!" | |
print len(a) # 4 | |
You can also do these: | |
a = L( 1, 2, 4, 8 , x="Hey!" ) # [1, 2, 4, 8] | |
a = L( 1, 2, 4, 8 )( x="Hey!" ) # [1, 2, 4, 8] | |
a = L( [1, 2, 4, 8] , x="Hey!" ) # [1, 2, 4, 8] | |
a = L( {1, 2, 4, 8} , x="Hey!" ) # [1, 2, 4, 8] | |
a = L( [2 ** b for b in range(4)] , x="Hey!" ) # [1, 2, 4, 8] | |
a = L( (2 ** b for b in range(4)) , x="Hey!" ) # [1, 2, 4, 8] | |
a = L( 2 ** b for b in range(4) )( x="Hey!" ) # [1, 2, 4, 8] | |
a = L( 2 ) # [2] | |
""" | |
def __new__(self, *args, **kwargs): | |
return super(L, self).__new__(self, args, kwargs) | |
def __init__(self, *args, **kwargs): | |
if len(args) == 1 and hasattr(args[0], '__iter__'): | |
list.__init__(self, args[0]) | |
else: | |
list.__init__(self, args) | |
self.__dict__.update(kwargs) | |
def __call__(self, **kwargs): | |
self.__dict__.update(kwargs) | |
return self |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment