diff --git a/src/quart_imp/_cli/filelib/extensions.py b/src/quart_imp/_cli/filelib/extensions.py index 5a71d3a..9b964c1 100644 --- a/src/quart_imp/_cli/filelib/extensions.py +++ b/src/quart_imp/_cli/filelib/extensions.py @@ -4,3 +4,12 @@ from quart_imp import Imp imp = Imp() """ + +def extensions_sql() -> str: + return """\ +from quart import Quart + + +def register_module(app:Quart): + pass +""" \ No newline at end of file diff --git a/src/quart_imp/_cli/filelib/init.py b/src/quart_imp/_cli/filelib/init.py index c6a6284..71fc5c8 100644 --- a/src/quart_imp/_cli/filelib/init.py +++ b/src/quart_imp/_cli/filelib/init.py @@ -1,8 +1,10 @@ def init_full_py(app_name: str, secret_key: str) -> str: return f"""\ from quart import Quart +from quart_imp.config import QuartConfig, ImpConfig from {app_name}.extensions import imp +from {app_name}.extensions.sql import db def create_app(): @@ -23,9 +25,6 @@ def create_app(): db.init_app(app) - with app.app_context(): - db.create_all() - return app """ diff --git a/src/quart_imp/_cli/init.py b/src/quart_imp/_cli/init.py index e70680f..24ff9e8 100644 --- a/src/quart_imp/_cli/init.py +++ b/src/quart_imp/_cli/init.py @@ -67,7 +67,7 @@ def minimal_app(app_folder: Path) -> None: def slim_app(app_folder: Path) -> None: from .filelib.init import init_slim_py - from .filelib.extensions import extensions_init + from .filelib.extensions import extensions_init, extensions_sql from .filelib.resources import resources_cli_py from .filelib.resources import resources_error_handlers_py from .filelib.templates import templates_error_html @@ -161,6 +161,10 @@ def full_app(app_folder: Path) -> None: folders["extensions"] / "__init__.py", extensions_init(), ), + "extensions/sql.py": ( + folders["extensions"] / "sql.py", + extensions_sql(), + ), "resources/cli/cli.py": ( folders["resources/cli"] / "cli.py", resources_cli_py(),