osm_common/temporal_exceptions.py

   1 # Copyright 2023 Canonical Ltd.
   2
   3 # Licensed under the Apache License, Version 2.0 (the "License");
   4 # you may not use this file except in compliance with the License.
   5 # You may obtain a copy of the License at
   6 #
   7 #    http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 # Unless required by applicable law or agreed to in writing, software
  10 # distributed under the License is distributed on an "AS IS" BASIS,
  11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
  12 # implied.
  13 # See the License for the specific language governing permissions and
  14 # limitations under the License.
  15
  16 from functools import wraps
  17
  18 from temporalio.exceptions import ApplicationError
  19
  20
  21 class RetryableException(ApplicationError):
  22     def __init__(self, message):
  23         super().__init__(message, non_retryable=False)
  24
  25
  26 def wrap_exceptions(error_message="Unhandled exception"):
  27     def Inner(func):
  28         @wraps(func)
  29         async def wrapper(*args, **kwargs):
  30             try:
  31                 return await func(*args, **kwargs)
  32             except Exception as err:
  33                 if isinstance(err, RetryableException):
  34                     raise err
  35                 raise ApplicationError(
  36                     f"{error_message}: {str(err)}", non_retryable=True
  37                 ) from err
  38
  39         return wrapper
  40
  41     return Inner