1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
from pypy.interpreter.error import OperationError, oefmt
from pypy.interpreter.typedef import TypeDef
from pypy.interpreter.gateway import interp2app
from pypy.interpreter.baseobjspace import W_Root
class W_IdentityDict(W_Root):
def __init__(self, space):
self.dict = {}
def descr_new(space, w_subtype):
self = space.allocate_instance(W_IdentityDict, w_subtype)
W_IdentityDict.__init__(self, space)
return self
def descr_len(self, space):
return space.newint(len(self.dict))
def descr_contains(self, space, w_key):
return space.newbool(w_key in self.dict)
def descr_setitem(self, space, w_key, w_value):
self.dict[w_key] = w_value
def descr_getitem(self, space, w_key):
try:
return self.dict[w_key]
except KeyError:
raise OperationError(space.w_KeyError, w_key)
def descr_delitem(self, space, w_key):
try:
del self.dict[w_key]
except KeyError:
raise OperationError(space.w_KeyError, w_key)
def descr_iter(self, space):
raise oefmt(space.w_TypeError,
"'identity_dict' object does not support iteration; "
"iterate over x.keys()")
def get(self, space, w_key, w_default=None):
if w_default is None:
w_default = space.w_None
return self.dict.get(w_key, w_default)
def keys(self, space):
return space.newlist(self.dict.keys())
def values(self, space):
return space.newlist(self.dict.values())
def clear(self, space):
self.dict.clear()
W_IdentityDict.typedef = TypeDef("identity_dict",
__doc__="""\
A dictionary that considers keys by object identity.
Distinct objects will have separate entries even if they
compare equal. All objects can be used as keys, even
non-hashable ones --- but avoid using immutable objects
like integers: two int objects 42 may or may not be
internally the same object.
""",
__new__ = interp2app(W_IdentityDict.descr_new.im_func),
__len__ = interp2app(W_IdentityDict.descr_len),
__contains__ = interp2app(W_IdentityDict.descr_contains),
__setitem__ = interp2app(W_IdentityDict.descr_setitem),
__getitem__ = interp2app(W_IdentityDict.descr_getitem),
__delitem__ = interp2app(W_IdentityDict.descr_delitem),
__iter__ = interp2app(W_IdentityDict.descr_iter),
get = interp2app(W_IdentityDict.get),
keys = interp2app(W_IdentityDict.keys),
values = interp2app(W_IdentityDict.values),
clear = interp2app(W_IdentityDict.clear),
)
|
