mirror of
https://github.com/Monadical-SAS/cubbi.git
synced 2025-12-20 20:29:06 +00:00
69 lines
1.6 KiB
Python
69 lines
1.6 KiB
Python
from enum import Enum
|
|
from typing import Dict, List, Optional
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class SessionStatus(str, Enum):
|
|
CREATING = "creating"
|
|
RUNNING = "running"
|
|
STOPPED = "stopped"
|
|
FAILED = "failed"
|
|
|
|
|
|
class DriverEnvironmentVariable(BaseModel):
|
|
name: str
|
|
description: str
|
|
required: bool = False
|
|
default: Optional[str] = None
|
|
sensitive: bool = False
|
|
|
|
|
|
class PersistentConfig(BaseModel):
|
|
source: str
|
|
target: str
|
|
type: str # "directory" or "file"
|
|
description: str = ""
|
|
|
|
|
|
class VolumeMount(BaseModel):
|
|
mountPath: str
|
|
description: str = ""
|
|
|
|
|
|
class DriverInit(BaseModel):
|
|
pre_command: Optional[str] = None
|
|
command: str
|
|
|
|
|
|
class Driver(BaseModel):
|
|
name: str
|
|
description: str
|
|
version: str
|
|
maintainer: str
|
|
image: str
|
|
init: Optional[DriverInit] = None
|
|
environment: List[DriverEnvironmentVariable] = []
|
|
ports: List[int] = []
|
|
volumes: List[VolumeMount] = []
|
|
persistent_configs: List[PersistentConfig] = []
|
|
|
|
|
|
class Session(BaseModel):
|
|
id: str
|
|
name: str
|
|
driver: str
|
|
status: SessionStatus
|
|
container_id: Optional[str] = None
|
|
environment: Dict[str, str] = Field(default_factory=dict)
|
|
project: Optional[str] = None
|
|
created_at: str
|
|
ports: Dict[int, int] = Field(default_factory=dict)
|
|
|
|
|
|
class Config(BaseModel):
|
|
docker: Dict[str, str] = Field(default_factory=dict)
|
|
drivers: Dict[str, Driver] = Field(default_factory=dict)
|
|
defaults: Dict[str, object] = Field(
|
|
default_factory=dict
|
|
) # Can store strings, booleans, or other values
|