Using pydantic dataclass

Author: harishkannarao

app/model/customer.py

@@ -1,7 +1,7 @@
-import dataclasses
+from pydantic.dataclasses import dataclass
 
 
-@dataclasses.dataclass(frozen=True)
+@dataclass(frozen=True)
 class Customer:
     first_name: str
     last_name: str

app/routers/customer.py

@@ -13,14 +13,10 @@
 router = APIRouter()
 
 
-def to_customer(row) -> Customer:
-    return Customer(first_name=row.first_name, last_name=row.last_name)
-
-
 @router.get("/customers/", tags=["users"])
 async def read_customers() -> list[Customer]:
     rows = await database.fetch_all(query=SELECT_ALL_CUSTOMERS)
-    customers: list[Customer] = list(map(to_customer, rows))
+    customers: list[Customer] = list(map(lambda row: Customer(**dict(row)), rows))
     return customers