set user product

2025-08-30 06:28:39 -04:00
parent 74d78bd883
commit e8a53e63cc
2 changed files with 13 additions and 1 deletions
@@ -113,12 +113,21 @@ class Members(db.Model):
    def set_user_stripe_id(cls, member_uid, member_id, stripe_customer_id):
        member = cls.query.filter_by(uid=str(member_uid),id=member_id).first()
        if not member:
-            raise ValueError(f"Reset with UID {member_uid} does not exist.")
+            raise ValueError(f"Member with UID {member_uid} does not exist.")
        # Update stripe_customer_id
        member.stripe_customer_id = stripe_customer_id

        return stripe_customer_id

+    def set_user_option_name(cls, member_id, option_name):
+        member = cls.query.filter_by(id=member_id).first()
+        if not member:
+            raise ValueError(f"Member with ID {member_id} does not exist.")
+        # Update stripe_customer_id
+        member.option_name = option_name
+
+        return option_name
+

    @classmethod
    def get_all_member(cls, email=None, username=None, page=1, limit=20):