21 lines
404 B
Python
21 lines
404 B
Python
from dataclasses import dataclass, field
|
|
from enum import Enum
|
|
from typing import Optional
|
|
|
|
|
|
DEFAULT_GRPC_TIMEOUT = 5
|
|
|
|
|
|
class TimeUnit(Enum):
|
|
MINUTES = "MINUTES"
|
|
SECONDS = "SECONDS"
|
|
MILLISECONDS = "MILLISECONDS"
|
|
|
|
@dataclass
|
|
class CallContextParam:
|
|
timeout: int = DEFAULT_GRPC_TIMEOUT
|
|
time_unit: TimeUnit = TimeUnit.SECONDS
|
|
|
|
@classmethod
|
|
def default(cls):
|
|
return cls()
|