Expect UUID instead of str for ids on queries now.

Author: andrew-chang-dewitt

db_wrapper/model/async_model.py

@@ -1,6 +1,7 @@
 """Asynchronous Model objects."""
 
 from typing import Any, Dict, List
+from uuid import UUID
 
 from db_wrapper.client import AsyncClient
 from .base import (
@@ -45,7 +46,7 @@ def __init__(self, client: AsyncClient, table: sql.Composable) -> None:
         super().__init__(table)
         self._client = client
 
-    async def one_by_id(self, id_value: str) -> T:
+    async def one_by_id(self, id_value: UUID) -> T:
         """Read a row by it's id."""
         result: List[T] = await self._client.execute_and_return(
             self._query_one_by_id(id_value))

db_wrapper/model/base.py

@@ -105,15 +105,15 @@ class ReadABC(Generic[T]):
     def __init__(self, table: sql.Composable) -> None:
         self._table = table
 
-    def _query_one_by_id(self, id_value: str) -> sql.Composed:
+    def _query_one_by_id(self, id_value: UUID) -> sql.Composed:
         """Build query to read a row by it's id."""
         query = sql.SQL(
             'SELECT * '
             'FROM {table} '
             'WHERE id = {id_value};'
         ).format(
             table=self._table,
-            id_value=sql.Literal(id_value)
+            id_value=sql.Literal(str(id_value))
         )
 
         return query

db_wrapper/model/sync_model.py

@@ -1,6 +1,7 @@
 """Synchronous Model objects."""
 
 from typing import Any, Dict, List
+from uuid import UUID
 
 from db_wrapper.client import SyncClient
 from .base import (
@@ -45,7 +46,7 @@ def __init__(self, client: SyncClient, table: sql.Composable) -> None:
         super().__init__(table)
         self._client = client
 
-    def one_by_id(self, id_value: str) -> T:
+    def one_by_id(self, id_value: UUID) -> T:
         """Read a row by it's id."""
         result: List[T] = self._client.execute_and_return(
             self._query_one_by_id(id_value))