class TreatmentType(models.Model):
    index = models.TextField(primary_key=True)
    code = models.IntegerField(db_column='Code', blank=True, null=True)  # Field name made lowercase.
    meaning = models.TextField(db_column='Meaning', blank=True, null=True)  # Field name made lowercase.

    class Meta:
        managed = False
        db_table = 'TreatmentType'


class Treatment(models.Model):
    index = models.TextField(primary_key=True)
    patient = models.IntegerField(db_column='PATIENT', blank=True, null=True)  # Field name made lowercase.
    date = models.TextField(db_column='DATE', blank=True, null=True)  # Field name made lowercase. This field type is a guess.
    treatmenttype = models.IntegerField(db_column='TREATTYPE', blank=True, null=True)  # Field name made lowercase.

    class Meta:
        managed = False
        db_table = 'Treatment'

class TreatmentModelSerializer(serializers.ModelSerializer):
    meaning = serializers.SerializerMethodField('get_meaning')

    def get_meaning(self, instance):
        meaning = TreatmentType.objects.get(code=instance.treatmenttype).meaning
        return meaning

    class Meta:
        model = Treatment
        fields = [
            'patient',
            'date',
            'meaning',
        ]