da1fb2a058
Refactor token validation and saving endpoints to accept a Token model instead of a raw string. Set default values for token generation request fields. Update TokenService cleanup to use datetime.now().
59 lines
1.9 KiB
Python
59 lines
1.9 KiB
Python
"""
|
|
Pydantic models for Token Management Service
|
|
"""
|
|
|
|
from pydantic import BaseModel, Field
|
|
from typing import List, Optional, Dict, Any
|
|
from datetime import datetime
|
|
|
|
class Token(BaseModel):
|
|
token: str = Field(..., description="JWT token")
|
|
|
|
class TokenGenerateRequest(BaseModel):
|
|
"""Request model for token generation"""
|
|
name: str = Field(..., description="Token owner name")
|
|
list_of_resources: List[str] = Field(..., description="List of accessible resources")
|
|
data_aggregation: bool = Field(default=False, description="Allow data aggregation")
|
|
time_aggregation: bool = Field(default=False, description="Allow time aggregation")
|
|
embargo: int = Field(default=0, description="Embargo period in seconds")
|
|
exp_hours: int = Field(default=24, description="Token expiration in hours")
|
|
|
|
class TokenResponse(BaseModel):
|
|
"""Response model for token operations"""
|
|
token: str = Field(..., description="JWT token")
|
|
|
|
class TokenValidationResponse(BaseModel):
|
|
"""Response model for token validation"""
|
|
valid: bool = Field(..., description="Whether token is valid")
|
|
token: str = Field(..., description="Original token")
|
|
decoded: Optional[Dict[str, Any]] = Field(None, description="Decoded token payload")
|
|
error: Optional[str] = Field(None, description="Error message if invalid")
|
|
|
|
class TokenRecord(BaseModel):
|
|
"""Token database record model"""
|
|
token: str
|
|
datetime: str
|
|
active: bool
|
|
name: str
|
|
resources: List[str]
|
|
expires_at: str
|
|
created_at: str
|
|
updated_at: str
|
|
|
|
class TokenListResponse(BaseModel):
|
|
"""Response model for token list"""
|
|
tokens: List[Dict[str, Any]]
|
|
count: int
|
|
|
|
class HealthResponse(BaseModel):
|
|
"""Health check response"""
|
|
service: str
|
|
status: str
|
|
timestamp: datetime
|
|
version: str
|
|
|
|
class Config:
|
|
json_encoders = {
|
|
datetime: lambda v: v.isoformat()
|
|
}
|