metagenedb.py

from requests.exceptions import HTTPError

from bioapi import MetageneDBCatalogGeneAPI
from django.urls import reverse


class MetageneDBAPIMock(MetageneDBCatalogGeneAPI):
    """
    Just a simple mock to go through the Test client. The idea is to test the upsert behaviour and not
    the insertion to the db.
    """
    KEY_ID = ''
    BASE_REVERSE = 'api'
    REVERSE_PATH = ''
    BAD_REQUESTS = range(400, 452)

    def __init__(self, client):
        self.client = client
        self.reverse_path = ':'.join([self.BASE_REVERSE, self.REVERSE_PATH])

    def get_all(self, params=None):
        response = self.client.get(reverse(f'{self.reverse_path}-list'), params)
        if response.status_code in self.BAD_REQUESTS:
            raise HTTPError
        return response.json()

    def get(self, entry_id, params=None):
        response = self.client.get(reverse(f'{self.reverse_path}-detail', kwargs={self.KEY_ID: entry_id}), params)
        if response.status_code in self.BAD_REQUESTS:
            raise HTTPError
        return response.json()

    def post(self, data):
        response = self.client.post(reverse(f'{self.reverse_path}-list'), data, format='json')
        if response.status_code in self.BAD_REQUESTS:
            raise HTTPError
        return response.json()

    def put(self, data, entry_id=None):
        if entry_id:
            return self.client.put(reverse(f'{self.reverse_path}-detail', kwargs={self.KEY_ID: entry_id}),
                                   data, format='json').json()
        return self.client.put(reverse(f'{self.reverse_path}-list'), data, format='json').json()


class MetageneDBCatalogGeneAPIMock(MetageneDBAPIMock):
    KEY_ID = 'gene_id'
    REVERSE_PATH = 'catalog:v1:genes'

    def get_gene_length(self, params=None):
        reverse_path = f"{self.reverse_path}-gene-length"
        response = self.client.get(reverse(reverse_path), params)
        if response.status_code in self.BAD_REQUESTS:
            raise HTTPError
        if response.status_code == 204:  # no content
            return {}
        return response.json()

    def get_tax_counts(self, params=None):
        reverse_path = f"{self.reverse_path}-taxonomy-counts"
        response = self.client.get(reverse(reverse_path), params)
        if response.status_code in self.BAD_REQUESTS:
            raise HTTPError
        if response.status_code == 204:  # no content
            return {}
        return response.json()


class MetageneDBCatalogTaxonomyAPIMock(MetageneDBAPIMock):
    KEY_ID = 'gene_id'
    REVERSE_PATH = 'catalog:v1:taxonomy'


class MetageneDBCatalogFunctionAPIMock(MetageneDBAPIMock):
    KEY_ID = 'function_id'
    REVERSE_PATH = 'catalog:v1:functions'