Updating VimCreateInput and UpdateVimStatusInput Dataclass
Change-Id: I2e0e0ebb943dbdf0a5ae7b33b36fda93b999d9d8
Signed-off-by: Gulsum Atici <gulsum.atici@canonical.com>
Signed-off-by: Mark Beierl <mark.beierl@canonical.com>
diff --git a/osm_common/temporal_exceptions.py b/osm_common/temporal_exceptions.py
index 275bc45..df730a4 100644
--- a/osm_common/temporal_exceptions.py
+++ b/osm_common/temporal_exceptions.py
@@ -13,29 +13,14 @@
# See the License for the specific language governing permissions and
# limitations under the License.
-from functools import wraps
-
from temporalio.exceptions import ApplicationError
-class RetryableException(ApplicationError):
+class NonRetryableException(ApplicationError):
def __init__(self, message):
- super().__init__(message, non_retryable=False)
+ super().__init__(message, non_retryable=True)
-def wrap_exceptions(error_message="Unhandled exception"):
- def Inner(func):
- @wraps(func)
- async def wrapper(*args, **kwargs):
- try:
- return await func(*args, **kwargs)
- except Exception as err:
- if isinstance(err, RetryableException):
- raise err
- raise ApplicationError(
- f"{error_message}: {str(err)}", non_retryable=True
- ) from err
-
- return wrapper
-
- return Inner
+class TimeOutError(ApplicationError):
+ def __init__(self, message):
+ super().__init__(message)