diff --git a/fastapi_template/template/{{cookiecutter.project_name}}/{{cookiecutter.project_name}}/db_piccolo/piccolo_conf.py b/fastapi_template/template/{{cookiecutter.project_name}}/{{cookiecutter.project_name}}/db_piccolo/piccolo_conf.py new file mode 100644 index 0000000000000000000000000000000000000000..ebcc5862b2fd430483c5ca636ab21943ea901a38 --- /dev/null +++ b/fastapi_template/template/{{cookiecutter.project_name}}/{{cookiecutter.project_name}}/db_piccolo/piccolo_conf.py @@ -0,0 +1,22 @@ +from piccolo.conf.apps import AppRegistry + +from {{cookiecutter.project_name}}.settings import settings + +{%- if cookiecutter.db_info.name == "postgresql" %} +from piccolo.engine.postgres import PostgresEngine + +DB = PostgresEngine(config={'host': settings.db_host, + 'port': settings.db_port, + 'user': settings.db_user, + 'password': settings.db_pass, + 'database': settings.db_base}) + +{%- elif cookiecutter.db_info.name == "sqlite" %} +from piccolo.engine.sqlite import SQLiteEngine + +DB = SQLiteEngine(path=settings.db_file) +{%- endif %} + +# A list of paths to piccolo apps +# e.g. ['blog.piccolo_app'] +APP_REGISTRY = AppRegistry(apps=[])