|
| 1 | +from abc import ABC, abstractmethod |
1 | 2 | from typing import Any, Dict, Optional |
2 | 3 |
|
3 | 4 | from pydantic import BaseModel |
|
6 | 7 | from .job import Job |
7 | 8 |
|
8 | 9 |
|
9 | | -class InvokeProcess(BaseModel): |
| 10 | +class Serializable(ABC): |
| 11 | + @abstractmethod |
| 12 | + def serialize(self) -> str: |
| 13 | + pass |
| 14 | + |
| 15 | + |
| 16 | +class InvokeProcess(BaseModel, Serializable): |
10 | 17 | name: str |
11 | 18 | input_arguments: Optional[Dict[str, Any]] |
12 | 19 |
|
| 20 | + def serialize(self) -> str: |
| 21 | + try: |
| 22 | + return self.model_dump_json() |
| 23 | + except ValueError: |
| 24 | + return str(self) |
| 25 | + |
13 | 26 |
|
14 | | -class WaitJob(BaseModel): |
| 27 | +class WaitJob(BaseModel, Serializable): |
15 | 28 | job: Job |
16 | 29 |
|
| 30 | + def serialize(self) -> str: |
| 31 | + try: |
| 32 | + return self.model_dump_json() |
| 33 | + except ValueError: |
| 34 | + return str(self) |
17 | 35 |
|
18 | | -class CreateAction(BaseModel): |
| 36 | + |
| 37 | +class CreateAction(BaseModel, Serializable): |
19 | 38 | name: Optional[str] = None |
20 | 39 | key: Optional[str] = None |
21 | 40 | title: str |
22 | 41 | data: Optional[Dict[str, Any]] = None |
23 | 42 | app_version: Optional[int] = 1 |
24 | 43 | assignee: Optional[str] = "" |
25 | 44 |
|
| 45 | + def serialize(self) -> str: |
| 46 | + try: |
| 47 | + return self.model_dump_json() |
| 48 | + except ValueError: |
| 49 | + return str(self) |
| 50 | + |
26 | 51 |
|
27 | | -class WaitAction(BaseModel): |
| 52 | +class WaitAction(BaseModel, Serializable): |
28 | 53 | action: Action |
| 54 | + |
| 55 | + def serialize(self) -> str: |
| 56 | + try: |
| 57 | + return self.model_dump_json() |
| 58 | + except ValueError: |
| 59 | + return str(self) |
0 commit comments