Merge pull request #140 from TTWShell/bugfix/tpl

Bugfix/tpl

Author: TTWShell

.circleci/config.yml

@@ -43,13 +43,12 @@ jobs:
           command: |
             pip install flake8 pytest pytest-cov pytest-env
             pip install --editable ".[hobbit,hobbit_core]"
-            pip install celery
       - run:
           name: use flake8 check self
           command: flake8 .
       - run:
           name: run hobbit cmd
-          command: hobbit --echo startproject -n demo -d ~/haha -f -p 1024 --no-celery
+          command: hobbit --echo startproject -n demo -d ~/haha -f -p 1024
       - run:
           name: tree flask project
           command: |
@@ -75,17 +74,16 @@ jobs:
           command: |
             pip install flake8 pytest pytest-cov pytest-env
             pip install --editable ".[hobbit,hobbit_core]"
-            pip install celery
       - run:
           name: use flake8 check self
           command: flake8 .
       - run:
           name: run hobbit cmd
           command: |
             mkdir ~/haha && cd ~/haha
-            hobbit --echo startproject -n demo -f -p 1024 --celery
+            hobbit --echo startproject -n demo -f -p 1024
             mkdir ~/hahaha && cd ~/hahaha
-            hobbit --echo startproject -n demo -t rivendell -p 1025 --celery
+            hobbit --echo startproject -n demo -t rivendell -p 1025
       - run:
           name: tree flask project
           command: |

Pipfile.lock

@@ -213,6 +213,7 @@
 from hobbit_core import HobbitManager  # noqa
 
 app = Flask('', root_path='.')
+app.config["SQLALCHEMY_DATABASE_URI"] = "sqlite:///:memory"
 db = SQLAlchemy()
 db.init_app(app)
 hobbit = HobbitManager()

docs/conf.py

@@ -36,10 +36,8 @@ def common_options(func):
 @click.option('-p', '--port', help='Port of web server.', required=True,
               type=click.IntRange(1024, 65535))
 @common_options
-@click.option('--celery/--no-celery', default=False,
-              help='Generate celery files or not.')
 @click.pass_context
-def new(ctx, name, port, dist, template, force, celery):
+def new(ctx, name, port, dist, template, force):
     """Create a new flask project, render from different template.
 
     Examples::
@@ -52,14 +50,12 @@ def new(ctx, name, port, dist, template, force, celery):
     """  # noqa
     dist = os.getcwd() if dist is None else os.path.abspath(dist)
     ctx.obj['FORCE'] = force
-    ctx.obj['CELERY'] = celery
     ctx.obj['JINJIA_CONTEXT'] = {
         'project_name': name,
         'port': port,
         'secret_key': ''.join(random.choice(
             string.ascii_letters) for i in range(38)),
         'version': pkg_resources.get_distribution("hobbit-core").version,
-        'celery': celery,
     }
 
     echo(f'Start init a hobbit project `{name}` to `{dist}`,'

hobbit/bootstrap.py

@@ -29,7 +29,6 @@ def chdir(dist):
 
 @click.pass_context
 def render_project(ctx, dist, tpl_path):
-    celery = ctx.obj.get('CELERY')  # gen cmd not have this arg
     context = ctx.obj['JINJIA_CONTEXT']
 
     jinjia_env = Environment(loader=FileSystemLoader(tpl_path))
@@ -48,9 +47,6 @@ def render_project(ctx, dist, tpl_path):
                 render_file(dist, fn[:-len(SUFFIX)], data)
                 continue
 
-            if not celery and origin_path.endswith('tasks'):
-                continue
-
             dir_name = Template(fn).render(context)
             render_project(os.path.join(dist, dir_name),
                            os.path.join(tpl_path, fn))

hobbit/handlers/bootstrap.py

@@ -7,8 +7,4 @@ SECRET_KEY = '{{ secret_key }}'
 SQLALCHEMY_DATABASE_URI = 'sqlite:///{}'.format(
     os.path.join(ROOT_PATH, '{{ project_name }}.db'))
 SQLALCHEMY_TRACK_MODIFICATIONS = False
-{% if celery %}
-CELERY_TIMEZONE = 'Asia/Shanghai'
-BROKER_URL = 'sqla+sqlite:///celerydb.sqlite'  # CELERY_BROKER_URL
-CELERY_RESULT_BACKEND = 'db+sqlite:///results.sqlite'
-{% endif %}
+

hobbit/static/bootstrap/rivendell/app/configs/default.py.jinja2

@@ -1,17 +1,11 @@
 from flask_sqlalchemy import SQLAlchemy
 from flask_migrate import Migrate
 from flask_marshmallow import Marshmallow
-{%- if celery %}
-from celery import Celery
-{%- endif %}
 
 from hobbit_core import HobbitManager
 
 db = SQLAlchemy()
 migrate = Migrate()
 ma = Marshmallow()
-{%- if celery %}
-celery = Celery()
-{%- endif %}
 hobbit = HobbitManager()
 

hobbit/static/bootstrap/rivendell/app/exts.py.jinja2

@@ -6,7 +6,7 @@ from flask.helpers import get_env
 
 from hobbit_core.err_handler import ErrHandler
 
-from app.exts import db, migrate, ma, hobbit{% if celery %}, celery{% endif %}
+from app.exts import db, migrate, ma, hobbit
 
 
 def register_extensions(app):
@@ -33,21 +33,6 @@ def register_cmds(app):
     pass
 
 
-{% if celery -%}
-def make_celery(app):
-    celery.conf.update(app.config)
-
-    class ContextTask(celery.Task):
-        def __call__(self, *args, **kwargs):
-            with app.app_context():
-                return self.run(*args, **kwargs)
-
-    celery.Task = ContextTask
-    from app import tasks  # NOQA
-    return celery
-
-
-{% endif -%}
 def create_app():
     app = Flask(__name__, instance_relative_config=True)
     app.config.from_object('app.configs.{}'.format(get_env()))
@@ -57,9 +42,6 @@ def create_app():
         register_blueprints(app)
     register_error_handler(app)
     register_cmds(app)
-    {%- if celery %}
-    make_celery(app)
-    {%- endif %}
 
     @app.before_request
     def log_request_info():

hobbit/static/bootstrap/rivendell/app/run.py.jinja2

@@ -1,4 +1 @@
-hobbit-core[hobbit,hobbit_core]=={{ version }}
-{% if celery -%}
-celery
-{% endif %}
+hobbit-core[hobbit,hobbit_core]=={{ version }}