Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def method_wrapper(*args, **kwargs):
""" Wrap the method and perform error handling
:param *args:
:param **kwargs:
:return: result of the wrapped method
"""
try:
if not request.is_jsonapi and fun.__name__ not in ["get"]:
# reuire jsonapi content type for requests to these routes
raise GenericError('Unsupported Content-Type', 415)
result = fun(*args, **kwargs)
safrs.DB.session.commit()
return result
except (ValidationError, GenericError, NotFoundError) as exc:
safrs.log.exception(exc)
status_code = getattr(exc, "status_code")
message = exc.message
except werkzeug.exceptions.NotFound:
status_code = 404
message = "Not Found"
except Exception as exc:
status_code = getattr(exc, "status_code", 500)
traceback.print_exc()
safrs.log.error(exc)
if safrs.log.getEffectiveLevel() > logging.DEBUG:
def _s_clone(self, **kwargs):
"""
Clone an object: copy the parameters and create a new id
:param *kwargs: TBD
"""
make_transient(self)
# pylint: disable=attribute-defined-outside-init
self.id = self.id_type()
for parameter in self._s_jsonapi_attrs:
value = kwargs.get(parameter, None)
if value is not None:
setattr(self, parameter, value)
safrs.DB.session.add(self)
if self._s_auto_commit:
safrs.DB.session.commit()
return self
# All subclasses should have the DB.Model as superclass.
# (SQLAlchemy doesn't work when using DB.Model as SAFRSBase superclass)
try:
safrs.DB.Model.__init__(self, **db_args)
except Exception as exc:
# OOPS .. things are going bad, this might happen using sqla automap
safrs.log.error("Failed to instantiate {}".format(self))
safrs.log.debug("db args: {}".format(db_args))
safrs.log.exception(exc)
safrs.DB.Model.__init__(self)
if self._s_auto_commit:
# Add the object to the database if specified by the class parameters
safrs.DB.session.add(self)
try:
safrs.DB.session.commit()
except sqlalchemy.exc.SQLAlchemyError as exc:
# Exception may arise when a DB constrained has been violated (e.g. duplicate key)
raise GenericError(exc)
id = data.get("id")
self.SAFRSObject.id_type.get_pks(id)
# Create the object instance with the specified id and json data
# If the instance (id) already exists, it will be updated with the data
# pylint: disable=not-callable
instance = self.SAFRSObject(**attributes)
if not instance._s_auto_commit:
#
# The item has not yet been added/commited by the SAFRSBase,
# in that case we have to do it ourselves
#
safrs.DB.session.add(instance)
try:
safrs.DB.session.commit()
except sqlalchemy.exc.SQLAlchemyError as exc:
# Exception may arise when a db constrained has been violated
# (e.g. duplicate key)
safrs.log.warning(str(exc))
raise GenericError(str(exc))
return instance