# -*- coding: utf-8 -*-

#

# Copyright (C) 2013-2017 Gauvain Pocentek <gauvain@pocentek.net>

#

# This program is free software: you can redistribute it and/or modify

# it under the terms of the GNU Lesser General Public License as published by

# the Free Software Foundation, either version 3 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

# GNU Lesser General Public License for more details.

#

# You should have received a copy of the GNU Lesser General Public License

# along with this program. If not, see <http://www.gnu.org/licenses/>.

import functools

class GitlabError(Exception):

def __init__(self, error_message="", response_code=None,

response_body=None):

Exception.__init__(self, error_message)

# Http status code

self.response_code = response_code

# Full http response

self.response_body = response_body

# Parsed error message from gitlab

self.error_message = error_message

def __str__(self):

if self.response_code is not None:

return "{0}: {1}".format(self.response_code, self.error_message)

else:

return "{0}".format(self.error_message)

class GitlabAuthenticationError(GitlabError):

pass

class GitlabParsingError(GitlabError):

pass

class GitlabConnectionError(GitlabError):

pass

class GitlabOperationError(GitlabError):

pass

class GitlabHttpError(GitlabError):

pass

class GitlabListError(GitlabOperationError):

pass

class GitlabGetError(GitlabOperationError):

pass

class GitlabCreateError(GitlabOperationError):

pass

class GitlabUpdateError(GitlabOperationError):

pass

class GitlabDeleteError(GitlabOperationError):

pass

class GitlabSetError(GitlabOperationError):

pass

class GitlabProtectError(GitlabOperationError):

pass

class GitlabTransferProjectError(GitlabOperationError):

pass

class GitlabProjectDeployKeyError(GitlabOperationError):

pass

class GitlabCancelError(GitlabOperationError):

pass

class GitlabPipelineCancelError(GitlabCancelError):

pass

class GitlabRetryError(GitlabOperationError):

pass

class GitlabBuildCancelError(GitlabCancelError):

pass

class GitlabBuildRetryError(GitlabRetryError):

pass

class GitlabBuildPlayError(GitlabRetryError):

pass

class GitlabBuildEraseError(GitlabRetryError):

pass

class GitlabJobCancelError(GitlabCancelError):

pass

class GitlabJobRetryError(GitlabRetryError):

pass

class GitlabJobPlayError(GitlabRetryError):

pass

class GitlabJobEraseError(GitlabRetryError):

pass

class GitlabPipelineRetryError(GitlabRetryError):

pass

class GitlabBlockError(GitlabOperationError):

pass

class GitlabUnblockError(GitlabOperationError):

pass

class GitlabSubscribeError(GitlabOperationError):

pass

class GitlabUnsubscribeError(GitlabOperationError):

pass

class GitlabMRForbiddenError(GitlabOperationError):

pass

class GitlabMRClosedError(GitlabOperationError):

pass

class GitlabMROnBuildSuccessError(GitlabOperationError):

pass

class GitlabTodoError(GitlabOperationError):

pass

class GitlabTimeTrackingError(GitlabOperationError):

pass

class GitlabUploadError(GitlabOperationError):

pass

class GitlabAttachFileError(GitlabOperationError):

pass

class GitlabCherryPickError(GitlabOperationError):

pass

class GitlabHousekeepingError(GitlabOperationError):

pass

class GitlabOwnershipError(GitlabOperationError):

pass

class GitlabSearchError(GitlabOperationError):

pass

class GitlabStopError(GitlabOperationError):

pass

class GitlabMarkdownError(GitlabOperationError):

pass

class GitlabVerifyError(GitlabOperationError):

pass

class GitlabRenderError(GitlabOperationError):

pass

class GitlabRepairError(GitlabOperationError):

pass

class GitlabLicenseError(GitlabOperationError):

pass

def on_http_error(error):

"""Manage GitlabHttpError exceptions.

This decorator function can be used to catch GitlabHttpError exceptions

raise specialized exceptions instead.

Args:

error(Exception): The exception type to raise -- must inherit from

GitlabError

"""

def wrap(f):

@functools.wraps(f)

def wrapped_f(*args, **kwargs):

try:

return f(*args, **kwargs)

except GitlabHttpError as e:

raise error(e.error_message, e.response_code, e.response_body)

return wrapped_f

return wrap