mirror of
https://github.com/bunny-lab-io/Borealis.git
synced 2025-12-15 01:55:48 -07:00
110 lines
3.0 KiB
Python
110 lines
3.0 KiB
Python
# ======================================================
|
|
# Data\Engine\assembly_management\models.py
|
|
# Description: Dataclasses describing assemblies, payload descriptors, and cache state.
|
|
#
|
|
# API Endpoints (if applicable): None
|
|
# ======================================================
|
|
|
|
"""Model definitions for assembly persistence."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import datetime as _dt
|
|
from dataclasses import dataclass, field
|
|
from enum import Enum
|
|
from typing import Any, Dict, Optional
|
|
|
|
|
|
class AssemblyDomain(str, Enum):
|
|
"""Logical source domains mapped to dedicated SQLite databases."""
|
|
|
|
OFFICIAL = "official"
|
|
COMMUNITY = "community"
|
|
USER = "user"
|
|
|
|
@property
|
|
def database_name(self) -> str:
|
|
mapping = {
|
|
AssemblyDomain.OFFICIAL: "official.db",
|
|
AssemblyDomain.COMMUNITY: "community.db",
|
|
AssemblyDomain.USER: "user_created.db",
|
|
}
|
|
return mapping[self]
|
|
|
|
|
|
class PayloadType(str, Enum):
|
|
"""Supported payload classifications."""
|
|
|
|
SCRIPT = "script"
|
|
WORKFLOW = "workflow"
|
|
BINARY = "binary"
|
|
UNKNOWN = "unknown"
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class PayloadDescriptor:
|
|
"""Represents on-disk payload material referenced by an assembly."""
|
|
|
|
guid: str
|
|
payload_type: PayloadType
|
|
file_name: str
|
|
file_extension: str
|
|
size_bytes: int
|
|
checksum: Optional[str]
|
|
created_at: _dt.datetime
|
|
updated_at: _dt.datetime
|
|
|
|
def as_dict(self) -> Dict[str, Any]:
|
|
return {
|
|
"guid": self.guid,
|
|
"payload_type": self.payload_type.value,
|
|
"file_name": self.file_name,
|
|
"file_extension": self.file_extension,
|
|
"size_bytes": self.size_bytes,
|
|
"checksum": self.checksum,
|
|
"created_at": self.created_at.isoformat(),
|
|
"updated_at": self.updated_at.isoformat(),
|
|
}
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class AssemblyRecord:
|
|
"""Represents an assembly row hydrated from persistence."""
|
|
|
|
assembly_id: str
|
|
display_name: str
|
|
summary: Optional[str]
|
|
category: Optional[str]
|
|
assembly_kind: str
|
|
assembly_type: Optional[str]
|
|
version: int
|
|
payload: PayloadDescriptor
|
|
metadata: Dict[str, Any] = field(default_factory=dict)
|
|
tags: Dict[str, Any] = field(default_factory=dict)
|
|
checksum: Optional[str] = None
|
|
created_at: _dt.datetime = field(default_factory=_dt.datetime.utcnow)
|
|
updated_at: _dt.datetime = field(default_factory=_dt.datetime.utcnow)
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class CachedAssembly:
|
|
"""Wrapper stored in memory with dirty state tracking."""
|
|
|
|
domain: AssemblyDomain
|
|
record: AssemblyRecord
|
|
is_dirty: bool = False
|
|
last_persisted: Optional[_dt.datetime] = None
|
|
dirty_since: Optional[_dt.datetime] = None
|
|
|
|
def mark_dirty(self) -> None:
|
|
now = _dt.datetime.utcnow()
|
|
self.is_dirty = True
|
|
self.dirty_since = self.dirty_since or now
|
|
|
|
def mark_clean(self) -> None:
|
|
now = _dt.datetime.utcnow()
|
|
self.is_dirty = False
|
|
self.last_persisted = now
|
|
self.dirty_since = None
|
|
|