Skip to content

Commit 33db77a

Browse files
author
Roland Hedberg
committed
Added some extra functionality.
1 parent ba26a7b commit 33db77a

File tree

2 files changed

+17
-4
lines changed

2 files changed

+17
-4
lines changed

src/saml2/mdstore.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -130,15 +130,21 @@ def keys(self):
130130
def values(self):
131131
return self.entity.values()
132132

133+
def __len__(self):
134+
return len(self.entity)
135+
133136
def __contains__(self, item):
134-
return item in self.entity
137+
return item in self.entity.keys()
135138

136139
def __getitem__(self, item):
137140
return self.entity[item]
138141

139142
def __setitem__(self, key, value):
140143
self.entity[key] = value
141144

145+
def __delitem__(self, key):
146+
del self.entity[key]
147+
142148
def do_entity_descriptor(self, entity_descr):
143149
if self.check_validity:
144150
try:
@@ -913,7 +919,8 @@ def dumps(self, format="local"):
913919
res = EntitiesDescriptor()
914920
for _md in self.metadata.values():
915921
try:
916-
res.entity_descriptor.extend(_md.entities_descr.entity_descriptor)
922+
res.entity_descriptor.extend(
923+
_md.entities_descr.entity_descriptor)
917924
except AttributeError:
918925
res.entity_descriptor.append(_md.entity_descr)
919926

src/saml2/sigver.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -999,9 +999,15 @@ def security_context(conf, debug=None):
999999
return None
10001000

10011001
if debug is None:
1002-
debug = conf.debug
1002+
try:
1003+
debug = conf.debug
1004+
except AttributeError:
1005+
pass
10031006

1004-
metadata = conf.metadata
1007+
try:
1008+
metadata = conf.metadata
1009+
except AttributeError:
1010+
metadata = None
10051011

10061012
_only_md = conf.only_use_keys_in_metadata
10071013
if _only_md is None:

0 commit comments

Comments
 (0)