sqlc-dev · kellen-miller · Aug 23, 2025
diff --git a/.gitignore b/.gitignore
@@ -5,3 +5,5 @@ bin
 .direnv
 .devenv*
 devenv.local.nix
+
+.idea/
diff --git a/examples/src/authors/models.py b/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
 

diff --git a/examples/src/authors/query.py b/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.executemany(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.executemany(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})
 

diff --git a/examples/src/authors/query.sql b/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);
diff --git a/examples/src/booktest/models.py b/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

diff --git a/examples/src/booktest/query.py b/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

diff --git a/examples/src/jets/models.py b/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
 
 

diff --git a/examples/src/jets/query-building.py b/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
 

diff --git a/examples/src/ondeck/city.py b/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
 

diff --git a/examples/src/ondeck/models.py b/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

diff --git a/examples/src/ondeck/venue.py b/examples/src/ondeck/venue.py
@@ -1,6 +1,6 @@
 # Code generated by sqlc. DO NOT EDIT.
 # versions:
-#   sqlc v1.28.0
+#   sqlc v1.29.0
 # source: venue.sql
 import dataclasses
 from typing import AsyncIterator, List, Optional

diff --git a/examples/src/tests/test_authors.py b/examples/src/tests/test_authors.py
@@ -29,6 +29,18 @@ def test_authors(db: sqlalchemy.engine.Connection):
     assert len(author_list) == 1
     assert author_list[0] == new_author
 
+    # Test batch insert with copyfrom
+    batch_authors = [
+        {"p1": "Dennis Ritchie", "p2": "Creator of C Programming Language"},
+        {"p1": "Ken Thompson", "p2": "Creator of Unix and Go Programming Language"},
+        {"p1": "Rob Pike", "p2": "Co-creator of Go Programming Language"},
+    ]
+    rows_affected = querier.create_authors_batch(batch_authors)
+    assert rows_affected == 3
+
+    all_authors = list(querier.list_authors())
+    assert len(all_authors) == 4  # 1 existing + 3 batch inserted
+
 
 @pytest.mark.asyncio
 async def test_authors_async(async_db: sqlalchemy.ext.asyncio.AsyncConnection):
@@ -54,3 +66,17 @@ async def test_authors_async(async_db: sqlalchemy.ext.asyncio.AsyncConnection):
         author_list.append(author)
     assert len(author_list) == 1
     assert author_list[0] == new_author
