def create_sample(self, name=None, db=False, **kw): if name is not None: kw['name'] = name kw.setdefault('percent', 100) sample = Sample(**kw) sample.save() if db else cache_sample(instance=sample) return sample
def test_get_samples_for_site(self): self.assertTrue(len(Sample.get_samples_for_site(self.site1)) == 0) name1 = "foo" Sample.objects.create(name=name1, percent='100.0', site=self.site1) self.assertEqual([name1], [s.name for s in Sample.get_samples_for_site(self.site1)]) # by default sample is sites-global self.assertEqual([name1], [s.name for s in Sample.get_samples_for_site(self.site2)]) name2 = "bar" Sample.objects.create(name=name2, percent='0.0', site=self.site2, all_sites_override=False) self.assertEqual({name1, name2}, set([s.name for s in Sample.get_samples_for_site(self.site2)])) self.assertEqual([name1], [s.name for s in Sample.get_samples_for_site(self.site1)])
def whoami(request): """ Return a JSON object representing the current user, either authenticated or anonymous. """ user = request.user if user.is_authenticated: data = { 'username': user.username, 'timezone': user.timezone, 'is_authenticated': True, 'is_staff': user.is_staff, 'is_superuser': user.is_superuser, 'is_beta_tester': user.is_beta_tester, 'gravatar_url': { 'small': gravatar_url(user.email, size=50), 'large': gravatar_url(user.email, size=200), } } else: data = { 'username': None, 'timezone': settings.TIME_ZONE, 'is_authenticated': False, 'is_staff': False, 'is_superuser': False, 'is_beta_tester': False, 'gravatar_url': { 'small': None, 'large': None, } } # Add waffle data to the dict we're going to be returning. # This is what the waffle.wafflejs() template tag does, but we're # doing it via an API instead of hardcoding the settings into # the HTML page. See also from waffle.views._generate_waffle_js. # # Note that if we upgrade django-waffle, version 15 introduces a # pluggable flag model, and the approved way to get all flag # objects will then become: # get_waffle_flag_model().get_all() # data['waffle'] = { 'flags': {f.name: f.is_active(request) for f in Flag.get_all()}, 'switches': {s.name: s.is_active() for s in Switch.get_all()}, 'samples': {s.name: s.is_active() for s in Sample.get_all()}, } return JsonResponse(data)
def whoami(request): """ Return a JSON object representing the current user, either authenticated or anonymous. """ user = request.user if user.is_authenticated: data = { "username": user.username, "is_authenticated": True, "avatar_url": get_avatar_url(user), "email": user.email, "subscriber_number": user.subscriber_number, } if UserSubscription.objects.filter(user=user, canceled__isnull=True).exists(): data["is_subscriber"] = True if user.is_staff: data["is_staff"] = True if user.is_superuser: data["is_superuser"] = True if user.is_beta_tester: data["is_beta_tester"] = True # This is rather temporary field. Once we're off the Wiki and into Yari # this no longer makes sense to keep. data["wiki_contributions"] = user.created_revisions.count() else: data = {} # Add waffle data to the dict we're going to be returning. # This is what the waffle.wafflejs() template tag does, but we're # doing it via an API instead of hardcoding the settings into # the HTML page. See also from waffle.views._generate_waffle_js. # # Note that if we upgrade django-waffle, version 15 introduces a # pluggable flag model, and the approved way to get all flag # objects will then become: # get_waffle_flag_model().get_all() # data["waffle"] = { "flags": {f.name: True for f in Flag.get_all() if f.is_active(request)}, "switches": {s.name: True for s in Switch.get_all() if s.is_active()}, "samples": {s.name: True for s in Sample.get_all() if s.is_active()}, } return JsonResponse(data)
def whoami(request): """ Return a JSON object representing the current user, either authenticated or anonymous. """ user = request.user if user.is_authenticated: data = { "username": user.username, "timezone": user.timezone, "is_authenticated": True, "is_staff": user.is_staff, "is_superuser": user.is_superuser, "is_beta_tester": user.is_beta_tester, "avatar_url": get_avatar_url(user), "is_subscriber": UserSubscription.objects.filter( user=user, canceled__isnull=True ).exists(), } else: data = { "username": None, "timezone": settings.TIME_ZONE, "is_authenticated": False, "is_staff": False, "is_superuser": False, "is_beta_tester": False, "avatar_url": None, "is_subscriber": False, } # Add waffle data to the dict we're going to be returning. # This is what the waffle.wafflejs() template tag does, but we're # doing it via an API instead of hardcoding the settings into # the HTML page. See also from waffle.views._generate_waffle_js. # # Note that if we upgrade django-waffle, version 15 introduces a # pluggable flag model, and the approved way to get all flag # objects will then become: # get_waffle_flag_model().get_all() # data["waffle"] = { "flags": {f.name: f.is_active(request) for f in Flag.get_all()}, "switches": {s.name: s.is_active() for s in Switch.get_all()}, "samples": {s.name: s.is_active() for s in Sample.get_all()}, } return JsonResponse(data)
def _generate_waffle_js(request): flags = get_waffle_flag_model().get_all() flag_values = [(f.name, f.is_active(request)) for f in flags] switches = Switch.get_all() switch_values = [(s.name, s.is_active()) for s in switches] samples = Sample.get_all() sample_values = [(s.name, s.is_active()) for s in samples] return loader.render_to_string('waffle/waffle.js', { 'flags': flag_values, 'switches': switch_values, 'samples': sample_values, 'flag_default': get_setting('FLAG_DEFAULT'), 'switch_default': get_setting('SWITCH_DEFAULT'), 'sample_default': get_setting('SAMPLE_DEFAULT'), })
def create_sample(self, name=None, db=False, **kw): if name is not None: kw["name"] = name kw.setdefault("percent", 100) sample = Sample(**kw) sample.save() if db else cache_sample(instance=sample)
def list(self, *args, **kwargs): return Switch.get_all() + Sample.get_all() + get_waffle_flag_model().get_all()