import xml.dom.minidom
import time

from v3Helpers import *

__all__ = [
   'hl7version3'
]

def v3Header():
   value = '<?xml version="1.0" encoding="UTF-8"?>\n'
   value += '<!--\n'
   value += 'This example message is provided for illustrative purposes only. It has had no clinical validation. Whilst every effort has been taken to ensure that the examples are consistent with the\n'
   value += ' message specification contained within the MiM, where there are conflicts with the written message specification or schema, the specification or schema shall be considered to take\n'
   value += 'precedence\n'
   value += '-->\n'
   return value

def v3LabRequestHeader(EffectiveTime):
	value = '<LaboratoryRequestGroup classCode="ACT" moodCode="RQO" xmlns="urn:hl7-org:v3" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="urn:hl7-org:v3 ..\..\schemas\POLB_MT001000UK01.xsd">\n'
	value += '	<id root="9921D007-89C1-4983-83E3-13E4BAEF1B79"/>\n'
	value += '	<code codeSystem="2.16.840.1.113883.2.1.3.2.4.15" code="193831000000104" displayName="Pathology Request Event"/>\n'
	value += '	<statusCode code="active"/>\n'
	value += '	<effectiveTime value="' + EffectiveTime + '"/>\n'
	value += '	<priorityCode codeSystem="2.16.840.1.113883.2.1.3.2.4.17.95" code="1" displayName="Routine"/>\n'
	value += '	<confidentialityCode codeSystem="2.16.840.1.113883.2.1.3.2.4.17.249" code="L" displayName="Limited sharing permitted"/>\n'
	return value

def v3Name(GivenName, FamilyName, i):
   value = v3Indent(i) + '<name>\n'
   value += v3Indent(i + 1) + '<given>' + esc(GivenName) + '</given>\n'
   value += v3Indent(i + 1) + '<family>' + esc(FamilyName) + '</family>\n'
   value += v3Indent(i) + '</name>\n'
   return value

def v3Address(Type, Line1, Line2, Line3, PostalCode, i):
   value = v3Indent(i) + '<addr use="' + esc(Type) + '">\n'
   value += v3Indent(i+1) + '<streetAddressLine>' + esc(Line1) + '</streetAddressLine>\n'
   value += v3Indent(i+1) + '<streetAddressLine>' + esc(Line2) + '</streetAddressLine>\n'
   value += v3Indent(i+1) + '<streetAddressLine>' + esc(Line3) + '</streetAddressLine>\n'
   value += v3Indent(i+1) + '<postalCode>' + esc(PostalCode) + '</postalCode>\n'
   value += v3Indent(i) + '</addr>\n'
   return value

def v3Sex(Sex):
	if Sex=='M':
		return '0';
	else:
		return '1';

def v3Doctor(Doctor, i):
	value = v3Indent(i) + '<performer typeCode="PRF">\n'
	value += v3Indent(i+1) + '<functionCode codeSystem="2.16.840.1.113883.5.88" code="PCP"/>\n'
	value += v3Indent(i+1) + '<assignedProvider classCode="ASSIGNED">\n'
	value += v3Indent(i+2) + '<id root="1.2.826.0.1285.0.2.0.67" extension="151854257025"/>\n'
	value += v3Indent(i+2) + '<id root="1.2.826.0.1285.0.2.0.65" extension="910493857097"/>\n'
	value += v3Indent(i+2) + '<code nullFlavor="NA"/>\n'
	value += v3Indent(i+2) + '<assignedPersonProvider determinerCode="INSTANCE" classCode="PSN">\n'
	value += v3Indent(i+3) + '<name>' + esc(Doctor) + '</name>\n'
	value += v3Indent(i+3) + '<asHealthCareProvider classCode="PROV"/>\n'
	value += v3Indent(i+2) + '</assignedPersonProvider>\n'
	value += v3Indent(i+1) + '</assignedProvider>\n'
	value += v3Indent(i) + '</performer>\n'
	return value


