def save(self, *args, **kwargs): """Generates the number of bonus point codes.""" _ = args _ = kwargs num = self.cleaned_data.get("num_codes") p = self.cleaned_data.get("point_value") # Generate if num > 0: BonusPoint.generate_bonus_points(p, num)
def generate_codes(request): """Handles the generate_codes_form from and creates the BonusPoints.""" if request.method == "POST": form = GenerateBonusPointsForm(request.POST) if form.is_valid(): points = form.cleaned_data['point_value'] num = form.cleaned_data['num_codes'] BonusPoint.generate_bonus_points(points, num) response = HttpResponseRedirect("/admin/bonus_points/bonuspoint/") return response raise Http404
def testViewBonusPoint(self): """Test view bonus points.""" BonusPoint.generate_bonus_points(5, 5) BonusPoint.generate_bonus_points(10, 2) BonusPoint.generate_bonus_points(20, 3) response = self.client.get(reverse('bonus_view_codes')) self.failUnlessEqual(response.status_code, 404) self.user.is_staff = True self.user.save() response = self.client.get(reverse('bonus_view_codes')) self.assertEqual(response.status_code, 200) self.assertTemplateUsed(response, 'view_bonus_points.html')