Continue on ynab import

postgres/accounts.sql.go

@@ -11,13 +11,18 @@ import (
 
 const createAccount = `-- name: CreateAccount :one
 INSERT INTO accounts
-(name)
-VALUES ($1)
+(name, budget_id)
+VALUES ($1, $2)
 RETURNING id, budget_id, name
 `
 
-func (q *Queries) CreateAccount(ctx context.Context, name string) (Account, error) {
-	row := q.db.QueryRow(ctx, createAccount, name)
+type CreateAccountParams struct {
+	Name     string
+	BudgetID uuid.UUID
+}
+
+func (q *Queries) CreateAccount(ctx context.Context, arg CreateAccountParams) (Account, error) {
+	row := q.db.QueryRow(ctx, createAccount, arg.Name, arg.BudgetID)
 	var i Account
 	err := row.Scan(&i.ID, &i.BudgetID, &i.Name)
 	return i, err

postgres/payees.sql.go

@@ -11,13 +11,18 @@ import (
 
 const createPayee = `-- name: CreatePayee :one
 INSERT INTO payees
-(name)
-VALUES ($1)
+(name, budget_id)
+VALUES ($1, $2)
 RETURNING id, budget_id, name
 `
 
-func (q *Queries) CreatePayee(ctx context.Context, name string) (Payee, error) {
-	row := q.db.QueryRow(ctx, createPayee, name)
+type CreatePayeeParams struct {
+	Name     string
+	BudgetID uuid.UUID
+}
+
+func (q *Queries) CreatePayee(ctx context.Context, arg CreatePayeeParams) (Payee, error) {
+	row := q.db.QueryRow(ctx, createPayee, arg.Name, arg.BudgetID)
 	var i Payee
 	err := row.Scan(&i.ID, &i.BudgetID, &i.Name)
 	return i, err

postgres/queries/accounts.sql

@@ -1,7 +1,7 @@
 -- name: CreateAccount :one
 INSERT INTO accounts
-(name)
-VALUES ($1)
+(name, budget_id)
+VALUES ($1, $2)
 RETURNING *;
 
 -- name: GetAccounts :many

postgres/queries/payees.sql

@@ -1,7 +1,7 @@
 -- name: CreatePayee :one
 INSERT INTO payees
-(name)
-VALUES ($1)
+(name, budget_id)
+VALUES ($1, $2)
 RETURNING *;
 
 -- name: GetPayees :many