Fixed database initialization. (#43)

Description:
* Added function to populate tables on startup for sqlalchemy and ormar.
* Added tortoise startup parameter.
* Fixed tortoise config.
* Fixed k8s config files.

Signed-off-by: Pavel Kirilin <[email protected]>

fastapi_template/template/{{cookiecutter.project_name}}/deploy/kube/app.yml

@@ -28,24 +28,24 @@ spec:
           args:
             - -c
             - >-
+              {%- if cookiecutter.enable_migrations == "True" %}
               {%- if cookiecutter.orm in ['sqlalchemy', 'ormar'] %}
               alembic upgrade head &&
               {%- elif cookiecutter.orm == 'tortoise' %}
               aerich upgrade &&
               {%- endif %}
+              {%- endif %}
               python -m {{cookiecutter.project_name }}
           {%- endif %}
           env:
             - name: {{cookiecutter.project_name | upper }}_HOST
               value: "0.0.0.0"
             - name: {{cookiecutter.project_name | upper }}_WORKERS_COUNT
               value: "10"
-            {%- if cookiecutter.db_info.name != "none" %}
-            {%- if cookiecutter.db_info.name != "sqlite" %}
+            {%- if cookiecutter.db_info.name not in ["none", "sqlite"] %}
             - name: {{cookiecutter.project_name | upper }}_DB_HOST
               value: "{{cookiecutter.kube_name}}-db-service"
             {%- endif %}
-            {%- endif %}
             {%- if cookiecutter.enable_redis == 'True' %}
             - name: {{cookiecutter.project_name | upper }}_REDIS_HOST
               value: "{{cookiecutter.kube_name}}-redis-service"

fastapi_template/template/{{cookiecutter.project_name}}/deploy/kube/db.yml

@@ -53,6 +53,7 @@ spec:
     - port: {{cookiecutter.db_info.port}}
       targetPort: {{cookiecutter.db_info.port}}
 ---
+{%- if cookiecutter.enable_migrations == "True" %}
 apiVersion: batch/v1
 kind: Job
 metadata:
@@ -87,5 +88,5 @@ spec:
           command: ["./wait-for-it.sh", "-t", "60", "{{cookiecutter.kube_name}}-db-service:{{cookiecutter.db_info.port}}"]
       restartPolicy: Never
 
-
 ---
+{%- endif %}

fastapi_template/template/{{cookiecutter.project_name}}/{{cookiecutter.project_name}}/db_tortoise/config.py

@@ -9,7 +9,7 @@
     },
     "apps": {
         "models": {
-            "models": ["aerich.models"] + MODELS_MODULES,
+            "models": MODELS_MODULES {%- if cookiecutter.enable_migrations == "True" %} + ["aerich.models"] {%- endif %} ,
             "default_connection": "default",
         },
     },

fastapi_template/template/{{cookiecutter.project_name}}/{{cookiecutter.project_name}}/web/application.py

@@ -57,7 +57,14 @@ def get_app() -> FastAPI:
     {% endif %}
 
     {%- if cookiecutter.orm == 'tortoise' %}
-    register_tortoise(app, config=TORTOISE_CONFIG, add_exception_handlers=True)
+    register_tortoise(
+        app,
+        config=TORTOISE_CONFIG,
+        add_exception_handlers=True,
+        {%- if cookiecutter.enable_migrations == "False" %}
+        generate_schemas=True,
+        {%- endif %}
+    )
     {%- endif %}
 
     return app

fastapi_template/template/{{cookiecutter.project_name}}/{{cookiecutter.project_name}}/web/lifetime.py

@@ -10,6 +10,11 @@
 
 {%- if cookiecutter.orm == "ormar" %}
 from {{cookiecutter.project_name}}.db.config import database
+{%- if cookiecutter.db_info.name != "none" and cookiecutter.enable_migrations == "False" %}
+from sqlalchemy.engine import create_engine
+from {{cookiecutter.project_name}}.db.meta import meta
+from {{cookiecutter.project_name}}.db.models import load_all_models
+{%- endif %}
 {%- endif %}
 
 {%- if cookiecutter.orm == "sqlalchemy" %}
@@ -21,6 +26,11 @@
 )
 from sqlalchemy.orm import sessionmaker
 
+{%- if cookiecutter.db_info.name != "none" and cookiecutter.enable_migrations == "False" %}
+from {{cookiecutter.project_name}}.db.meta import meta
+from {{cookiecutter.project_name}}.db.models import load_all_models
+{%- endif %}
+
 
 def _setup_db(app: FastAPI) -> None:
     """
@@ -47,11 +57,34 @@ def _setup_db(app: FastAPI) -> None:
 
 {% if cookiecutter.enable_redis == "True" %}
 def _setup_redis(app: FastAPI) -> None:
+    """
+    Initialize redis connection.
+
+    :param app: current FastAPI app.
+    """
     app.state.redis_pool = aioredis.ConnectionPool.from_url(
         str(settings.redis_url),
     )
 {%- endif %}
 
+{%- if cookiecutter.db_info.name != "none" and cookiecutter.enable_migrations == "False" %}
+{%- if cookiecutter.orm in ["ormar", "sqlalchemy"] %}
+async def _create_tables() -> None:
+    """Populates tables in the database."""
+    load_all_models()
+    {%- if cookiecutter.orm == "ormar" %}
+    engine = create_engine(str(settings.db_url))
+    with engine.connect() as connection:
+        meta.create_all(connection)
+    engine.dispose()
+    {%- elif cookiecutter.orm == "sqlalchemy" %}
+    engine = create_async_engine(str(settings.db_url))
+    async with engine.begin() as connection:
+        await connection.run_sync(meta.create_all)
+    await engine.dispose()
+    {%- endif %}
+{%- endif %}
+{%- endif %}
 
 def startup(app: FastAPI) -> Callable[[], Awaitable[None]]:
     """
@@ -67,9 +100,14 @@ def startup(app: FastAPI) -> Callable[[], Awaitable[None]]:
     async def _startup() -> None:  # noqa: WPS430
         {%- if cookiecutter.orm == "sqlalchemy" %}
         _setup_db(app)
-        {% elif cookiecutter.orm == "ormar" %}
+        {%- elif cookiecutter.orm == "ormar" %}
         await database.connect()
         {%- endif %}
+        {%- if cookiecutter.db_info.name != "none" and cookiecutter.enable_migrations == "False" %}
+        {%- if cookiecutter.orm in ["ormar", "sqlalchemy"] %}
+        await _create_tables()
+        {%- endif %}
+        {%- endif %}
         {%- if cookiecutter.enable_redis == "True" %}
         _setup_redis(app)
         {%- endif %}