77from psycopg2 .extensions import register_adapter
88from 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