+
+    # Test batch insert with copyfrom
+    batch_authors = [
+        {"p1": "Dennis Ritchie", "p2": "Creator of C Programming Language"},
+        {"p1": "Ken Thompson", "p2": "Creator of Unix and Go Programming Language"},
+        {"p1": "Rob Pike", "p2": "Co-creator of Go Programming Language"},
+    ]
+    rows_affected = await querier.create_authors_batch(batch_authors)
+    assert rows_affected == 3
+
+    all_authors = []
+    async for author in querier.list_authors():
+        all_authors.append(author)
+    assert len(all_authors) == 4  # 1 existing + 3 batch inserted
diff --git a/internal/endtoend/testdata/copyfrom/python/models.py b/internal/endtoend/testdata/copyfrom/python/models.py
@@ -0,0 +1,24 @@
+# Code generated by sqlc. DO NOT EDIT.
+# versions:
+#   sqlc v1.29.0
+import dataclasses
+import datetime
+from typing import Optional
+
+
+@dataclasses.dataclass()
+class Author:
+    id: int
+    name: str
+    bio: str
+
+
+@dataclasses.dataclass()
+class User:
+    id: int
+    email: str
+    name: str
+    bio: Optional[str]
+    age: Optional[int]
+    active: Optional[bool]
+    created_at: datetime.datetime
diff --git a/internal/endtoend/testdata/copyfrom/python/query.py b/internal/endtoend/testdata/copyfrom/python/query.py
@@ -0,0 +1,158 @@
+# Code generated by sqlc. DO NOT EDIT.
+# versions:
+#   sqlc v1.29.0
+# source: query.sql
+import dataclasses
+from typing import Any, List, Optional
+
+import sqlalchemy
+import sqlalchemy.ext.asyncio
+
+from copyfrom import models
+
+
+CREATE_AUTHOR = """-- name: create_author \\:one
+INSERT INTO authors (name, bio) VALUES (:p1, :p2) RETURNING id, name, bio
+"""
+
+
+CREATE_AUTHORS = """-- name: create_authors \\:copyfrom
+INSERT INTO authors (name, bio) VALUES (:p1, :p2)
+"""
+
+
+CREATE_AUTHORS_NAMED = """-- name: create_authors_named \\:copyfrom
+INSERT INTO authors (name, bio) VALUES (:p1, :p2)
+"""
+
+
+CREATE_USER = """-- name: create_user \\:one
+INSERT INTO users (email, name) VALUES (:p1, :p2) RETURNING id, email, name, bio, age, active, created_at
+"""
+
+
+CREATE_USERS_BATCH = """-- name: create_users_batch \\:copyfrom
+INSERT INTO users (email, name) VALUES (:p1, :p2)
+"""
+
+
+CREATE_USERS_WITH_DETAILS = """-- name: create_users_with_details \\:copyfrom
+INSERT INTO users (email, name, bio, age, active) VALUES (:p1, :p2, :p3, :p4, :p5)
+"""
+
+
+@dataclasses.dataclass()
+class CreateUsersWithDetailsParams:
+    email: str
+    name: str
+    bio: Optional[str]
+    age: Optional[int]
+    active: Optional[bool]
+
+
+class Querier:
+    def __init__(self, conn: sqlalchemy.engine.Connection):
+        self._conn = conn
+
+    def create_author(self, *, name: str, bio: str) -> Optional[models.Author]:
+        row = self._conn.execute(sqlalchemy.text(CREATE_AUTHOR), {"p1": name, "p2": bio}).first()
+        if row is None:
+            return None
+        return models.Author(
+            id=row[0],
+            name=row[1],
+            bio=row[2],
+        )
+
+    def create_authors(self, arg_list: List[Any]) -> int:
+        result = self._conn.executemany(sqlalchemy.text(CREATE_AUTHORS), arg_list)
+        return result.rowcount
+
+    def create_authors_named(self, arg_list: List[Any]) -> int:
+        result = self._conn.executemany(sqlalchemy.text(CREATE_AUTHORS_NAMED), arg_list)
+        return result.rowcount
+
+    def create_user(self, *, email: str, name: str) -> Optional[models.User]:
+        row = self._conn.execute(sqlalchemy.text(CREATE_USER), {"p1": email, "p2": name}).first()
+        if row is None:
+            return None
+        return models.User(
+            id=row[0],
+            email=row[1],
+            name=row[2],
+            bio=row[3],
+            age=row[4],
+            active=row[5],
+            created_at=row[6],
+        )
+
+    def create_users_batch(self, arg_list: List[Any]) -> int:
+        result = self._conn.executemany(sqlalchemy.text(CREATE_USERS_BATCH), arg_list)
+        return result.rowcount
+
+    def create_users_with_details(self, arg_list: List[CreateUsersWithDetailsParams]) -> int:
+        data = list()
+        for item in arg_list:
+            data.append({
+                "p1": item.email,
+                "p2": item.name,
+                "p3": item.bio,
+                "p4": item.age,
+                "p5": item.active,
+            })
+        result = self._conn.executemany(sqlalchemy.text(CREATE_USERS_WITH_DETAILS), data)
+        return result.rowcount
+
+
+class AsyncQuerier:
+    def __init__(self, conn: sqlalchemy.ext.asyncio.AsyncConnection):
+        self._conn = conn
+
+    async def create_author(self, *, name: str, bio: str) -> Optional[models.Author]:
+        row = (await self._conn.execute(sqlalchemy.text(CREATE_AUTHOR), {"p1": name, "p2": bio})).first()
+        if row is None:
+            return None
+        return models.Author(
+            id=row[0],
+            name=row[1],
+            bio=row[2],
+        )
+
+    async def create_authors(self, arg_list: List[Any]) -> int:
+        result = await self._conn.executemany(sqlalchemy.text(CREATE_AUTHORS), arg_list)
+        return result.rowcount
+
+    async def create_authors_named(self, arg_list: List[Any]) -> int:
+        result = await self._conn.executemany(sqlalchemy.text(CREATE_AUTHORS_NAMED), arg_list)
+        return result.rowcount
+
+    async def create_user(self, *, email: str, name: str) -> Optional[models.User]:
+        row = (await self._conn.execute(sqlalchemy.text(CREATE_USER), {"p1": email, "p2": name})).first()
+        if row is None:
+            return None
+        return models.User(
+            id=row[0],
+            email=row[1],
+            name=row[2],
+            bio=row[3],
+            age=row[4],
+            active=row[5],
+            created_at=row[6],
+        )
+
+    async def create_users_batch(self, arg_list: List[Any]) -> int:
+        result = await self._conn.executemany(sqlalchemy.text(CREATE_USERS_BATCH), arg_list)
+        return result.rowcount
+
+    async def create_users_with_details(self, arg_list: List[CreateUsersWithDetailsParams]) -> int:
+        data = list()
+        for item in arg_list:
+            data.append({
+                "p1": item.email,
+                "p2": item.name,
+                "p3": item.bio,
+                "p4": item.age,
+                "p5": item.active,
+            })
+        result = await self._conn.executemany(sqlalchemy.text(CREATE_USERS_WITH_DETAILS), data)
+        return result.rowcount
diff --git a/internal/endtoend/testdata/copyfrom/query.sql b/internal/endtoend/testdata/copyfrom/query.sql
@@ -0,0 +1,17 @@
+-- name: CreateAuthors :copyfrom
+INSERT INTO authors (name, bio) VALUES ($1, $2);
+
+-- name: CreateAuthor :one
+INSERT INTO authors (name, bio) VALUES ($1, $2) RETURNING *;
+
+-- name: CreateAuthorsNamed :copyfrom
+INSERT INTO authors (name, bio) VALUES (@name, @bio);
+
+-- name: CreateUser :one
+INSERT INTO users (email, name) VALUES (@email, @name) RETURNING *;
+
+-- name: CreateUsersBatch :copyfrom
+INSERT INTO users (email, name) VALUES (@email, @name);
+
+-- name: CreateUsersWithDetails :copyfrom
+INSERT INTO users (email, name, bio, age, active) VALUES ($1, $2, $3, $4, $5);
diff --git a/internal/endtoend/testdata/copyfrom/schema.sql b/internal/endtoend/testdata/copyfrom/schema.sql
@@ -0,0 +1,15 @@
+CREATE TABLE authors (
+  id         SERIAL PRIMARY KEY,
+  name       text   NOT NULL,
+  bio        text   NOT NULL
+);
+
+CREATE TABLE users (
+  id         SERIAL PRIMARY KEY,
+  email      text   NOT NULL,
+  name       text   NOT NULL,
+  bio        text,
+  age        int,
+  active     boolean DEFAULT true,
+  created_at timestamp NOT NULL DEFAULT NOW()
+);
diff --git a/internal/endtoend/testdata/copyfrom/sqlc.yaml b/internal/endtoend/testdata/copyfrom/sqlc.yaml
@@ -0,0 +1,17 @@
+version: "2"
+plugins:
+  - name: py
+    wasm:
+      url: file://../../../../bin/sqlc-gen-python.wasm
+      sha256: "9aedc973afcc3c089934aaf509843a761e21ac92a4ce34d7a4ba3acebfb49bf0"
+sql:
+  - schema: "schema.sql"
+    queries: "query.sql"
+    engine: postgresql
+    codegen:
+      - out: python
+        plugin: py
+        options:
+          package: copyfrom
+          emit_sync_querier: true
+          emit_async_querier: true
-Original file line number
+Diff line change
@@ Expand Up / @@ -5,3 +5,5 @@ bin @@
     .direnv
     .devenv*
     devenv.local.nix
+    .idea/