feat(db): add support for PostgreSQL and SQLite

Author: penggaolai

README.md

@@ -60,7 +60,7 @@
 
 + [ ] 提供英文文档版本
 
-+ [ ] 提供 PostgreSQL、SQLite 数据库支持
++ [x] 提供 PostgreSQL、SQLite 数据库支持
 
 + [x] 提供完整的 run 方法
 

fastapi_builder/constants.py

@@ -41,8 +41,9 @@ class License(BaseEnum):
 
 
 class Database(BaseEnum):
-    # POSTGRES = "Postgres"
+    POSTGRESQL = "PostgreSQL"
     MYSQL = "MySQL"
+    SQLITE = "SQLite"
 
 
 class DBCmd(BaseEnum):

fastapi_builder/templates/project/{{ cookiecutter.folder_name }}/pyproject.toml

@@ -12,14 +12,23 @@ readme = "README.md"
 [tool.poetry.dependencies]
 python = "^{{ cookiecutter.python }}"
 fastapi = "{{ cookiecutter.fastapi }}"
+{% if cookiecutter.database == 'MySQL' -%}
 aiomysql = "^0.2.0"
+pymysql = "^1.1.1"
+{% endif -%}
+{% if cookiecutter.database == 'PostgreSQL' -%}
+asyncpg = "^0.29.0"
+psycopg2-binary = "^2.9.9"
+{% endif -%}
+{% if cookiecutter.database == 'SQLite' -%}
+aiosqlite = "^0.19.0"
+{% endif -%}
 alembic = "^1.13.2"
 bcrypt = "^4.2.0"
 databases = "^0.9.0"
 email-validator = "^2.2.0"
 loguru = "^0.7.2"
 passlib = "^1.7.4"
-pymysql = "^1.1.1"
 python-jose = "^3.3.0"
 python-multipart = "^0.0.9"
 redis = "^5.0.8"

fastapi_builder/templates/project/{{ cookiecutter.folder_name }}/requirements.txt

@@ -1,4 +1,14 @@
+{% if cookiecutter.database == 'MySQL' -%}
 aiomysql==0.2.0
+pymysql==1.1.1
+{% endif -%}
+{% if cookiecutter.database == 'PostgreSQL' -%}
+asyncpg==0.29.0
+psycopg2-binary==2.9.9
+{% endif -%}
+{% if cookiecutter.database == 'SQLite' -%}
+aiosqlite==0.19.0
+{% endif -%}
 alembic==1.13.2
 bcrypt==4.2.0
 databases==0.9.0
@@ -7,7 +17,6 @@ fastapi==^{{ cookiecutter.fastapi }}
 loguru==0.7.2
 passlib==1.7.4
 pydantic==2.8.2
-pymysql==1.1.1
 python-jose==3.3.0
 python-multipart==0.0.9
 redis==5.0.8