Generalize decorator code.

2017-02-12 00:22:29 -05:00 · 2017-02-12 00:22:29 -05:00 · d3df810065
commit d3df810065
parent 9076e48fee
21 changed files with 53 additions and 160 deletions
--- a/moto/core/models.py
+++ b/moto/core/models.py
@ -186,3 +186,16 @@ class BaseBackend(object):
            return MockAWS({'global': self})(func)
        else:
            return MockAWS({'global': self})
+
+
+class base_decorator(object):
+    mock_backend = MockAWS
+
+    def __init__(self, backends):
+        self.backends = backends
+
+    def __call__(self, func=None):
+        if func:
+            return self.mock_backend(self.backends)(func)
+        else:
+            return self.mock_backend(self.backends)