def v3PatientRecordTarget(InMsg, PatientIndex, i):
	value = v3Indent(i) + '<recordTarget typeCode="RCT" contextControlCode="OP">\n'
	value += v3Indent(i+1) + '<patient classCode="PAT">\n' 
	value += v3Indent(i+2) + '<id root="2.16.840.1.113883.2.1.4.1" extension="99000002768"/>\n'
	value += v3Indent(i+2) + '<statusCode code="active"/>\n'
	value += v3Indent(i+2) + '<patientPerson determinerCode="INSTANCE" classCode="PSN">\n'
	value += v3Name(getItem(InMsg, "Patient", PatientIndex, "PatientFirstName"), getItem(InMsg, "Patient", PatientIndex, "PatientLastName"), i+3)
	value += v3Indent(i+3) + '<administrativeGenderCode codeSystem="2.16.840.1.113883.2.1.3.2.4.16.25" code="' + v3Sex(getItem(InMsg, "Patient", PatientIndex, "PatientSex")) + '"/>\n'
	value += v3Indent(i+3) + '<birthTime value="'+ getItem(InMsg, "Patient", PatientIndex, "PatientBirth") + '"/>\n'
	value += v3Address("H", getItem(InMsg, "Patient", PatientIndex, "PatientStreetAddress"), getItem(InMsg, "Patient", PatientIndex, "PatientAddressOtherDesignation"), getItem(InMsg, "Patient", PatientIndex, "PatientAddressCity"), getItem(InMsg, "Patient", PatientIndex, "PatientAddressZip"), i+3)
	value += v3Indent(i+3) + '<asPatient classCode="PAT">\n'
	value += v3Indent(i+4) + '<subjectOf typeCode="SBJ">\n'
	value += v3Indent(i+5) + '<patientCareProvision moodCode="EVN" classCode="PCPR">\n'
	value += v3Indent(i+6) + '<code codeSystem="2.16.840.1.113883.2.1.3.2.4.16.43" code="01"/>\n'
	value += v3Doctor("Dr Responsible", i+6) # TODO - need to map in a real doctor. 
	value += v3Indent(i+5) + '</patientCareProvision>\n'
	value += v3Indent(i+4) + '</subjectOf>\n'
	value += v3Indent(i+3) + '</asPatient>\n'
	value += v3Indent(i+2) + '</patientPerson>\n'
	value += v3Indent(i+1) + '</patient>\n'
	value += v3Indent(i) +   '</recordTarget>\n'
	return value

def v3HospitalTestProvider(HospitalName,i):
	value  = v3Indent(i)   + '<performer contextControlCode="ON" typeCode="PRF">\n'
	value += v3Indent(i+1) + '<assignedOrgSDS classCode="ASSIGNED">\n'
	value += v3Indent(i+2) + '<id root="1.2.826.0.1285.0.1.10" extension="RBL07"/>\n'
	value += v3Indent(i+2) + '<assignedOrganizationSDS classCode="ORG" determinerCode="INSTANCE">\n'
	value += v3Indent(i+3) + '<name>' + esc(HospitalName) + '</name>\n'
	value += v3Indent(i+2) + '</assignedOrganizationSDS>\n'
	value += v3Indent(i+1) + '</assignedOrgSDS>\n'
	value += v3Indent(i)   + '</performer>\n'
	return value

def v3DoctorAuthor(DoctorName, i):
	value  = v3Indent(i)   + '<author contextControlCode="OP" typeCode="AUT">\n'
	value += v3Indent(i+1) + '<time value="20050822105532"/>\n'
	value += v3Indent(i+1) + '<assignedPersonSDS classCode="ASSIGNED">\n'
	value += v3Indent(i+2) + '<id root="1.2.826.0.1285.0.2.0.67" extension="151854257025"/>\n'
	value += v3Indent(i+2) + '<id root="1.2.826.0.1285.0.2.0.65" extension="910493857097"/>\n'
	value += v3Indent(i+2) + '<code nullFlavor="NA"/>\n'
	value += v3Indent(i+2) + '<assignedPersonSDS determinerCode="INSTANCE" classCode="PSN">\n'
	value += v3Indent(i+3) + '<name>' + esc(DoctorName) + '</name>\n'
	value += v3Indent(i+2) + '</assignedPersonSDS>\n'
	value += v3Indent(i+1) + '</assignedPersonSDS>\n'
	value += v3Indent(i)   + '</author>\n'
	return value

