def wizard_by_id(wizard_id): service = WizardService(WizardRepository()) wizard = service.get_wizard_by_id(wizard_id) return render_template('wizard/wizard.html', wizard=wizard) if wizard else render_template( 'wizard/wizard_not_found.html')
def delete_wizard(): service = WizardService(WizardRepository()) wizards = service.get_all_wizards() if request.method == 'POST': service.delete_wizard(request.form['wizards_id']) return render_template('wizard/wizards.html', wizards=wizards) return render_template('wizard/delete_wizard.html', wizards=wizards)
def test_wizard_is_created_with_right_parameters(self): w = Wizard(1, 'Harry', 17, 1) has_received_letter = 1 wont_turn_into_dark_wizard = 0 is_not_voldemort_friend = 0 b = BaseWizardRepository() service = WizardService(b) create_wizard = service.create_wizard(w.name, w.age, has_received_letter, wont_turn_into_dark_wizard, is_not_voldemort_friend, w.house_id) self.assertTrue(create_wizard)
def test_cannot_register_if_is_friend_with_voldemort(self): w = Wizard(1, 'Harry', 15, 1) has_received_letter = 1 wont_turn_into_dark_wizard = 0 is_not_voldemort_friend = 1 b = BaseWizardRepository() service = WizardService(b) create_wizard = service.create_wizard(w.name, w.age, has_received_letter, wont_turn_into_dark_wizard, is_not_voldemort_friend, w.house_id) self.assertFalse(create_wizard)
def create_wizard(): h_service = HouseService(HouseRepository()) houses = h_service.get_all_houses() if request.method == 'POST': service = WizardService(WizardRepository()) new_wizard = service.create_wizard( request.form['wizard_name'], request.form['wizard_age'], request.form['has_received_letter'], request.form['voldemort_friend'], request.form['dark_wizard'], request.form['house_options'], ) if new_wizard: return render_template('wizard/wizard_success.html') return render_template('wizard/wizard_error.html') return render_template('wizard/new_wizard.html', houses=houses)
def update_wizard(wizard_id): w_service = WizardService(WizardRepository()) h_service = HouseService(HouseRepository()) houses = h_service.get_all_houses() if request.method == 'POST': can_update_wizard = w_service.update_wizard( request.form['wizard_id'], request.form['wizard_name'], request.form['wizard_age'], request.form['wizard_house']) if not can_update_wizard: return render_template('wizard/wizard_error.html') wizard = w_service.get_wizard_by_id(wizard_id) current_house = h_service.get_house_by_id(wizard.house_id) if wizard is not None: return render_template('wizard/update_wizard.html', wizard=wizard, houses=houses, current_house=current_house) else: return render_template('wizard/wizard_not_found.html')
def wizards(): service = WizardService(WizardRepository()) wizards = service.get_all_wizards() return render_template('wizard/wizards.html', wizards=wizards)