84 lines
2.2 KiB
Python
84 lines
2.2 KiB
Python
|
from dataclasses import dataclass, field, asdict
|
||
|
|
from typing import Dict, Any, List, Optional
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class EpochInfo:
|
||
|
|
major: int
|
||
|
|
minor: int = 0
|
||
|
|
stability: str = "stable"
|
||
|
|
|
||
|
|
def to_dict(self) -> Dict[str, Any]:
|
||
|
|
return asdict(self)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def from_dict(cls, data: Dict[str, Any]) -> "EpochInfo":
|
||
|
|
return cls(**data)
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class ContributorInfo:
|
||
|
|
name: str
|
||
|
|
registered_at: Optional[int] = None
|
||
|
|
contribution_count: int = 0
|
||
|
|
|
||
|
|
def to_dict(self) -> Dict[str, Any]:
|
||
|
|
return asdict(self)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def from_dict(cls, data: Dict[str, Any]) -> "ContributorInfo":
|
||
|
|
return cls(**data)
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class SegmentLineage:
|
||
|
|
segment_id: str
|
||
|
|
start_line: int
|
||
|
|
end_line: int
|
||
|
|
start_byte: int
|
||
|
|
end_byte: int
|
||
|
|
|
||
|
|
def to_dict(self) -> Dict[str, Any]:
|
||
|
|
return asdict(self)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def from_dict(cls, data: Dict[str, Any]) -> "SegmentLineage":
|
||
|
|
return cls(**data)
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class CodexEntry:
|
||
|
|
source_file: str
|
||
|
|
source_type: str
|
||
|
|
version: str
|
||
|
|
origin: str
|
||
|
|
epoch: EpochInfo
|
||
|
|
contributor: ContributorInfo
|
||
|
|
segments: List[SegmentLineage] = field(default_factory=list)
|
||
|
|
timestamp: Optional[str] = None
|
||
|
|
|
||
|
|
def to_dict(self) -> Dict[str, Any]:
|
||
|
|
return {
|
||
|
|
"source_file": self.source_file,
|
||
|
|
"source_type": self.source_type,
|
||
|
|
"version": self.version,
|
||
|
|
"origin": self.origin,
|
||
|
|
"epoch": self.epoch.to_dict(),
|
||
|
|
"contributor": self.contributor.to_dict(),
|
||
|
|
"segments": [s.to_dict() for s in self.segments],
|
||
|
|
"timestamp": self.timestamp
|
||
|
|
}
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def from_dict(cls, data: Dict[str, Any]) -> "CodexEntry":
|
||
|
|
return cls(
|
||
|
|
source_file=data.get("source_file", "unknown"),
|
||
|
|
source_type=data.get("source_type", ".py"),
|
||
|
|
version=data.get("version", "1.0.0"),
|
||
|
|
origin=data.get("origin", "unknown"),
|
||
|
|
epoch=EpochInfo.from_dict(data.get("epoch", {"major": 1})),
|
||
|
|
contributor=ContributorInfo.from_dict(data.get("contributor", {"name": "unknown"})),
|
||
|
|
segments=[SegmentLineage.from_dict(s) for s in data.get("segments", [])],
|
||
|
|
timestamp=data.get("timestamp")
|
||
|
|
)
|