perfect test

aerich/cli.py

@@ -154,7 +154,7 @@ def history(ctx):
 )
 @click.pass_context
 async def init(
-        ctx: Context, tortoise_orm, location,
+    ctx: Context, tortoise_orm, location,
 ):
     config = ctx.obj["config"]
     name = ctx.obj["name"]

aerich/ddl/postgres/__init__.py

@@ -0,0 +1,8 @@
+from tortoise.backends.asyncpg.schema_generator import AsyncpgSchemaGenerator
+
+from aerich.ddl import BaseDDL
+
+
+class PostgresDDL(BaseDDL):
+    schema_generator_cls = AsyncpgSchemaGenerator
+    DIALECT = AsyncpgSchemaGenerator.DIALECT

aerich/ddl/sqlite/__init__.py

@@ -0,0 +1,8 @@
+from tortoise.backends.sqlite.schema_generator import SqliteSchemaGenerator
+
+from aerich.ddl import BaseDDL
+
+
+class SqliteDDL(BaseDDL):
+    schema_generator_cls = SqliteSchemaGenerator
+    DIALECT = SqliteSchemaGenerator.DIALECT

aerich/migrate.py

@@ -13,11 +13,15 @@ from tortoise import (
     Model,
     Tortoise,
 )
+from tortoise.backends.asyncpg.schema_generator import AsyncpgSchemaGenerator
 from tortoise.backends.mysql.schema_generator import MySQLSchemaGenerator
+from tortoise.backends.sqlite.schema_generator import SqliteSchemaGenerator
 from tortoise.fields import Field
 
 from aerich.ddl import BaseDDL
 from aerich.ddl.mysql import MysqlDDL
+from aerich.ddl.postgres import PostgresDDL
+from aerich.ddl.sqlite import SqliteDDL
 from aerich.exceptions import ConfigurationError
 from aerich.utils import get_app_connection
 
@@ -76,6 +80,10 @@ class Migrate:
         connection = get_app_connection(config, app)
         if connection.schema_generator is MySQLSchemaGenerator:
             cls.ddl = MysqlDDL(connection)
+        elif connection.schema_generator is SqliteSchemaGenerator:
+            cls.ddl = SqliteDDL(connection)
+        elif connection.schema_generator is AsyncpgSchemaGenerator:
+            cls.ddl = PostgresDDL(connection)
         else:
             raise NotImplementedError("Current only support MySQL")