renalreg/radar

import logging

from radar.models.medications import Medication, MEDICATION_DOSE_UNITS, MEDICATION_ROUTES
from radar.models.source_types import SOURCE_TYPE_MANUAL
from radar.utils import date_to_datetime


logger = logging.getLogger(__name__)


def export_medications(sda_container, patient, group):
    q = Medication.query
    q = q.filter(Medication.patient == patient)
    q = q.filter(Medication.source_group == group)
    q = q.filter(Medication.source_type == SOURCE_TYPE_MANUAL)
    medications = q.all()

    if not medications:
        return

    sda_medications = sda_container.setdefault('medications', list())

    for medication in medications:
        sda_medication = dict()
        sda_medication['external_id'] = str(medication.id)
        sda_medication['from_time'] = date_to_datetime(
            medication.from_date,
            with_timezone=False
        )

        if medication.to_date:
            sda_medication['to_time'] = date_to_datetime(
                medication.to_date,
                with_timezone=False
            )

        if medication.drug_text:
            sda_medication['drug_product'] = {
                'product_name': medication.drug_text
            }

            sda_medication['order_item'] = {
                'code': medication.drug_text,
                'description': medication.drug_text
            }
        elif medication.drug:
            sda_medication['drug_product'] = {
                'product_name': medication.drug.name
            }

            sda_medication['order_item'] = {
                'code': medication.drug.name,
                'description': medication.drug.name
            }
        else:
            continue

        if medication.dose_text:
            sda_medication['dose_uom'] = {
                'code': medication.dose_text,
                'description': medication.dose_text
            }
        else:
            if medication.dose_quantity is not None:
                sda_medication['dose_quantity'] = medication.dose_quantity

            if medication.dose_unit:
                code = medication.dose_unit
                description = MEDICATION_DOSE_UNITS.get(code)

                if description:
                    sda_medication['dose_uom'] = {
                        'sda_coding_standard': 'RADAR',
                        'code': code,
                        'description': description,
                    }
                else:
                    logger.error('Unknown dose unit code={}'.format(code))

            if medication.route:
                code = medication.route
                description = MEDICATION_ROUTES.get(code)

                if description:
                    sda_medication['route'] = {
                        'sda_coding_standard': 'RADAR',
                        'code': code,
                        'description': description,
                    }
                else:
                    logger.error('Unknown route code={}'.format(code))

            if medication.frequency:
                sda_medication['frequency'] = {
                    'code': medication.frequency,
                    'description': medication.frequency,
                }

        sda_medication['entering_organization'] = {
            'code': medication.source_group.code,
            'description': medication.source_group.name
        }

        sda_medication['entered_at'] = {
            'code': 'RADAR',
            'description': 'RaDaR'
        }

        sda_medications.append(sda_medication)