Fix missing empty lines when using asyncpg (#41)

* refactor: fix missing empty lines in asyncpg.go, sqlc.yaml, queries.py (#42)

* fragment

Author: rayakame

.changes/unreleased/Fixed-20250506-001607.yaml

@@ -0,0 +1,6 @@
+kind: Fixed
+body: Fixed missing empty lines when using `asyncpg` driver.
+time: 2025-05-06T00:16:07.0299503+02:00
+custom:
+    Author: rayakame
+    PR: "41"

internal/codegen/drivers/asyncpg.go

@@ -91,10 +91,10 @@ func AsyncpgBuildPyQueryFunc(query *core.Query, body *builders.IndentStringBuild
 				}
 			}
 			body.WriteLine("))")
-			body.WriteIndentedString(indentLevel+1, "return return_rows")
+			body.WriteIndentedLine(indentLevel+1, "return return_rows")
 		} else {
 			body.WriteIndentedLine(indentLevel+2, fmt.Sprintf("return_rows.append(%s(row[0]))", retType))
-			body.WriteIndentedString(indentLevel+1, "return return_rows")
+			body.WriteIndentedLine(indentLevel+1, "return return_rows")
 		}
 	}
 	return nil

sqlc.yaml

@@ -3,7 +3,7 @@ plugins:
   - name: python
     wasm:
       url: file://sqlc-gen-better-python.wasm
-      sha256: beb883283e1e6fbe0f807a5294c4418f7674aa3cb23c89b28807fa8cd78c58ee
+      sha256: e5f90e5f52b4847a35d9c60a2828134bcd2a89f798793899fed7a0738aefed1a
 sql:
   - schema: test/schema.sql
     queries: test/queries.sql
@@ -15,7 +15,7 @@ sql:
           package: test
           sql_driver: asyncpg
           model_type: msgspec
-          emit_classes: false
+          emit_classes: true
           omit_unused_models: true
           emit_init_file: true
 

test/queries.py

@@ -7,12 +7,7 @@
 __all__: typing.Sequence[str] = (
     "GetStudentAndScoreRow",
     "GetStudentAndScoresRow",
-    "delete_author",
-    "get_student_and_score",
-    "get_student_and_scores",
-    "list_authors",
-    "list_authors2",
-    "update_author",
+    "Queries",
 )
 
 import msgspec
@@ -74,37 +69,41 @@ class GetStudentAndScoresRow(msgspec.Struct):
 """
 
 
-async def delete_author(conn: asyncpg.Connection, *, id: int) -> None:
-    await conn.execute(DELETE_AUTHOR, id)
-
-
-async def get_student_and_score(conn: asyncpg.Connection, *, id: int) -> typing.Optional[GetStudentAndScoreRow]:
-    row = await conn.fetchrow(GET_STUDENT_AND_SCORE, id)
-    if row is None:
-        return None
-    return GetStudentAndScoreRow(student=models.Student(id=row[0], name=row[1], age=row[2]), test_score=models.TestScore(student_id=row[3], score=row[4], grade=row[5]))
-
-
-async def get_student_and_scores(conn: asyncpg.Connection) -> typing.Sequence[GetStudentAndScoresRow]:
-    rows = await conn.fetch(GET_STUDENT_AND_SCORES)
-    return_rows: typing.List[GetStudentAndScoresRow] = []
-    for row in rows:
-        return_rows.append(GetStudentAndScoresRow(student=models.Student(id=row[0], name=row[1], age=row[2]), test_score=models.TestScore(student_id=row[3], score=row[4], grade=row[5])))
-    return return_rows
-
-async def list_authors(conn: asyncpg.Connection, *, ids: typing.Sequence[int]) -> typing.Sequence[int]:
-    rows = await conn.fetch(LIST_AUTHORS, ids)
-    return_rows: typing.List[int] = []
-    for row in rows:
-        return_rows.append(int(row[0]))
-    return return_rows
-
-async def list_authors2(conn: asyncpg.Connection, *, id: int) -> typing.Sequence[int]:
-    rows = await conn.fetch(LIST_AUTHORS2, id)
-    return_rows: typing.List[int] = []
-    for row in rows:
-        return_rows.append(int(row[0]))
-    return return_rows
-
-async def update_author(conn: asyncpg.Connection, *, name: str, bio: str, id: int) -> None:
-    await conn.execute(UPDATE_AUTHOR, name, bio, id)
+class Queries:
+    __slots__ = ("_conn",)
+
+    def __init__(self, conn: asyncpg.Connection):
+        self._conn = conn
+
+    async def delete_author(self, *, id: int) -> None:
+        await self._conn.execute(DELETE_AUTHOR, id)
+
+    async def get_student_and_score(self, *, id: int) -> typing.Optional[GetStudentAndScoreRow]:
+        row = await self._conn.fetchrow(GET_STUDENT_AND_SCORE, id)
+        if row is None:
+            return None
+        return GetStudentAndScoreRow(student=models.Student(id=row[0], name=row[1], age=row[2]), test_score=models.TestScore(student_id=row[3], score=row[4], grade=row[5]))
+
+    async def get_student_and_scores(self) -> typing.Sequence[GetStudentAndScoresRow]:
+        rows = await self._conn.fetch(GET_STUDENT_AND_SCORES)
+        return_rows: typing.List[GetStudentAndScoresRow] = []
+        for row in rows:
+            return_rows.append(GetStudentAndScoresRow(student=models.Student(id=row[0], name=row[1], age=row[2]), test_score=models.TestScore(student_id=row[3], score=row[4], grade=row[5])))
+        return return_rows
+
+    async def list_authors(self, *, ids: typing.Sequence[int]) -> typing.Sequence[int]:
+        rows = await self._conn.fetch(LIST_AUTHORS, ids)
+        return_rows: typing.List[int] = []
+        for row in rows:
+            return_rows.append(int(row[0]))
+        return return_rows
+
+    async def list_authors2(self, *, id: int) -> typing.Sequence[int]:
+        rows = await self._conn.fetch(LIST_AUTHORS2, id)
+        return_rows: typing.List[int] = []
+        for row in rows:
+            return_rows.append(int(row[0]))
+        return return_rows
+
+    async def update_author(self, *, name: str, bio: str, id: int) -> None:
+        await self._conn.execute(UPDATE_AUTHOR, name, bio, id)