base

Complexity

Total Complexity

11

Size/Duplication

Total Lines	90
Duplicated Lines	0 %

Importance

Changes

0

7 Methods

Rating	Name	Duplication	Size	Complexity
A	Resource._post()	0	13	3
A	Resource._prepare_url()	0	6	3
A	Resource._get_headers()	0	4	1
A	Resource._delete()	0	4	1
A	Resource._put()	0	4	1
A	Resource.__init__()	0	2	1
A	Resource._get()	0	4	1

import json
import requests
from enum import Enum

# =======================================================
# Parameter enums
# =======================================================
class BasicOptions(Enum):
    TRUE  = "true"
    FALSE = "false"
    ANY   = "any"


class AssignedUserOptions(Enum):
    ME   = "me"
    ANY  = "any"
    NONE = "none"


class AssignedTeamOptions(Enum):
    MINE = "mine"
    NONE = "none"


class SortByOptions(Enum):
    LAST_ACTIVITY = "last_activity"
    CREATION_TIME = "creation_time"


class UserRoles:
    ADMIN        = ("admin",        20)
    AGENT        = ("agent",        10)
    COLLABORATOR = ("collaborator", 9)
    CUSTOMER     = ("customer",     None)


class DataPointType(Enum):
    AVG_FIRST_RESPONSE_TIME = "avg_first_response_time"
    TICKETS_COUNT = "tickets_count"
    REPLIES_COUNT = "replies_count"

# =======================================================
# Resource - Base class with default requests
# =======================================================
class Resource:
    def __init__(self, api):
        self.api = api

    def _prepare_url(self, endpoint, **kwargs):
        url = self.api.BASE_URL.format(url=endpoint)
        for key, value in kwargs.items():
            if value is not None:
                url += "&" + key + "=" + str(value)
        return url

    def _get_headers(self):
        return {
            "Content-Type": "application/json",
            "Accept": "application/json"
        }

    def _get(self, endpoint, **kwargs):
        url = self._prepare_url(endpoint, **kwargs)
        print("GET", url)
        return requests.get(url, headers=self._get_headers()).json()

    def _post(self, endpoint, data=None, files=None, **kwargs):
        url = self._prepare_url(endpoint, **kwargs)
        print("POST", url)
        headers = self._get_headers()
        if files is not None: 
            headers.pop("Content-Type")
        response = requests.post(url, headers=headers, json=data, files=files)
        # print(response)
        # print(response.text)
        try:
            return response.json()
        except json.decoder.JSONDecodeError:
            return response.text

    def _put(self, endpoint, data=None, **kwargs):
        url = self._prepare_url(endpoint, **kwargs)
        print("PUT", url)
        return requests.put(url, headers=self._get_headers(), json=data).json()

    def _delete(self, endpoint, **kwargs):
        url = self._prepare_url(endpoint, **kwargs)
        print("DELETE", url)
        return requests.delete(url, headers=self._get_headers()).json()