Update app/models/membership_invite.py

app/models/membership_invite.py

@@ -1,9 +1,27 @@
 # app/models/membership_invite.py
 from datetime import datetime
+import json
 from sqlalchemy import Column, Integer, String, Text, DateTime, ForeignKey
+from sqlalchemy.types import TypeDecorator, TEXT
 from core.database import Base
 
 
+class JSONEncodedDict(TypeDecorator):
+    """Represents an immutable structure as a json-encoded string."""
+    impl = TEXT
+    cache_ok = True
+
+    def process_bind_param(self, value, dialect):
+        if value is not None:
+            return json.dumps(value)
+        return None
+
+    def process_result_value(self, value, dialect):
+        if value is not None:
+            return json.loads(value)
+        return None
+
+
 class MembershipInvite(Base):
     __tablename__ = "membership_invites"
 
@@ -19,6 +37,7 @@ class MembershipInvite(Base):
     invited_by = Column(String(255), nullable=True)  # Admin email/name
     invite_date = Column(DateTime, default=datetime.utcnow)
     completed_at = Column(DateTime, nullable=True)
+    students_data = Column(JSONEncodedDict, nullable=True)  # Store student data temporarily before payment
     created_at = Column(DateTime, default=datetime.utcnow)
     updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)