File size: 1,271 Bytes
cc69c66 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
"""
This file should contain all relevant data-only information, it should contain minimal to no behavior.
"""
from typing import Dict, List, Optional
from pydantic import BaseModel, Field, model_validator
class Property(BaseModel):
name: Optional[str] = None
type: Optional[str] = None
description: Optional[str] = None
model_config = {"extra": "ignore"}
def upsert_dict(mapping, kvp):
k, v = kvp
mapping[k] = v
return mapping
class DataModel(BaseModel):
name: Optional[str] = None
description: Optional[str] = None
links: Optional[List[str]] = Field(default={})
properties: Optional[Dict[str, Property]] = Field(default={})
model_config = {"extra": "ignore"}
@model_validator(mode="before")
def _normalize_properties(cls, node_content):
props = node_content.get("properties", {})
if not isinstance(props, list):
return node_content
props_to_use = [prop for prop in props if "name" in prop]
prop_items = [Property(**prop) for prop in props_to_use]
prop_dict = {
prop.name: Property(type=prop.type, description=prop.description)
for prop in prop_items
}
return upsert_dict(node_content, ("properties", prop_dict))
|