Update app/schemas/class_.py

app/schemas/class_.py

@@ -43,6 +43,12 @@ class ClassBase(BaseModel):
     
     # Coach assignment
     coach_email: Optional[str] = None
+    
+    # Location - can be physical address or online meeting link
+    location: Optional[str] = None
+    
+    # Membership plans - list of plan IDs this class is connected to
+    membership_plan_ids: List[int] = Field(default_factory=list)
 
     @field_validator("class_time", mode="before")
     @classmethod
@@ -133,6 +139,10 @@ class ClassUpdate(BaseModel):
     is_active: Optional[bool] = None
     
     coach_email: Optional[str] = None
+    
+    location: Optional[str] = None
+    
+    membership_plan_ids: Optional[List[int]] = None
 
     @field_validator("days_of_week", mode="before")
     @classmethod
@@ -153,6 +163,7 @@ class ClassOut(ClassBase):
     id: int
     created_at: Optional[datetime] = None
     updated_at: Optional[datetime] = None
+    membership_plan_ids: List[int] = Field(default_factory=list)
 
     model_config = ConfigDict(from_attributes=True)