CHIEFSOFT\ameye
77 lines
2.8 KiB
Python
77 lines
2.8 KiB
Python
from datetime import datetime, timezone
|
|
from app.extensions import db
|
|
from sqlalchemy.sql import func
|
|
|
|
# Inside Stripe_Product ===== : {'product_id': 'prod_SvB4RlFx6qXK6v', 'price_id': 'price_1RzKi4LjZLojw6IZA246JDAa'}
|
|
|
|
class SubscriptionOptions(db.Model):
|
|
__tablename__ = 'subscription_options'
|
|
|
|
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
|
|
uid = db.Column(db.String(150), nullable=True)
|
|
display_name = db.Column(db.String(25), nullable=False)
|
|
option_name = db.Column(db.String(100), nullable=False)
|
|
monthly = db.Column(db.Integer, nullable=True, default=0)
|
|
status = db.Column(db.Integer, nullable=True, default=1)
|
|
added = db.Column(db.DateTime(timezone=False), server_default=func.now())
|
|
updated = db.Column(db.DateTime(timezone=False), server_default=func.now(), onupdate=func.now())
|
|
stripe_product_id = db.Column(db.String(100), nullable=True)
|
|
stripe_price_id = db.Column(db.String(100), nullable=True)
|
|
country_code = db.Column(db.String(3), nullable=False)
|
|
|
|
|
|
@classmethod
|
|
def get_subscription_options(cls, status):
|
|
sub_options = cls.query.filter_by(status=status).all()
|
|
if not sub_options:
|
|
return None
|
|
return sub_options
|
|
|
|
@classmethod
|
|
def get_subscription_options_by_option_name(cls, option_name):
|
|
sub_options = cls.query.filter_by(option_name=option_name).all()
|
|
if not sub_options:
|
|
return None
|
|
return sub_options
|
|
|
|
|
|
@classmethod
|
|
def set_stripe_product_id(cls,package_uid, stripe_product_id):
|
|
# Retrieve
|
|
sub_option = cls.query.filter_by(uid=str(package_uid)).first()
|
|
|
|
if not sub_option:
|
|
raise ValueError(f"Reset with ID {package_uid} does not exist.")
|
|
|
|
# Update stripe_product_id
|
|
sub_option.stripe_product_id = stripe_product_id
|
|
|
|
@classmethod
|
|
def set_stripe_price_id(cls, package_uid, stripe_price_id):
|
|
# Retrieve
|
|
sub_option = cls.query.filter_by(uid=str(package_uid)).first()
|
|
|
|
if not sub_option:
|
|
raise ValueError(f"Reset with ID {package_uid} does not exist.")
|
|
|
|
# Update stripe_price_id
|
|
sub_option.stripe_price_id = stripe_price_id
|
|
|
|
|
|
def to_dict(self):
|
|
return {
|
|
"id": self.id,
|
|
"uid": self.uid,
|
|
"display_name": self.display_name,
|
|
"status": self.status,
|
|
"monthly": self.monthly,
|
|
"country_code": self.country_code,
|
|
"stripe_product_id": self.stripe_product_id,
|
|
"stripe_price_id": self.stripe_price_id,
|
|
"added": self.added.isoformat() if self.added else None,
|
|
"updated": self.updated.isoformat() if self.updated else None
|
|
}
|
|
|
|
def __repr__(self):
|
|
return f'<SubscriptionOptions {self.id} - {self.amount}>'
|