Missing Patient class

[svenvanderburg]

trafficstars

The extras episodes on persistence and databases assume a Patient class is in the inflammation project. This class was removed with the new section 3.

The test test_patient.py refers to a models.Patient class but it does not exist in models.py.

We could easily fix this by adding the Patient class back to models.py:

class Observation:
    def __init__(self, day, value):
        self.day = day
        self.value = value

    def __str__(self):
        return str(self.value)

class Patient:
    """A patient in an inflammation study."""
    def __init__(self, name):
        self.name = name
        self.observations = []

    def add_observation(self, value, day=None):
        if day is None:
            try:
                day = self.observations[-1].day + 1

            except IndexError:
                day = 0

        new_observation = Observation(day, value)

        self.observations.append(new_observation)
        return new_observation

    def __str__(self):
        return self.name