Add support for :copyfrom in Python code generation

Author: kellen-miller

.gitignore

@@ -5,3 +5,5 @@ bin
 .direnv
 .devenv*
 devenv.local.nix
+
+.idea/

examples/src/authors/models.py

@@ -1,6 +1,6 @@
 # Code generated by sqlc. DO NOT EDIT.
 # versions:
-#   sqlc v1.28.0
+#   sqlc v1.29.0
 import dataclasses
 from typing import Optional
 

examples/src/authors/query.py

@@ -1,8 +1,8 @@
 # Code generated by sqlc. DO NOT EDIT.
 # versions:
-#   sqlc v1.28.0
+#   sqlc v1.29.0
 # source: query.sql
-from typing import AsyncIterator, Iterator, Optional
+from typing import Any, AsyncIterator, Iterator, List, Optional
 
 import sqlalchemy
 import sqlalchemy.ext.asyncio
@@ -20,6 +20,11 @@
 """
 
 
+CREATE_AUTHORS_BATCH = """-- name: create_authors_batch \\:copyfrom
+INSERT INTO authors (name, bio) VALUES (:p1, :p2)
+"""
+
+
 DELETE_AUTHOR = """-- name: delete_author \\:exec
 DELETE FROM authors
 WHERE id = :p1
@@ -52,6 +57,10 @@ def create_author(self, *, name: str, bio: Optional[str]) -> Optional[models.Aut
             bio=row[2],
         )
 
+    def create_authors_batch(self, arg_list: List[Any]) -> int:
+        result = self._conn.execute(sqlalchemy.text(CREATE_AUTHORS_BATCH), arg_list)
+        return result.rowcount
+
     def delete_author(self, *, id: int) -> None:
         self._conn.execute(sqlalchemy.text(DELETE_AUTHOR), {"p1": id})
 
@@ -89,6 +98,10 @@ async def create_author(self, *, name: str, bio: Optional[str]) -> Optional[mode
             bio=row[2],
         )
 
+    async def create_authors_batch(self, arg_list: List[Any]) -> int:
+        result = await self._conn.execute(sqlalchemy.text(CREATE_AUTHORS_BATCH), arg_list)
+        return result.rowcount
+
     async def delete_author(self, *, id: int) -> None:
         await self._conn.execute(sqlalchemy.text(DELETE_AUTHOR), {"p1": id})
 

examples/src/authors/query.sql

@@ -17,3 +17,6 @@ RETURNING *;
 -- name: DeleteAuthor :exec
 DELETE FROM authors
 WHERE id = $1;
+
+-- name: CreateAuthorsBatch :copyfrom
+INSERT INTO authors (name, bio) VALUES ($1, $2);

examples/src/booktest/models.py

@@ -1,6 +1,6 @@
 # Code generated by sqlc. DO NOT EDIT.
 # versions:
-#   sqlc v1.28.0
+#   sqlc v1.29.0
 import dataclasses
 import datetime
 import enum

examples/src/booktest/query.py

@@ -1,6 +1,6 @@
 # Code generated by sqlc. DO NOT EDIT.
 # versions:
-#   sqlc v1.28.0
+#   sqlc v1.29.0
 # source: query.sql
 import dataclasses
 import datetime

examples/src/jets/models.py

@@ -1,6 +1,6 @@
 # Code generated by sqlc. DO NOT EDIT.
 # versions:
-#   sqlc v1.28.0
+#   sqlc v1.29.0
 import dataclasses
 
 

examples/src/jets/query-building.py

@@ -1,6 +1,6 @@
 # Code generated by sqlc. DO NOT EDIT.
 # versions:
-#   sqlc v1.28.0
+#   sqlc v1.29.0
 # source: query-building.sql
 from typing import AsyncIterator, Optional
 

examples/src/ondeck/city.py

@@ -1,6 +1,6 @@
 # Code generated by sqlc. DO NOT EDIT.
 # versions:
-#   sqlc v1.28.0
+#   sqlc v1.29.0
 # source: city.sql
 from typing import AsyncIterator, Optional
 

examples/src/ondeck/models.py

@@ -1,6 +1,6 @@
 # Code generated by sqlc. DO NOT EDIT.
 # versions:
-#   sqlc v1.28.0
+#   sqlc v1.29.0
 import dataclasses
 import datetime
 import enum