Skip to content

Commit 9ce2379

Browse files
committed
Late truncate_schema apply
1 parent 5baf072 commit 9ce2379

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/dipdup/utils/database.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,6 @@ async def generate_schema(conn: BaseDBAsyncClient, name: str) -> None:
137137
await create_schema(conn, name)
138138
await set_schema(conn, name)
139139
await Tortoise.generate_schemas()
140-
await conn.execute_script(_truncate_schema_sql)
141140
else:
142141
raise NotImplementedError
143142

@@ -146,6 +145,7 @@ async def truncate_schema(conn: BaseDBAsyncClient, name: str) -> None:
146145
if isinstance(conn, SqliteClient):
147146
raise NotImplementedError
148147

148+
await conn.execute_script(_truncate_schema_sql)
149149
await conn.execute_script(f"SELECT truncate_schema('{name}')")
150150

151151

0 commit comments

Comments
 (0)