335 lines
No EOL
11 KiB
Python
335 lines
No EOL
11 KiB
Python
"""
|
|
Models for User Service
|
|
"""
|
|
|
|
import uuid
|
|
from pydantic import BaseModel, Field, EmailStr
|
|
from typing import List, Dict, Any, Optional
|
|
from datetime import datetime
|
|
from enum import Enum
|
|
from modules.shared.attributeUtils import register_model_labels, AttributeDefinition, ModelMixin
|
|
from modules.shared.timezoneUtils import get_utc_timestamp
|
|
|
|
class AuthAuthority(str, Enum):
|
|
"""Authentication authority enum"""
|
|
LOCAL = "local"
|
|
GOOGLE = "google"
|
|
MSFT = "msft"
|
|
|
|
class UserPrivilege(str, Enum):
|
|
"""User privilege levels"""
|
|
SYSADMIN = "sysadmin"
|
|
ADMIN = "admin"
|
|
USER = "user"
|
|
|
|
class ConnectionStatus(str, Enum):
|
|
"""Connection status"""
|
|
ACTIVE = "active"
|
|
EXPIRED = "expired"
|
|
REVOKED = "revoked"
|
|
PENDING = "pending"
|
|
|
|
class Mandate(BaseModel, ModelMixin):
|
|
"""Data model for a mandate"""
|
|
id: str = Field(
|
|
default_factory=lambda: str(uuid.uuid4()),
|
|
description="Unique ID of the mandate",
|
|
frontend_type="text",
|
|
frontend_readonly=True,
|
|
frontend_required=False
|
|
)
|
|
name: str = Field(
|
|
description="Name of the mandate",
|
|
frontend_type="text",
|
|
frontend_readonly=False,
|
|
frontend_required=True
|
|
)
|
|
language: str = Field(
|
|
default="en",
|
|
description="Default language of the mandate",
|
|
frontend_type="select",
|
|
frontend_readonly=False,
|
|
frontend_required=True,
|
|
frontend_options=[
|
|
{"value": "de", "label": {"en": "Deutsch", "fr": "Allemand"}},
|
|
{"value": "en", "label": {"en": "English", "fr": "Anglais"}},
|
|
{"value": "fr", "label": {"en": "Français", "fr": "Français"}},
|
|
{"value": "it", "label": {"en": "Italiano", "fr": "Italien"}}
|
|
]
|
|
)
|
|
enabled: bool = Field(
|
|
default=True,
|
|
description="Indicates whether the mandate is enabled",
|
|
frontend_type="checkbox",
|
|
frontend_readonly=False,
|
|
frontend_required=False
|
|
)
|
|
|
|
# Register labels for Mandate
|
|
register_model_labels(
|
|
"Mandate",
|
|
{"en": "Mandate", "fr": "Mandat"},
|
|
{
|
|
"id": {"en": "ID", "fr": "ID"},
|
|
"name": {"en": "Name", "fr": "Nom"},
|
|
"language": {"en": "Language", "fr": "Langue"},
|
|
"enabled": {"en": "Enabled", "fr": "Activé"}
|
|
}
|
|
)
|
|
|
|
class UserConnection(BaseModel, ModelMixin):
|
|
"""Data model for a user's connection to an external service"""
|
|
id: str = Field(
|
|
default_factory=lambda: str(uuid.uuid4()),
|
|
description="Unique ID of the connection",
|
|
frontend_type="text",
|
|
frontend_readonly=True,
|
|
frontend_required=False
|
|
)
|
|
userId: str = Field(
|
|
description="ID of the user this connection belongs to",
|
|
frontend_type="text",
|
|
frontend_readonly=True,
|
|
frontend_required=False
|
|
)
|
|
authority: AuthAuthority = Field(
|
|
description="Authentication authority",
|
|
frontend_type="select",
|
|
frontend_readonly=True,
|
|
frontend_required=False,
|
|
frontend_options=[
|
|
{"value": "local", "label": {"en": "Local", "fr": "Local"}},
|
|
{"value": "google", "label": {"en": "Google", "fr": "Google"}},
|
|
{"value": "msft", "label": {"en": "Microsoft", "fr": "Microsoft"}}
|
|
]
|
|
)
|
|
externalId: str = Field(
|
|
description="User ID in the external system",
|
|
frontend_type="text",
|
|
frontend_readonly=True,
|
|
frontend_required=False
|
|
)
|
|
externalUsername: str = Field(
|
|
description="Username in the external system",
|
|
frontend_type="text",
|
|
frontend_readonly=False,
|
|
frontend_required=False
|
|
)
|
|
externalEmail: Optional[EmailStr] = Field(
|
|
None,
|
|
description="Email in the external system",
|
|
frontend_type="email",
|
|
frontend_readonly=False,
|
|
frontend_required=False
|
|
)
|
|
status: ConnectionStatus = Field(
|
|
default=ConnectionStatus.ACTIVE,
|
|
description="Connection status",
|
|
frontend_type="select",
|
|
frontend_readonly=False,
|
|
frontend_required=False,
|
|
frontend_options=[
|
|
{"value": "active", "label": {"en": "Active", "fr": "Actif"}},
|
|
{"value": "inactive", "label": {"en": "Inactive", "fr": "Inactif"}},
|
|
{"value": "expired", "label": {"en": "Expired", "fr": "Expiré"}},
|
|
{"value": "pending", "label": {"en": "Pending", "fr": "En attente"}}
|
|
]
|
|
)
|
|
connectedAt: float = Field(
|
|
default_factory=get_utc_timestamp,
|
|
description="When the connection was established (UTC timestamp in seconds)",
|
|
frontend_type="timestamp",
|
|
frontend_readonly=True,
|
|
frontend_required=False
|
|
)
|
|
lastChecked: float = Field(
|
|
default_factory=get_utc_timestamp,
|
|
description="When the connection was last verified (UTC timestamp in seconds)",
|
|
frontend_type="timestamp",
|
|
frontend_readonly=True,
|
|
frontend_required=False
|
|
)
|
|
expiresAt: Optional[float] = Field(
|
|
None,
|
|
description="When the connection expires (UTC timestamp in seconds)",
|
|
frontend_type="timestamp",
|
|
frontend_readonly=True,
|
|
frontend_required=False
|
|
)
|
|
|
|
# Register labels for UserConnection
|
|
register_model_labels(
|
|
"UserConnection",
|
|
{"en": "User Connection", "fr": "Connexion utilisateur"},
|
|
{
|
|
"id": {"en": "ID", "fr": "ID"},
|
|
"userId": {"en": "User ID", "fr": "ID utilisateur"},
|
|
"authority": {"en": "Authority", "fr": "Autorité"},
|
|
"externalId": {"en": "External ID", "fr": "ID externe"},
|
|
"externalUsername": {"en": "External Username", "fr": "Nom d'utilisateur externe"},
|
|
"externalEmail": {"en": "External Email", "fr": "Email externe"},
|
|
"status": {"en": "Status", "fr": "Statut"},
|
|
"connectedAt": {"en": "Connected At", "fr": "Connecté le"},
|
|
"lastChecked": {"en": "Last Checked", "fr": "Dernière vérification"},
|
|
"expiresAt": {"en": "Expires At", "fr": "Expire le"}
|
|
}
|
|
)
|
|
|
|
|
|
|
|
class User(BaseModel, ModelMixin):
|
|
"""Data model for a user"""
|
|
id: str = Field(
|
|
default_factory=lambda: str(uuid.uuid4()),
|
|
description="Unique ID of the user",
|
|
frontend_type="text",
|
|
frontend_readonly=True,
|
|
frontend_required=False
|
|
)
|
|
username: str = Field(
|
|
description="Username for login",
|
|
frontend_type="text",
|
|
frontend_readonly=False,
|
|
frontend_required=True
|
|
)
|
|
email: Optional[EmailStr] = Field(
|
|
None,
|
|
description="Email address of the user",
|
|
frontend_type="email",
|
|
frontend_readonly=False,
|
|
frontend_required=True
|
|
)
|
|
fullName: Optional[str] = Field(
|
|
None,
|
|
description="Full name of the user",
|
|
frontend_type="text",
|
|
frontend_readonly=False,
|
|
frontend_required=False
|
|
)
|
|
language: str = Field(
|
|
default="en",
|
|
description="Preferred language of the user",
|
|
frontend_type="select",
|
|
frontend_readonly=False,
|
|
frontend_required=True,
|
|
frontend_options=[
|
|
{"value": "de", "label": {"en": "Deutsch", "fr": "Allemand"}},
|
|
{"value": "en", "label": {"en": "English", "fr": "Anglais"}},
|
|
{"value": "fr", "label": {"en": "Français", "fr": "Français"}},
|
|
{"value": "it", "label": {"en": "Italiano", "fr": "Italien"}}
|
|
]
|
|
)
|
|
enabled: bool = Field(
|
|
default=True,
|
|
description="Indicates whether the user is enabled",
|
|
frontend_type="checkbox",
|
|
frontend_readonly=False,
|
|
frontend_required=False
|
|
)
|
|
privilege: UserPrivilege = Field(
|
|
default=UserPrivilege.USER,
|
|
description="Permission level",
|
|
frontend_type="select",
|
|
frontend_readonly=False,
|
|
frontend_required=True,
|
|
frontend_options=[
|
|
{"value": "user", "label": {"en": "User", "fr": "Utilisateur"}},
|
|
{"value": "admin", "label": {"en": "Admin", "fr": "Administrateur"}},
|
|
{"value": "sysadmin", "label": {"en": "SysAdmin", "fr": "Administrateur système"}}
|
|
]
|
|
)
|
|
authenticationAuthority: AuthAuthority = Field(
|
|
default=AuthAuthority.LOCAL,
|
|
description="Primary authentication authority",
|
|
frontend_type="select",
|
|
frontend_readonly=True,
|
|
frontend_required=False,
|
|
frontend_options=[
|
|
{"value": "local", "label": {"en": "Local", "fr": "Local"}},
|
|
{"value": "google", "label": {"en": "Google", "fr": "Google"}},
|
|
{"value": "msft", "label": {"en": "Microsoft", "fr": "Microsoft"}}
|
|
]
|
|
)
|
|
mandateId: Optional[str] = Field(
|
|
None,
|
|
description="ID of the mandate this user belongs to",
|
|
frontend_type="text",
|
|
frontend_readonly=True,
|
|
frontend_required=False
|
|
)
|
|
|
|
# Register labels for User
|
|
register_model_labels(
|
|
"User",
|
|
{"en": "User", "fr": "Utilisateur"},
|
|
{
|
|
"id": {"en": "ID", "fr": "ID"},
|
|
"username": {"en": "Username", "fr": "Nom d'utilisateur"},
|
|
"email": {"en": "Email", "fr": "Email"},
|
|
"fullName": {"en": "Full Name", "fr": "Nom complet"},
|
|
"language": {"en": "Language", "fr": "Langue"},
|
|
"enabled": {"en": "Enabled", "fr": "Activé"},
|
|
"privilege": {"en": "Privilege", "fr": "Privilège"},
|
|
"authenticationAuthority": {"en": "Auth Authority", "fr": "Autorité d'authentification"},
|
|
"mandateId": {"en": "Mandate ID", "fr": "ID de mandat"}
|
|
}
|
|
)
|
|
|
|
class UserInDB(User):
|
|
"""Extended user class with password hash"""
|
|
hashedPassword: Optional[str] = Field(None, description="Hash of the user password")
|
|
|
|
# Register labels for UserInDB
|
|
register_model_labels(
|
|
"UserInDB",
|
|
{"en": "User Access", "fr": "Accès de l'utilisateur"},
|
|
{
|
|
"hashedPassword": {"en": "Password hash", "fr": "Hachage de mot de passe"}
|
|
}
|
|
)
|
|
|
|
# Token Models
|
|
class Token(BaseModel, ModelMixin):
|
|
"""Token model for all authentication types"""
|
|
id: Optional[str] = None
|
|
userId: str
|
|
authority: AuthAuthority
|
|
connectionId: Optional[str] = Field(None, description="ID of the connection this token belongs to")
|
|
tokenAccess: str
|
|
tokenType: str = "bearer"
|
|
expiresAt: float = Field(description="When the token expires (UTC timestamp in seconds)")
|
|
tokenRefresh: Optional[str] = None
|
|
createdAt: Optional[float] = Field(None, description="When the token was created (UTC timestamp in seconds)")
|
|
|
|
class Config:
|
|
useEnumValues = True
|
|
|
|
# Register labels for Token
|
|
register_model_labels(
|
|
"Token",
|
|
{"en": "Token", "fr": "Jeton"},
|
|
{
|
|
"id": {"en": "ID", "fr": "ID"},
|
|
"userId": {"en": "User ID", "fr": "ID utilisateur"},
|
|
"authority": {"en": "Authority", "fr": "Autorité"},
|
|
"connectionId": {"en": "Connection ID", "fr": "ID de connexion"},
|
|
"tokenAccess": {"en": "Access Token", "fr": "Jeton d'accès"},
|
|
"tokenType": {"en": "Token Type", "fr": "Type de jeton"},
|
|
"expiresAt": {"en": "Expires At", "fr": "Expire le"},
|
|
"tokenRefresh": {"en": "Refresh Token", "fr": "Jeton de rafraîchissement"},
|
|
"createdAt": {"en": "Created At", "fr": "Créé le"}
|
|
}
|
|
)
|
|
|
|
class LocalToken(Token):
|
|
"""Local authentication token model"""
|
|
pass
|
|
|
|
class GoogleToken(Token):
|
|
"""Google OAuth token model"""
|
|
pass
|
|
|
|
class MsftToken(Token):
|
|
"""Microsoft OAuth token model"""
|
|
pass
|
|
|