Improve subcription support
This commit is contained in:
@@ -4,7 +4,7 @@ Tenant models for bakery management - FIXED
|
||||
Removed cross-service User relationship to eliminate circular dependencies
|
||||
"""
|
||||
|
||||
from sqlalchemy import Column, String, Boolean, DateTime, Float, ForeignKey, Text, Integer
|
||||
from sqlalchemy import Column, String, Boolean, DateTime, Float, ForeignKey, Text, Integer, JSON
|
||||
from sqlalchemy.dialects.postgresql import UUID
|
||||
from sqlalchemy.orm import relationship
|
||||
from datetime import datetime, timezone
|
||||
@@ -35,7 +35,7 @@ class Tenant(Base):
|
||||
|
||||
# Status
|
||||
is_active = Column(Boolean, default=True)
|
||||
subscription_tier = Column(String(50), default="basic")
|
||||
subscription_tier = Column(String(50), default="starter")
|
||||
|
||||
# ML status
|
||||
model_trained = Column(Boolean, default=False)
|
||||
@@ -92,7 +92,7 @@ class Subscription(Base):
|
||||
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
||||
tenant_id = Column(UUID(as_uuid=True), ForeignKey("tenants.id", ondelete="CASCADE"), nullable=False)
|
||||
|
||||
plan = Column(String(50), default="basic") # basic, professional, enterprise
|
||||
plan = Column(String(50), default="starter") # starter, professional, enterprise
|
||||
status = Column(String(50), default="active") # active, suspended, cancelled
|
||||
|
||||
# Billing
|
||||
@@ -102,10 +102,13 @@ class Subscription(Base):
|
||||
trial_ends_at = Column(DateTime(timezone=True))
|
||||
|
||||
# Limits
|
||||
max_users = Column(Integer, default=1)
|
||||
max_users = Column(Integer, default=5)
|
||||
max_locations = Column(Integer, default=1)
|
||||
max_products = Column(Integer, default=50)
|
||||
|
||||
# Features - Store plan features as JSON
|
||||
features = Column(JSON)
|
||||
|
||||
# Timestamps
|
||||
created_at = Column(DateTime(timezone=True), default=lambda: datetime.now(timezone.utc))
|
||||
updated_at = Column(DateTime(timezone=True), default=lambda: datetime.now(timezone.utc), onupdate=lambda: datetime.now(timezone.utc))
|
||||
|
||||
Reference in New Issue
Block a user