Code organization and refactoring
This commit is contained in:
18
models/__init__.py
Normal file
18
models/__init__.py
Normal file
@@ -0,0 +1,18 @@
|
||||
"""API models package."""
|
||||
from .api_models import (
|
||||
FactionRequest,
|
||||
AssignMemberRequest,
|
||||
RemoveAssignmentRequest,
|
||||
BotControl,
|
||||
DiscordMappingRequest,
|
||||
ConfigUpdateRequest,
|
||||
)
|
||||
|
||||
__all__ = [
|
||||
"FactionRequest",
|
||||
"AssignMemberRequest",
|
||||
"RemoveAssignmentRequest",
|
||||
"BotControl",
|
||||
"DiscordMappingRequest",
|
||||
"ConfigUpdateRequest",
|
||||
]
|
||||
BIN
models/__pycache__/__init__.cpython-311.pyc
Normal file
BIN
models/__pycache__/__init__.cpython-311.pyc
Normal file
Binary file not shown.
BIN
models/__pycache__/api_models.cpython-311.pyc
Normal file
BIN
models/__pycache__/api_models.cpython-311.pyc
Normal file
Binary file not shown.
32
models/api_models.py
Normal file
32
models/api_models.py
Normal file
@@ -0,0 +1,32 @@
|
||||
"""Pydantic models for API requests and responses."""
|
||||
from typing import Optional
|
||||
from pydantic import BaseModel
|
||||
|
||||
|
||||
class FactionRequest(BaseModel):
|
||||
faction_id: int
|
||||
interval: int = 30
|
||||
|
||||
|
||||
class AssignMemberRequest(BaseModel):
|
||||
group_id: str # "1", "2", ... "5"
|
||||
kind: str # "friendly" or "enemy"
|
||||
member_id: int
|
||||
|
||||
|
||||
class RemoveAssignmentRequest(BaseModel):
|
||||
member_id: int
|
||||
|
||||
|
||||
class BotControl(BaseModel):
|
||||
action: str # "start" or "stop"
|
||||
|
||||
|
||||
class DiscordMappingRequest(BaseModel):
|
||||
torn_id: int
|
||||
discord_id: int
|
||||
|
||||
|
||||
class ConfigUpdateRequest(BaseModel):
|
||||
key: str
|
||||
value: Optional[str | int]
|
||||
Reference in New Issue
Block a user