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

Replace globals() call with a static mapping

Browse Source 
(suggested in @spulec review)
This commit is contained in:
Jean-Baptiste Barth 2015-11-19 11:44:51 +01:00
parent 6b581edb55
commit 26980f41a6
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
moto/swf/models/__init__.py
View File

@ -25,6 +25,12 @@ from .workflow_type import WorkflowType
from .workflow_execution import WorkflowExecution from .workflow_execution import WorkflowExecution
KNOWN_SWF_TYPES = {
"activity": ActivityType,
"workflow": WorkflowType,
}
class SWFBackend(BaseBackend): class SWFBackend(BaseBackend):
def __init__(self, region_name): def __init__(self, region_name):
self.region_name = region_name self.region_name = region_name
@ -119,7 +125,7 @@ class SWFBackend(BaseBackend):
_type = domain.get_type(kind, name, version, ignore_empty=True) _type = domain.get_type(kind, name, version, ignore_empty=True)
if _type: if _type:
raise SWFTypeAlreadyExistsFault(_type) raise SWFTypeAlreadyExistsFault(_type)
_class = globals()["{0}Type".format(kind.capitalize())] _class = KNOWN_SWF_TYPES[kind]
_type = _class(name, version, **kwargs) _type = _class(name, version, **kwargs)
domain.add_type(_type) domain.add_type(_type)