def v3HospitalReceiver(Hospital, i):
	value = v3Indent(i)   + '<primaryInformationRecipient typeCode="PRCP">\n'
	value += v3Indent(i+1) + '<assignedOrgSDS classCode="ASSIGNED">\n'
	value += v3Indent(i+2) + '<id root="1.2.826.0.1285.0.1.10" extension="M85008"/>\n'
	value += v3Indent(i+2) + '<assignedOrganizationSDS classCode="ORG" determinerCode="INSTANCE">\n'
	value += v3Indent(i+3) + '<name>' + esc(Hospital) + '</name>\n'
	value += v3Indent(i+2) + '</assignedOrganizationSDS>\n'
	value += v3Indent(i+1) + '</assignedOrgSDS>\n'
	value += v3Indent(i)   + '</primaryInformationRecipient>\n'
	return value

def v3MedicationReason(InMsg, i):
	value = v3Indent(i)   + '<reason typeCode="RSON" contextConductionInd="true" contextControlCode="AN">\n'
	value += v3Indent(i+1) + '<givenMedication classCode="SBADM" moodCode="EVN">\n'
	value += v3Indent(i+2) + '<id root="13695EDE-AC85-4257-9D0D-1254C80E5925"/>\n'
	value += v3Indent(i+2) + '<code code="225426007" codeSystem="2.16.840.1.113883.2.1.3.2.4.15"/>\n'
	value += v3Indent(i+2) + '<statusCode code="active"/>\n'
	value += v3Indent(i+2) + '<effectiveTime>\n'
	# TODO - need to figure out what the effective time should be mapped from
	value += v3Indent(i+3) + '<low value="20071001"/>\n'
	value += v3Indent(i+2) + '</effectiveTime>\n'
	value += v3Indent(i+2) + '<consumable typeCode="CSM" contextControlCode="AN">\n'
	value += v3Indent(i+3) + '<manufacturedProduct classCode="ADMM">\n'
	value += v3Indent(i+4) + '<manufacturedMaterialKind classCode="MAT" determinerCode="KIND">\n'
	# TODO - need to map in a code for the drug here 
	value += v3Indent(i+5) + '<code code="371485002" codeSystem="2.16.840.1.113883.2.1.3.2.4.15" displayName="Insulin soluble human 100units/ml solution for injection 1.5ml cartridges"/>\n'
	value += v3Indent(i+4) + '</manufacturedMaterialKind>\n'
	value += v3Indent(i+3) + '</manufacturedProduct>\n'
	value += v3Indent(i+2) + '</consumable>\n'
	value += v3Indent(i+2) + '<component4 contextConductionInd="true" typeCode="COMP">\n'
	value += v3Indent(i+3) + '<seperatableInd value="false"/>\n'
	value += v3Indent(i+3) + '<administrationTypeCourseOrSingleDose classCode="OBS" moodCode="EVN">\n'
	# TODO - codes can go in here to describe how the drug should be taken.
	value += v3Indent(i+4) + '<code code="AT" codeSystem="2.16.840.1.113883.2.1.3.2.4.17.30" displayName="Medication administration type"/>\n'
	value += v3Indent(i+4) + '<value code="CO" codeSystem="2.16.840.1.113883.2.1.3.2.4.17.106" displayName="administration of a complete course"/>\n'
	value += v3Indent(i+3) + '</administrationTypeCourseOrSingleDose>\n'
	value += v3Indent(i+2) + '</component4>\n'
	value += v3Indent(i+1) + '</givenMedication>\n'
	value += v3Indent(i)   + '</reason>\n'
  	return value

def v3LabRequestFooter():
   return '</LaboratoryRequestGroup>\n';

def hl7version3(PlainLabDataXml):
	InMsg = xml.dom.minidom.parseString(PlainLabDataXml)
	v3 = v3Header()
	v3 += v3LabRequestHeader(time.strftime("%Y%m%d"))
	v3 += v3PatientRecordTarget(InMsg, 0, 1)  # TODO - we assume exactly 1 patient
	v3 += v3HospitalTestProvider("The Performing Organisation",1)
	v3 += v3DoctorAuthor("Dr Responsible", 1)
	v3 += v3HospitalReceiver("The Requesting Organization", 1)
	v3 += v3MedicationReason(InMsg, 1)
	v3 += v3LabRequestFooter()
	return v3