def test_key_error_on_update(self, bad_mock):
""" Test KeyError on update """
bad_mock.side_effect = KeyError()
pet = Pet("fido", "dog", False)
pet.save()
pet.name = 'Fifi'
pet.update()
def handle(self, *args, **options):
if Vaccine.objects.exists() or Pet.objects.exists():
print("[x] Data ALready Loaded ")
print(ALREDY_LOADED_ERROR_MESSAGE)
return
print("[+] Creating Vacine Data ")
for vaccine_name in VACCINES_NAMES:
vac = Vaccine(name=vaccine_name)
vac.save()
print("[+] Loading pet Data for adoption")
for row in DictReader(open("./pet_data.csv")):
pet = Pet()
pet.name = row['Pet']
pet.submitter = row['Submitter']
pet.species = row['Species']
pet.breed = row['Breed']
pet.description = row['Pet Description']
pet.sex = row['Sex']
pet.age = row['Age']
raw_submission_date = row['submission date']
submission_date = UTC.localize(
datetime.strptime(raw_submission_date, DATETIME_FORMAT))
pet.submission_date = submission_date
pet.save() ## saving data
raw_vaccination_names = row['vaccinations']
## check if the vaccine in the avilabe list
## if it is then save it
## dont allow other vaccination outside the vaccination list
vaccination_names = [
name for name in raw_vaccination_names.split('| ') if name
]
for vac_name in vaccination_names:
vac = Vaccine.objects.get(name=vac_name)
pet.vaccinations.add(vac)
pet.save()
