Coverage for backend/pinch_service/docs/types/VisualiseTypes.py: 100%

1from typing import Any, List 

2from pydantic import BaseModel 

3from OpenPinch.src.lib.enums import GraphType 

4from OpenPinch.src.lib.schema import GraphSet 

5 

6class Graph(BaseModel): 

7 type: GraphType 

8 data: Any 

9 

10 class Config: 

11 use_enum_values = True 

12 

13class Zone(BaseModel): 

14 name: str = "Zone" 

15 graphs: List[Graph] 

16 

17class VisualiseRequest(BaseModel): 

18 zones: List[Zone] 

19 

20class VisualiseResponse(BaseModel): 

21 graphs: List[GraphSet] 

22