forked from sqlc-dev/sqlc
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathcity.py
More file actions
73 lines (57 loc) · 1.72 KB
/
city.py
File metadata and controls
73 lines (57 loc) · 1.72 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
# Code generated by sqlc. DO NOT EDIT.
from typing import AsyncIterator, Optional
import sqlalchemy
import sqlalchemy.ext.asyncio
from ondeck import models
CREATE_CITY = """-- name: create_city \\:one
INSERT INTO city (
name,
slug
) VALUES (
:p1,
:p2
) RETURNING slug, name
"""
GET_CITY = """-- name: get_city \\:one
SELECT slug, name
FROM city
WHERE slug = :p1
"""
LIST_CITIES = """-- name: list_cities \\:many
SELECT slug, name
FROM city
ORDER BY name
"""
UPDATE_CITY_NAME = """-- name: update_city_name \\:exec
UPDATE city
SET name = :p2
WHERE slug = :p1
"""
class AsyncQuerier:
def __init__(self, conn: sqlalchemy.ext.asyncio.AsyncConnection):
self._conn = conn
async def create_city(self, *, name: str, slug: str) -> Optional[models.City]:
row = (await self._conn.execute(sqlalchemy.text(CREATE_CITY), {"p1": name, "p2": slug})).first()
if row is None:
return None
return models.City(
slug=row[0],
name=row[1],
)
async def get_city(self, *, slug: str) -> Optional[models.City]:
row = (await self._conn.execute(sqlalchemy.text(GET_CITY), {"p1": slug})).first()
if row is None:
return None
return models.City(
slug=row[0],
name=row[1],
)
async def list_cities(self) -> AsyncIterator[models.City]:
result = await self._conn.stream(sqlalchemy.text(LIST_CITIES))
async for row in result:
yield models.City(
slug=row[0],
name=row[1],
)
async def update_city_name(self, *, slug: str, name: str) -> None:
await self._conn.execute(sqlalchemy.text(UPDATE_CITY_NAME), {"p1": slug, "p2": name})