NextCloud.response

Complexity

Total Complexity

12

Size/Duplication

Total Lines	60
Duplicated Lines	0 %

Importance

Changes

0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	WebDAVResponse.__repr__()	0	3	2
A	OCSResponse.__init__()	0	13	3
A	NextCloudResponse.__init__()	0	6	3
A	WebDAVResponse.__init__()	0	6	2
A	OCSResponse.__repr__()	0	3	2

from .api_wrappers.webdav import WebDAVStatusCodes


class NextCloudResponse(object):

    def __init__(self, response, json_output=True, data=None):
        self.raw = response
        if not data:
            self.data = response.json() if json_output else response.content.decode("UTF-8")
        else:
            self.data = data


class WebDAVResponse(NextCloudResponse):
    """ Response class for WebDAV api methods """

    METHODS_SUCCESS_CODES = {
        "PROPFIND": [WebDAVStatusCodes.MULTISTATUS_CODE],
        "PROPPATCH": [WebDAVStatusCodes.MULTISTATUS_CODE],
        "REPORT": [WebDAVStatusCodes.MULTISTATUS_CODE],
        "MKCOL": [WebDAVStatusCodes.CREATED_CODE],
        "COPY": [WebDAVStatusCodes.CREATED_CODE, WebDAVStatusCodes.NO_CONTENT_CODE],
        "MOVE": [WebDAVStatusCodes.CREATED_CODE, WebDAVStatusCodes.NO_CONTENT_CODE],
        "PUT": [WebDAVStatusCodes.CREATED_CODE],
        "DELETE": [WebDAVStatusCodes.NO_CONTENT_CODE]
    }

    def __init__(self, response, data=None):
        super(WebDAVResponse, self).__init__(response=response, data=data, json_output=False)
        request_method = response.request.method
        self.is_ok = False
        if request_method in self.METHODS_SUCCESS_CODES:
            self.is_ok = response.status_code in self.METHODS_SUCCESS_CODES[request_method]

    def __repr__(self):
        is_ok_str = "OK" if self.is_ok else "Failed"
        return "<OCSResponse: Status: {}>".format(is_ok_str)


class OCSResponse(NextCloudResponse):
    """ Response class for OCS api methods """

    def __init__(self, response, json_output=True, success_code=None):
        self.raw = response
        self.is_ok = None
        if json_output:
            self.full_data = response.json()
            self.meta = self.full_data['ocs']['meta']
            self.status_code = self.full_data['ocs']['meta']['statuscode']
            self.data = self.full_data['ocs']['data']
            if success_code:
                self.is_ok = self.full_data['ocs']['meta']['statuscode'] == success_code

        else:
            self.data = response.content.decode("UTF-8")

    def __repr__(self):
        is_ok_str = "OK" if self.is_ok else "Failed"
        return "<OCSResponse: Status: {}>".format(is_ok_str)