acazares
52b8fcd434
- Enhanced TenantMiddleware to validate tenant information from JWT tokens. - Added LicenseValidationMiddleware to check tenant licenses before processing requests. - Updated security utilities to extract tenant information from tokens and validate company access. - Introduced CompanyStore to manage active company state and handle company switching in the frontend. - Modified API routes to include company_id in requests for better resource management. - Improved logging and error handling throughout the middleware and API layers. - Updated frontend components to reflect changes in company management and selection. - Added new API route for fetching user's companies with proper authentication handling.
121 lines
3.5 KiB
Python
121 lines
3.5 KiB
Python
"""
|
|
Modelos ORM para gestión de licencias
|
|
"""
|
|
|
|
from datetime import datetime
|
|
from sqlalchemy import (
|
|
Column,
|
|
Integer,
|
|
String,
|
|
DateTime,
|
|
Boolean,
|
|
ForeignKey,
|
|
Enum as SQLEnum,
|
|
)
|
|
from sqlalchemy.sql import func
|
|
from sqlalchemy.orm import relationship
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
from core.database import Base
|
|
import enum
|
|
|
|
|
|
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):
|
|
"""
|
|
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)
|
|
|
|
# Timestamps
|
|
created_at: Mapped[datetime] = mapped_column(
|
|
DateTime, nullable=False, default=func.now()
|
|
)
|
|
updated_at: Mapped[datetime] = mapped_column(
|
|
DateTime, nullable=False, default=func.now(), onupdate=func.now()
|
|
)
|
|
deleted_at: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
|
|
|
|
def __repr__(self):
|
|
return f"<License(tenant_id={self.tenant_id}, plan={self.plan.value}, status={self.status.value})>"
|
|
|
|
|
|
class LicenseUsage(Base):
|
|
"""
|
|
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)
|
|
|
|
# Timestamps
|
|
created_at: Mapped[datetime] = mapped_column(
|
|
DateTime, nullable=False, default=func.now()
|
|
)
|
|
updated_at: Mapped[datetime] = mapped_column(
|
|
DateTime, nullable=False, default=func.now(), onupdate=func.now()
|
|
)
|
|
deleted_at: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
|
|
|
|
def __repr__(self):
|
|
return f"<LicenseUsage(tenant_id={self.tenant_id}, operations={self.operations_count})>"
|