Skip to content

Commit bfa2adf

Browse files
Update example to specify Async API
1 parent 243b57a commit bfa2adf

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

example/example/models/extended_model.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,8 @@
77
from psycopg2.extensions import register_adapter
88
from psycopg2.extras import Json
99

10-
from db_wrapper.model import ModelData, Model, Read, Create, Client
10+
from db_wrapper import AsyncClient, AsyncModel, ModelData
11+
from db_wrapper.model import AsyncRead, AsyncCreate
1112

1213
# tell psycopg2 to adapt all dictionaries to json instead of
1314
# the default hstore
@@ -26,7 +27,7 @@ class ExtendedModelData(ModelData):
2627
data: Dict[str, Any]
2728

2829

29-
class ExtendedCreator(Create[ExtendedModelData]):
30+
class ExtendedCreator(AsyncCreate[ExtendedModelData]):
3031
"""Add custom json loading to Model.create."""
3132

3233
# pylint: disable=too-few-public-methods
@@ -60,7 +61,7 @@ async def one(self, item: ExtendedModelData) -> ExtendedModelData:
6061
return result[0]
6162

6263

63-
class ExtendedReader(Read[ExtendedModelData]):
64+
class ExtendedReader(AsyncRead[ExtendedModelData]):
6465
"""Add custom method to Model.read."""
6566

6667
async def all_by_string(self, string: str) -> List[ExtendedModelData]:
@@ -90,13 +91,13 @@ async def all(self) -> List[ExtendedModelData]:
9091
return result
9192

9293

93-
class ExtendedModel(Model[ExtendedModelData]):
94+
class ExtendedModel(AsyncModel[ExtendedModelData]):
9495
"""Build an ExampleItem Model instance."""
9596

9697
read: ExtendedReader
9798
create: ExtendedCreator
9899

99-
def __init__(self, client: Client) -> None:
100+
def __init__(self, client: AsyncClient) -> None:
100101
super().__init__(client, 'extended_model')
101102
self.read = ExtendedReader(self.client, self.table)
102103
self.create = ExtendedCreator(self.client, self.table)

0 commit comments

Comments
 (0)