Merge pull request #86 from legenove/tlp_for_tornado

Tlp for tornado

Author: jizhouli

README.md

@@ -1,10 +1,10 @@
-# Flask RESTful Application Code Generator
+# Flask/Tornado RESTful Application Code Generator
 
 [![Build Status][travis-image]][travis-url] [![PyPi Version][pypi-image]][pypi-url]
 
 ## Overview
 
-Generate Flask-RESTful application code from a Swagger Specification doc.
+Generate Flask/Tornado-RESTful application code from a Swagger Specification doc.
 
 **Alpha version for now, it can not handle all validation properly.**
 
@@ -25,13 +25,16 @@ swagger_py_codegen --swagger-doc api.yml example-app
 
 Command Options:
 
-    -s, --swagger, --swagger-doc        Swagger doc file.  [required]
-    -f, --force                         Force overwrite.
-    -p, --package                       Package name / application name.
-    -t, --template-dir                  Path of your custom templates directory.
-    --spec, --specification             Generate online specification json response.
-    --ui                                Generate swagger ui.
-    --help                              Show this message and exit.
+    -s, --swagger, --swagger-doc    Swagger doc file.  [required]
+	-f, --force                     Force overwrite.
+	-p, --package                   Package name / application name.
+	-t, --template-dir              Path of your custom templates directory.
+	--spec, --specification         Generate online specification json response.
+	--ui                            Generate swagger ui.
+	-j, --jobs INTEGER              Parallel jobs for processing.
+	-tlp, --templates               gen flask/tornado templates,default flask.
+	--version                       Show current version.
+	--help                          Show this message and exit.
 
 ## Examples:
 
@@ -42,7 +45,7 @@ Generate example-app from [apis.yml](https://github.com/guokr/swagger-py-codegen
 	|__ api.yml
 
     $ swagger_py_codegen -s  api.yml  example-app -p demo
-    $ tree
+    $ tree (flask-demo)
 	.
 	|__ api.yml
 	|__ example-app
@@ -62,6 +65,29 @@ Generate example-app from [apis.yml](https://github.com/guokr/swagger-py-codegen
 	   |     |__ validators.py
 	   |__ requirements.txt
 
+	$ swagger_py_codegen -s  docs/panel.yml  example-app -p demo -tlp=tornado
+    $ tree (tornado-demo)
+	.
+	|__ api.yml
+	|__ example-app
+	   |__ demo
+	   |  |__ __init__.py
+	   |  |__ core
+	   |     |__ __init.py
+	   |  |__ v1
+	   |     |__ api
+	   |     |  |__ __init__.py
+	   |     |  |__ oauth_auth_approach_approach.py
+	   |     |  |__ oauth_auth_approach.py
+	   |     |  |__ users_token.py
+	   |     |  |__ users_current.py
+	   |     |  |__ users.py
+	   |     |__ __init__.py
+	   |     |__ routes.py
+	   |     |__ schemas.py
+	   |     |__ validators.py
+	   |__ requirements.txt
+
 Install example-app requirements:
 
     $ cd example-app

swagger_py_codegen/command.py

@@ -13,6 +13,7 @@
 
 from ._version import __version__
 from .flask import FlaskGenerator
+from .tornado import TornadoGenerator
 from .parser import Swagger
 from .base import Template
 
@@ -81,16 +82,22 @@ def print_version(ctx, param, value):
               help='Generate swagger ui.')
 @click.option('-j', '--jobs',
               default=4, help='Parallel jobs for processing.')
+@click.option('-tlp', '--templates',
+              default='flask', help='gen flask/tornado templates,default flask.')
 @click.option('--version', is_flag=True, callback=print_version,
               expose_value=False, is_eager=True,
               help='Show current version.')
 def generate(destination, swagger_doc, force=False, package=None,
-             template_dir=None, specification=False, ui=False, jobs=4):
+             template_dir=None, templates='flask',
+             specification=False, ui=False, jobs=4):
     pool = Pool(processes=int(jobs))
     package = package or destination.replace('-', '_')
     data = spec_load(swagger_doc)
     swagger = Swagger(data, pool)
