leaf_test_api.py

# from rest_framework import status
from rest_framework.decorators import api_view
from rest_framework.response import Response

# from utils.firebase import get_collection, get_document, update_document
from utils.firebase import get_collection


@api_view(['GET', 'POST'])
def lab_results(request, format=None):
    """Get lab results data."""

    if request.method == 'GET':
        limit = request.query_params.get('limit', 1000)
        if limit:
            limit = int(limit)
        order_by = request.query_params.get('order_by', '')
        # TODO: Get any filters from dict(request.query_params)
        docs = get_collection(
            'tests/leaf/lab_results', order_by=order_by, limit=limit, filters=[]
        )
        return Response(docs, content_type='application/json')

    if request.method == 'POST':
        print('TODO: Create lab results')
        return NotImplementedError


@api_view(['GET'])
def mmes(request, format=None):
    """Get licensee (MME) data."""

    if request.method == 'GET':
        limit = request.query_params.get('limit', None)
        if limit:
            limit = int(limit)
        order_by = request.query_params.get('order_by', '')
        # TODO: Get any filters from dict(request.query_params)
        # e.g. {'key': 'name', 'operation': '==', 'value': 'xyz'}
        docs = get_collection(
            'tests/leaf/mmes', order_by=order_by, limit=limit, filters=[]
        )
        return Response(docs, content_type='application/json')