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