06/15 Phase 4 + 5 codes

2026-06-15 17:43:47 -04:00
parent f57a95013c
commit a9efd15a76
36 changed files with 1677 additions and 57 deletions
@@ -0,0 +1,108 @@
+"""Monetization models: Subscription, Transaction, Boost.
+
+Stripe is source of truth for subscriptions. Local rows mirror state
+and are reconciled via webhooks + nightly job.
+Money stored as integer cents throughout.
+"""
+from datetime import datetime
+from sqlalchemy import JSON
+from app.extensions import db
+
+
+class Subscription(db.Model):
+    __tablename__ = "subscriptions"
+
+    id = db.Column(db.BigInteger().with_variant(db.Integer, "sqlite"),
+                   primary_key=True, autoincrement=True)
+    user_id = db.Column(db.BigInteger().with_variant(db.Integer, "sqlite"),
+                        db.ForeignKey("users.id"), nullable=False,
+                        unique=True, index=True)
+    plan_id = db.Column(db.BigInteger().with_variant(db.Integer, "sqlite"),
+                        db.ForeignKey("plans.id"), nullable=False)
+    stripe_customer_id = db.Column(db.String(64), nullable=True, index=True)
+    stripe_sub_id = db.Column(db.String(64), nullable=True, unique=True, index=True)
+    status = db.Column(
+        db.Enum("active", "past_due", "canceled", "trialing",
+                name="sub_status"),
+        nullable=False, default="active")
+    current_period_end = db.Column(db.DateTime, nullable=True)
+    cancel_at_period_end = db.Column(db.Boolean, nullable=False, default=False)
+    created_at = db.Column(db.DateTime, default=datetime.utcnow, nullable=False)
+    updated_at = db.Column(db.DateTime, default=datetime.utcnow,
+                           onupdate=datetime.utcnow, nullable=False)
+
+    user = db.relationship("User",
+                           backref=db.backref("subscription", uselist=False))
+    plan = db.relationship("Plan",
+                           backref=db.backref("subscriptions", lazy="dynamic"))
+
+    @property
+    def is_active(self):
+        return self.status in ("active", "trialing")
+
+    def __repr__(self):
+        return f"<Subscription u{self.user_id} {self.status}>"
+
+
+class Transaction(db.Model):
+    __tablename__ = "transactions"
+
+    id = db.Column(db.BigInteger().with_variant(db.Integer, "sqlite"),
+                   primary_key=True, autoincrement=True)
+    user_id = db.Column(db.BigInteger().with_variant(db.Integer, "sqlite"),
+                        db.ForeignKey("users.id"), nullable=False, index=True)
+    type = db.Column(
+        db.Enum("subscription", "boost", "refund", name="txn_type"),
+        nullable=False)
+    amount_cents = db.Column(db.Integer, nullable=False, default=0)
+    currency = db.Column(db.String(3), nullable=False, default="usd")
+    stripe_object_id = db.Column(db.String(64), nullable=True, index=True)
+    status = db.Column(db.String(20), nullable=False, default="succeeded")
+    meta = db.Column(JSON, nullable=True)
+    created_at = db.Column(db.DateTime, default=datetime.utcnow, nullable=False)
+
+    user = db.relationship("User",
+                           backref=db.backref("transactions", lazy="dynamic"))
+
+    def __repr__(self):
+        return f"<Transaction {self.type} ${self.amount_cents/100:.2f}>"
+
+
+class Boost(db.Model):
+    __tablename__ = "boosts"
+
+    # Boost types and their default durations in days
+    TYPES = {
+        "featured": {"label": "Featured listing", "days": 7,  "cents": 499},
+        "bump":     {"label": "Bump to top",       "days": 3,  "cents": 199},
+        "highlight":{"label": "Highlight",         "days": 7,  "cents": 299},
+        "urgent":   {"label": "Urgent tag",        "days": 7,  "cents": 199},
+    }
+
+    id = db.Column(db.BigInteger().with_variant(db.Integer, "sqlite"),
+                   primary_key=True, autoincrement=True)
+    listing_id = db.Column(db.BigInteger().with_variant(db.Integer, "sqlite"),
+                           db.ForeignKey("listings.id"), nullable=False, index=True)
+    user_id = db.Column(db.BigInteger().with_variant(db.Integer, "sqlite"),
+                        db.ForeignKey("users.id"), nullable=False)
+    type = db.Column(
+        db.Enum("featured", "bump", "highlight", "urgent", name="boost_type"),
+        nullable=False)
+    expires_at = db.Column(db.DateTime, nullable=False, index=True)
+    transaction_id = db.Column(
+        db.BigInteger().with_variant(db.Integer, "sqlite"),
+        db.ForeignKey("transactions.id"), nullable=True)
+    created_at = db.Column(db.DateTime, default=datetime.utcnow, nullable=False)
+
+    listing = db.relationship("Listing",
+                              backref=db.backref("boosts", lazy="selectin"))
+    user = db.relationship("User",
+                           backref=db.backref("boosts", lazy="dynamic"))
+    transaction = db.relationship("Transaction")
+
+    @property
+    def is_active(self):
+        return self.expires_at > datetime.utcnow()
+
+    def __repr__(self):
+        return f"<Boost {self.type} L{self.listing_id} exp={self.expires_at.date()}>"