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

Cleanup backend finding

Browse Source
This commit is contained in:
Steve Pulec 2013-07-26 15:14:34 -04:00
parent 168b049b4d
commit 674a85ba0b
2 changed files with 20 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
moto/backends.py Normal file
View File

@ -0,0 +1,17 @@
from moto.dynamodb import dynamodb_backend
from moto.ec2 import ec2_backend
from moto.elb import elb_backend
from moto.s3 import s3_backend
from moto.ses import ses_backend
from moto.sqs import sqs_backend
from moto.sts import sts_backend
BACKENDS = {
'dynamodb': dynamodb_backend,
'ec2': ec2_backend,
'elb': elb_backend,
's3': s3_backend,
'ses': ses_backend,
'sqs': sqs_backend,
'sts': sts_backend,
}

18
moto/server.py
View File

@ -4,14 +4,7 @@ import argparse
from flask import Flask from flask import Flask
from werkzeug.routing import BaseConverter from werkzeug.routing import BaseConverter
from moto.dynamodb import dynamodb_backend # flake8: noqa from moto.backends import BACKENDS
from moto.ec2 import ec2_backend # flake8: noqa
from moto.elb import elb_backend # flake8: noqa
from moto.s3 import s3_backend # flake8: noqa
from moto.ses import ses_backend # flake8: noqa
from moto.sqs import sqs_backend # flake8: noqa
from moto.sts import sts_backend # flake8: noqa
from moto.core.utils import convert_flask_to_httpretty_response from moto.core.utils import convert_flask_to_httpretty_response
app = Flask(__name__) app = Flask(__name__)
@ -26,8 +19,7 @@ class RegexConverter(BaseConverter):
def configure_urls(service): def configure_urls(service):
module = sys.modules[__name__] backend = BACKENDS[service]
backend = getattr(module, "{}_backend".format(service))
from werkzeug.routing import Map from werkzeug.routing import Map
# Reset view functions to reset the app # Reset view functions to reset the app
app.view_functions = {} app.view_functions = {}
@ -38,11 +30,7 @@ def configure_urls(service):
def main(argv=sys.argv[1:]): def main(argv=sys.argv[1:]):
# Yes, I'm using those imports in the beginning of the file to create a available_services = BACKENDS.keys()
# dynamic list of available services to be shown in the help text when the
# user tries to interact with moto_server.
available_services = [
x.split('_')[0] for x in globals() if x.endswith('_backend')]
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument( parser.add_argument(