labs.py

from datetime import datetime
from django.template.defaultfilters import slugify
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 .auth import authenticate


@api_view(['GET'])
def lab(request, format=None):
    """Get or update information about a lab."""

    # Query labs.
    if request.method == 'GET':
        limit = request.query_params.get('limit', None)
        order_by = request.query_params.get('order_by', 'state')
        # TODO: Get any filters from dict(request.query_params)
        labs = get_collection('labs', order_by=order_by, limit=limit, filters=[])
        return Response({'data': labs}, content_type='application/json')


@api_view(['GET', 'POST'])
def labs(request, format=None):
    """Get or update information about labs."""

    # Query labs.
    if request.method == 'GET':
        limit = request.query_params.get('limit', None)
        order_by = request.query_params.get('order_by', 'state')
        # TODO: Get any filters from dict(request.query_params)
        labs = get_collection('labs', order_by=order_by, limit=limit, filters=[])
        return Response({'data': labs}, content_type='application/json')

    # Update a lab given a valid Firebase token.
    elif request.method == 'POST':

        # Check token.
        try:
            claims = authenticate(request)
        except Exception:  # TODO: what is the correct exception?
            return Response(
                {'error': 'Could not authenticate.'}, status=status.HTTP_400_BAD_REQUEST
            )

        # Get the posted lab data.
        lab = request.data
        org_id = lab['id']
        lab['slug'] = slugify(lab['name'])

        # TODO: Handle adding labs.
        # Create uuid, latitude, and longitude, other fields?

        # Determine any changes.
        existing_data = get_document(f'labs/{org_id}')
        changes = []
        for key, after in lab:
            before = existing_data[key]
            if before != after:
                changes.append({'key': key, 'before': before, 'after': after})

        # Get a timestamp.
        timestamp = datetime.now().isoformat()
        lab['updated_at'] = timestamp

        # Create a change log.
        log_entry = {
            'action': 'Updated lab data.',
            'type': 'change',
            'created_at': lab['updated_at'],
            'user': claims['uid'],
            'user_name': claims['display_name'],
            'user_email': claims['email'],
            'photo_url': claims['photo_url'],
            'changes': changes,
        }
        update_document(f'labs/{org_id}/logs/{timestamp}', log_entry)

        # Update the lab.
        update_document(f'labs/{org_id}', lab)

        return Response(log_entry, status=status.HTTP_201_CREATED)


@api_view(['GET', 'POST'])
def lab_logs(request, org_id, format=None):
    """Get or create lab logs."""

    if request.method == 'GET':
        data = get_collection(f'labs/{org_id}/logs')
        return Response({'data': data}, content_type='application/json')

    elif request.method == 'POST':
        # TODO: Create a log.
        return Response({'data': 'Under construction'}, content_type='application/json')


@api_view(['GET', 'POST'])
def lab_analyses(request, org_id, format=None):
    """
    Get or update (TODO) lab analyses.
    """

    if request.method == 'GET':
        data = get_collection(f'labs/{org_id}/analyses')
        return Response({'data': data}, content_type='application/json')

    elif request.method == 'POST':
        # TODO: Create an analysis.
        return Response({'data': 'Under construction'}, content_type='application/json')