Update app/models/class_.py

app/models/class_.py

@@ -35,9 +35,9 @@ class Class(Base):
     
     # Coach assignment
     coach_email = Column(String(255), nullable=True, index=True)
-
-    # No membership plan link anymore – kept fully independent
-    # (plan_id / membership_plan_id removed)
+    
+    # Location - can be physical address or online meeting link
+    location = Column(Text, nullable=True)
 
     # Status + timestamps
     is_active = Column(Boolean, default=True)
@@ -55,6 +55,12 @@ class Class(Base):
         back_populates="class_",
         cascade="all, delete-orphan",
     )
+    # Many-to-many relationship with membership plans
+    membership_plans = relationship(
+        "ClassMembershipPlan",
+        back_populates="class_",
+        cascade="all, delete-orphan",
+    )
 
 
 class ClassEnrollment(Base):
@@ -90,3 +96,29 @@ class ClassEnrollment(Base):
     removed_at = Column(DateTime, nullable=True)
 
     class_ = relationship("Class", back_populates="enrollments")
+
+
+class ClassMembershipPlan(Base):
+    """Many-to-many relationship between classes and membership plans."""
+    __tablename__ = "class_membership_plans"
+
+    id = Column(Integer, primary_key=True, index=True)
+    
+    class_id = Column(
+        Integer,
+        ForeignKey("classes.id"),
+        nullable=False,
+        index=True,
+    )
+    
+    membership_plan_id = Column(
+        Integer,
+        ForeignKey("membership_plans.id"),
+        nullable=False,
+        index=True,
+    )
+    
+    created_at = Column(DateTime, default=datetime.utcnow)
+    
+    # Relationships
+    class_ = relationship("Class", back_populates="membership_plans")