def save_tag(form_field): if str(form_field): try: tag = Tag.objects.get(name=form_field) #@UndefinedVariable except Tag.DoesNotExist: #@UndefinedVariable tag = Tag(name=form_field, poll_count=0) tag.save() return tag return None
def create_users(num_users): for x in range(num_users): random_num = str(random.randint(0, sys.maxint)) username = '******' + random_num email = random_num + '@RandomUser.com' password = '******' user = User.objects.create_user(username, email, password) tag_name = 'B' + str(random.randint(0, 15)) try: tag = Tag.objects.get(name=tag_name) #@UndefinedVariable except Tag.DoesNotExist: #@UndefinedVariable tag = Tag(name=tag_name, poll_count=0) tag.save() user_profile = UserProfile(user=user) user_profile.save() user_profile.education.add(tag)
def add_tag(request): id = request.GET.get("id", None) tag_name = request.GET.get("t", None) p = Poll.objects.get(id=id) #@UndefinedVariable # make sure user owns polls if p.user != request.user: return HttpResponseForbidden("unauthorized") # get existing poll tags tags = p.tags.all() vars = {} vars["tags"] = tags tag_name = tag_name.strip() if tag_name == "": vars["error"] = "Please provide a valid tag" return render_to_response("ajax/list_poll_tags.html", vars) tag_names = tag_name.split(",") for tag_name in tag_names: tag_name = tag_name.strip() try: tag = Tag.objects.get(name=tag_name) #@UndefinedVariable except Tag.DoesNotExist: #@UndefinedVariable tag = Tag() tag.name = tag_name #increment the count if tag.poll_count: tag.poll_count += 1 else: tag.poll_count = 1 tag.save() if p.tags.filter(name=tag_name).count() < 1: p.tags.add(tag) p.save() else: # poll already has that tag # silently fail pass vars["poll"] = p return render_to_response("ajax/list_poll_tags.html", vars)