-    generator = FlaskGenerator(swagger)
+    if templates == 'tornado':
+        generator = TornadoGenerator(swagger)
+    else:
+        generator = FlaskGenerator(swagger)
     generator.with_spec = specification
     generator.with_ui = ui
     template = Template()

swagger_py_codegen/templates/tornado/api.tpl

@@ -0,0 +1,15 @@
+# -*- coding: utf-8 -*-
+from __future__ import absolute_import
+
+from core import RequestHandler
+from .. import UserInfo
+from ..validators import validate_filter
+
+class ApiHandler(RequestHandler):
+    on_initialize_decorators = [validate_filter]
+
+    def get_current_user(self):
+        authorization = self.request.headers.get('Authorization', '')
+        user_id = self.request.headers.get('user_id')
+
+        return UserInfo(user_id, authorization, self.blueprint)

swagger_py_codegen/templates/tornado/app.tpl

@@ -0,0 +1,63 @@
+# -*- coding: utf-8 -*-
+import os
+import tornado.httpserver
+import tornado.ioloop
+import tornado.web
+import tornado.options
+
+from core import load_tornado_settings
+
+modules = ['{{ blueprint }}']
+config = load_tornado_settings(*modules)
+
+class Application(tornado.web.Application):
+    def __init__(self, url_list, **app_settings):
+        tornado.web.Application.__init__(self, url_list, **app_settings)
+        self.config = config
+
+
+class RestfulErrorHandler(tornado.web.ErrorHandler):
+    def write_error(self, status_code, **kwargs):
+        """Override to implement custom error pages.
+
+        ``write_error`` may call `write`, `render`, `set_header`, etc
+        to produce output as usual.
+
+        If this error was caused by an uncaught exception (including
+        HTTPError), an ``exc_info`` triple will be available as
+        ``kwargs["exc_info"]``.  Note that this exception may not be
+        the "current" exception for purposes of methods like
+        ``sys.exc_info()`` or ``traceback.format_exc``.
+        """
+        self.finish({
+            "code": status_code,
+            "message": self._reason,
+        })
+
+
+def main():
+    import socket
+
+    socket.setdefaulttimeout(2)
+    url_list = []
+    url_list.extend(config.URIS)
+
+    app_settings = {
+        "static_path": os.path.join(os.path.dirname(__file__), "static"),
+        "default_handler_class": RestfulErrorHandler,
+        "default_handler_args": dict(status_code=404)
+
+    }
+
+    app = Application(url_list,
+                      debug=config.DEBUG,
+                      **app_settings)
+
+    http_server = tornado.httpserver.HTTPServer(app)
+    http_server.bind(config.PORT)
+    http_server.start(config.WORKER)
+    tornado.ioloop.IOLoop.current().start()
+
+
+if __name__ == "__main__":
+    main()

swagger_py_codegen/templates/tornado/blueprint.tpl

@@ -0,0 +1,38 @@
+# -*- coding: utf-8 -*-
+from __future__ import absolute_import
+
+class UserInfo(object):
+    _scopes = None
+    client_id = None
+    _account = None
+
+    def __init__(self, user_id, authorization, blueprint):
+        self.authorization = authorization
+        self.user_id = user_id
+        self.blueprint = blueprint
+        self.valid = False
+
+    @property
+    def scopes(self):
+        if self._scopes is None:
+            self._scopes = self._loader()
+        return self._scopes
+
+    def _loader(self):
+        return {{ scopes_supported }}
+
+    @property
+    def account(self):
+        if self._account is None:
+            if self.valid and self.user_id:
+                # TODO: test
+                self._account = None
+        return self._account
+
+
+def before_request(obj):
+    pass
+
+
+def after_request(obj):
+    pass

swagger_py_codegen/templates/tornado/core.tpl

