From 6a5d9e5f8e720fc2b50dd363ac0f450593184895 Mon Sep 17 00:00:00 2001
From: Cameron <leaf33golf@gmail.com>
Date: Wed, 10 Dec 2025 21:49:13 +0000
Subject: [PATCH] Update app/schemas.py

---
 app/schemas.py | 31 +++++++++++++++++++++++++------
 1 file changed, 25 insertions(+), 6 deletions(-)

diff --git a/app/schemas.py b/app/schemas.py
index ab00851..010a678 100644
--- a/app/schemas.py
+++ b/app/schemas.py
@@ -1,26 +1,45 @@
 from pydantic import BaseModel
 from datetime import date
+from typing import Optional
 
+# -----------------------
+# Customer Schemas
+# -----------------------
 class CustomerBase(BaseModel):
     full_name: str
     email: str
-    phone: str
-    home_branch_id: int
-    customer_since: date
+    phone: Optional[str] = None
+    date_of_birth: Optional[date] = None
+    age: Optional[int] = None
+    gender: Optional[str] = None
+    street_address: Optional[str] = None
+    city: Optional[str] = None
+    state: Optional[str] = None
+    zip_code: Optional[str] = None
+    home_branch_id: Optional[int] = None
+    customer_since: Optional[date] = None
+    employment_status: Optional[str] = None
+    annual_income: Optional[float] = None
+    credit_score: Optional[int] = None
+    preferred_contact_method: Optional[str] = None
 
 class Customer(CustomerBase):
-    customer_id: int
+    customer_id: str  # now a string
     class Config:
         orm_mode = True
 
+# -----------------------
+# Account Schemas
+# -----------------------
 class AccountBase(BaseModel):
-    customer_id: int
+    customer_id: str  # now a string
     account_type: str
     open_date: date
     balance: float
+    branch_id: Optional[int] = None
 
 class Account(AccountBase):
-    account_id: int
+    account_id: str  # now a string
     account_number: str
     class Config:
         orm_mode = True