- Rearranged imports in multiple files for consistency and clarity. - Updated logging middleware to exclude specific paths from logging. - Enhanced security module by cleaning up token handling and improving tenant validation. - Added tenant and company scoped mixins for better database model management. - Implemented generic CRUD routes for tenant-scoped resources. - Improved error handling and response management in API routes. - Cleaned up login and logout processes to ensure proper session management. - Introduced mechanisms to clear local storage and cookies on tenant change. - Enhanced company store to detect tenant changes and clear data accordingly. - Added new DTO mixins for currency and value affect flags.
95 lines
2.8 KiB
Python
95 lines
2.8 KiB
Python
"""
|
|
Modelos ORM para gestión de licencias
|
|
"""
|
|
|
|
import enum
|
|
|
|
from api.v1.common.base_models import TimestampMixin
|
|
from core.database import Base
|
|
from sqlalchemy import Boolean, Column, DateTime
|
|
from sqlalchemy import Enum as SQLEnum
|
|
from sqlalchemy import ForeignKey, Integer
|
|
|
|
|
|
class LicensePlan(enum.Enum):
|
|
"""Planes de licencia disponibles"""
|
|
|
|
FREE = "free"
|
|
BASIC = "basic"
|
|
PROFESSIONAL = "professional"
|
|
ENTERPRISE = "enterprise"
|
|
|
|
|
|
class LicenseStatus(enum.Enum):
|
|
"""Estados de licencia"""
|
|
|
|
ACTIVE = "active"
|
|
EXPIRED = "expired"
|
|
SUSPENDED = "suspended"
|
|
PENDING = "pending"
|
|
CANCELLED = "cancelled"
|
|
|
|
|
|
class License(Base, TimestampMixin):
|
|
"""
|
|
Modelo de Licencia - Control de planes y límites por tenant
|
|
"""
|
|
|
|
__tablename__ = "licenses"
|
|
__table_args__ = {"schema": "a76"}
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
tenant_id = Column(
|
|
Integer, ForeignKey("a76.tenants.id"), nullable=False, unique=True, index=True
|
|
)
|
|
|
|
# Plan y características
|
|
plan = Column(SQLEnum(LicensePlan), default=LicensePlan.FREE, nullable=False)
|
|
status = Column(
|
|
SQLEnum(LicenseStatus), default=LicenseStatus.PENDING, nullable=False
|
|
)
|
|
|
|
# Límites del plan
|
|
max_users = Column(Integer, default=5, nullable=False)
|
|
max_storage_gb = Column(Integer, default=10, nullable=False)
|
|
max_monthly_operations = Column(Integer, default=1000, nullable=False)
|
|
|
|
# Features habilitadas (booleans)
|
|
feature_api_access = Column(Boolean, default=True)
|
|
feature_advanced_reports = Column(Boolean, default=False)
|
|
feature_integrations = Column(Boolean, default=False)
|
|
feature_dedicated_support = Column(Boolean, default=False)
|
|
|
|
# Vigencia
|
|
starts_at = Column(DateTime(timezone=True), nullable=False)
|
|
expires_at = Column(DateTime(timezone=True), nullable=False)
|
|
|
|
def __repr__(self):
|
|
return f"<License(tenant_id={self.tenant_id}, plan={self.plan.value}, status={self.status.value})>"
|
|
|
|
|
|
class LicenseUsage(Base, TimestampMixin):
|
|
"""
|
|
Modelo para tracking de uso de licencia
|
|
"""
|
|
|
|
__tablename__ = "license_usage"
|
|
__table_args__ = {"schema": "a76"}
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
tenant_id = Column(
|
|
Integer, ForeignKey("a76.tenants.id"), nullable=False, index=True
|
|
)
|
|
|
|
# Métricas de uso
|
|
period_start = Column(DateTime(timezone=True), nullable=False)
|
|
period_end = Column(DateTime(timezone=True), nullable=False)
|
|
|
|
active_users = Column(Integer, default=0)
|
|
storage_used_gb = Column(Integer, default=0)
|
|
operations_count = Column(Integer, default=0)
|
|
api_calls_count = Column(Integer, default=0)
|
|
|
|
def __repr__(self):
|
|
return f"<LicenseUsage(tenant_id={self.tenant_id}, operations={self.operations_count})>"
|