@@ -0,0 +1,105 @@
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals, absolute_import
+
+import six
+import importlib
+import tornado.web
+
+settings = {}
+
+def load_tornado_settings(*modules):
+    settings.update({'MODULES': modules})
+    kwargs = {}
+    mods = []
+    config = Config()
+    config.update(**settings)
+    try:
+        setting_mod = importlib.import_module('my_settings')
+        if hasattr(setting_mod, 'load_settings'):
+            getattr(setting_mod, 'load_uris')(config, **kwargs)
+    except ImportError:
+        pass
+
+    for module in modules:
+        try:
+            mods.append(importlib.import_module('%s.routes' % module))
+        except ImportError:
+            raise ImportError(
+                "Could not import routers '%s' (Is it on sys.path?)" % (
+                    module))
+
+    for mod in mods:
+        if hasattr(mod, 'load_uris'):
+            getattr(mod, 'load_uris')(config, **kwargs)
+
+    return config
+
+class RequestHandler(tornado.web.RequestHandler):
+    on_initialize_decorators = []
+
+    def initialize(self):
+        request = self.request
+        meth = getattr(self, self.request.method.lower(), None)
+        if meth is None and self.request.method == 'HEAD':
+            meth = getattr(self, 'get', None)
+        assert meth is not None, 'Unimplemented method %r' % request.method
+
+        for decorator in self.on_initialize_decorators:
+            meth = decorator(meth)
+
+        setattr(self, self.request.method.lower(), meth)
+
+    def set_headers(self, items):
+        if items is None:
+            return
+        for k, v in items:
+            self.set_header(k, v)
+
+    def write_error(self, status_code, **kwargs):
+        """Override to implement custom error pages.
+
+        ``write_error`` may call `write`, `render`, `set_header`, etc
+        to produce output as usual.
+
+        If this error was caused by an uncaught exception (including
+        HTTPError), an ``exc_info`` triple will be available as
+        ``kwargs["exc_info"]``.  Note that this exception may not be
+        the "current" exception for purposes of methods like
+        ``sys.exc_info()`` or ``traceback.format_exc``.
+        """
+        self.finish({
+            "code": status_code,
+            "message": self._reason,
+        })
+
+class Config(object):
+    def __getitem__(self, item):
+        return getattr(self, item)
+
+    def update(self, **kw):
+        for name, value in kw.items():
+            self.__setattr__(name, value)
+
+    def setdefault(self, key, default=None):
+        try:
+            return getattr(self, key)
+        except AttributeError:
+            setattr(self, key, default)
+        return default
+
+    def uri_tuple(self, route, url_prefix):
+        route['resource'].endpoint = route['endpoint']
+        route['resource'].blueprint = url_prefix.replace('/', '_')
+        if url_prefix:
+            return r'/' + url_prefix + route['urls'][0], route['resource']
+        return route['urls'][0], route['resource']
+
+    def update_uri(self, routes, url_prefix=r''):
+        self.ROUTES.extend(routes)
+        self.URIS.extend([self.uri_tuple(r, url_prefix) for r in routes])
+
+    URIS = []
+    ROUTES = []
+    DEBUG = True
+    PORT = 5000
+    WORKER = 1

swagger_py_codegen/templates/tornado/requirements.tpl

@@ -0,0 +1,3 @@
+tornado
+jsonschema
+six

swagger_py_codegen/templates/tornado/routers.tpl

@@ -0,0 +1,22 @@
+# -*- coding: utf-8 -*-
+
+{% include '_do_not_change.tpl' %}
+from __future__ import absolute_import
+
+{% for view in views -%}
+from .api.{{ view.endpoint }} import {{ view.name }}
+{% endfor %}
+
+url_prefix = '{{ blueprint }}'
+
+routes = [
+    {%- for view in views %}
+    dict(resource={{ view.name }}, urls=[r"{{ view.url }}"], endpoint='{{ view.endpoint }}'),
+    {%- endfor %}
+]
+
+def load_uris(config):
+    try:
+        config.update_uri(routes, url_prefix)
+    except:
+        pass

swagger_py_codegen/templates/tornado/specification.tpl

@@ -0,0 +1 @@
+{{ swagger }}