63 lines
2.5 KiB
Python
63 lines
2.5 KiB
Python
from django.http import HttpResponse, JsonResponse
|
|
|
|
from BaseLib.BaseView import BaseView
|
|
from Platform import settings
|
|
from experiments.models import Experiment
|
|
|
|
|
|
class ExperimentsView(BaseView):
|
|
required_login = True
|
|
endpoint = ''
|
|
view_file = 'experiments.html'
|
|
|
|
def pre_handle(self):
|
|
if 'stage' not in self.request.GET or self.request.GET['stage'] not in ['development', 'production']:
|
|
return '/experiments/?stage=production'
|
|
self.stage = self.request.GET['stage']
|
|
self.context['stage'] = self.stage
|
|
|
|
def get(self):
|
|
self.context['experiments'] = Experiment.objects.filter(project=self.request.user.selected_project,
|
|
stage=self.stage).order_by('name')
|
|
|
|
def post_create(self):
|
|
Experiment.objects.create(project=self.request.user.selected_project, stage=self.stage, name=self.request.POST['name'])
|
|
return '/experiments/?stage=' + self.stage
|
|
|
|
def post_delete(self):
|
|
Experiment.objects.get(id=self.request.POST['experiment_id']).delete()
|
|
return '/experiments/?stage=' + self.stage
|
|
def post_change(self):
|
|
exp = Experiment.objects.get(id=self.request.POST['experiment_id'])
|
|
exp.enabled = 'enabled' in self.request.POST
|
|
condition = self.request.POST['condition_select']
|
|
if condition == 'Другое (только для техлидов)':
|
|
exp.condition = self.request.POST['condition']
|
|
elif condition == 'Никому':
|
|
exp.condition = 'False'
|
|
elif condition == 'Только админам':
|
|
exp.condition = 'user.is_superuser'
|
|
elif condition == 'Только сотрудникам':
|
|
exp.condition = 'user.platform_staff'
|
|
else:
|
|
exp.condition = 'True'
|
|
exp.save()
|
|
return '/experiments/?stage=' + self.stage
|
|
|
|
|
|
def get_experiment(request):
|
|
if request.headers.get("X-Security-Token") != settings.PLATFORM_SECURITY_TOKEN:
|
|
return HttpResponse('', status=403)
|
|
project = request.GET.get('project')
|
|
stage = request.GET.get('stage')
|
|
name = request.GET.get('name')
|
|
if project is None or stage is None or name is None:
|
|
return HttpResponse('', status=400)
|
|
experiment = Experiment.objects.filter(stage=stage, project__name=project, name=name).first()
|
|
if experiment is None:
|
|
return HttpResponse('', status=404)
|
|
return JsonResponse({
|
|
'enabled': experiment.enabled,
|
|
'condition': experiment.condition
|
|
})
|