def save_subjects(bill, subjects): #TODO fix duplicate subjects for subject_name in subjects: subject = Subject.objects.filter(name=subject_name) # If this subject is not in the db, add her/him if not subject: subject = Subject() subject.name = subject_name subject.save() # Associate this subject with imported bill subject.bills.add(bill) subject.save()
def create(subject_name, bill_id): subject = Subject.objects.filter(name = subject_name) if subject: subject[0].bills.add(Bill.objects.get(id = bill_id)) return subject[0] else: subject = Subject() subject.name = subject_name subject.save() subject.bills.add(Bill.objects.get(id = bill_id)) subject.save() return subject