Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Simplify how we store SWF types inside a SWF domain

Browse Source
This commit is contained in:
Jean-Baptiste Barth 2015-10-01 21:33:47 +02:00
parent 5c02fcd94b
commit 080b79338d
1 changed files with 7 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
moto/swf/models.py
View File

@ -22,8 +22,10 @@ class Domain(object):
self.retention = retention self.retention = retention
self.description = description self.description = description
self.status = "REGISTERED" self.status = "REGISTERED"
self.activity_types = defaultdict(dict) self.types = {
self.workflow_types = defaultdict(dict) "activity": defaultdict(dict),
"workflow": defaultdict(dict),
}
def __repr__(self): def __repr__(self):
return "Domain(name: %(name)s, status: %(status)s)" % self.__dict__ return "Domain(name: %(name)s, status: %(status)s)" % self.__dict__
@ -39,8 +41,7 @@ class Domain(object):
def get_type(self, kind, name, version, ignore_empty=False): def get_type(self, kind, name, version, ignore_empty=False):
try: try:
_types = getattr(self, "{}_types".format(kind)) return self.types[kind][name][version]
return _types[name][version]
except KeyError: except KeyError:
if not ignore_empty: if not ignore_empty:
raise SWFUnknownResourceFault( raise SWFUnknownResourceFault(
@ -51,17 +52,11 @@ class Domain(object):
) )
def add_type(self, _type): def add_type(self, _type):
if isinstance(_type, ActivityType): self.types[_type.kind][_type.name][_type.version] = _type
self.activity_types[_type.name][_type.version] = _type
elif isinstance(_type, WorkflowType):
self.workflow_types[_type.name][_type.version] = _type
else:
raise ValueError("Unknown SWF type: {}".format(_type))
def find_types(self, kind, status): def find_types(self, kind, status):
_all = [] _all = []
_types = getattr(self, "{}_types".format(kind)) for _, family in self.types[kind].iteritems():
for _, family in _types.iteritems():
for _, _type in family.iteritems(): for _, _type in family.iteritems():
if _type.status == status: if _type.status == status:
_all.append(_type) _all.append(_type)