|
|
|
|
|
import json
|
|
|
|
|
|
from django.contrib import messages
|
|
|
from django.utils.safestring import mark_safe
|
|
|
from django.shortcuts import render, redirect, get_object_or_404, HttpResponse
|
|
|
from django.contrib.auth.decorators import login_required
|
|
|
|
|
|
from apps.main.models import Experiment, Device
|
|
|
from apps.main.views import sidebar
|
|
|
|
|
|
from .models import GeneratorConfiguration
|
|
|
from .forms import GeneratorConfigurationForm, GeneratorImportForm
|
|
|
|
|
|
def is_developer(user):
|
|
|
groups = [str(g.name) for g in user.groups.all()]
|
|
|
#return 'Developer' in groups or user.is_staff
|
|
|
return 'Developer' in groups or user.is_superuser
|
|
|
|
|
|
|
|
|
def is_operator(user):
|
|
|
groups = [str(g.name) for g in user.groups.all()]
|
|
|
#return 'Operator' in groups or user.is_staff
|
|
|
return 'Operator' in groups or user.is_superuser
|
|
|
|
|
|
|
|
|
def conf(request, conf_id):
|
|
|
|
|
|
conf = get_object_or_404(GeneratorConfiguration, pk=conf_id)
|
|
|
|
|
|
kwargs = {}
|
|
|
kwargs['dev_conf'] = conf
|
|
|
|
|
|
if conf.enable_2:
|
|
|
kwargs['dev_conf_keys'] = ['delay', 'periode_1', 'width_1', 'ntx_1', 'periode_2', 'width_2', 'ntx_2', 'selector']
|
|
|
else:
|
|
|
kwargs['dev_conf_keys'] = ['delay', 'periode_1', 'width_1', 'ntx_1', 'selector']
|
|
|
|
|
|
kwargs['title'] = 'Configuration'
|
|
|
kwargs['suptitle'] = 'Detail'
|
|
|
|
|
|
kwargs['button'] = 'Edit Configuration'
|
|
|
|
|
|
conf.status_device()
|
|
|
|
|
|
###### SIDEBAR ######
|
|
|
kwargs.update(sidebar(conf=conf))
|
|
|
|
|
|
return render(request, 'generator_conf.html', kwargs)
|
|
|
|
|
|
@login_required
|
|
|
def conf_edit(request, conf_id):
|
|
|
|
|
|
conf = get_object_or_404(GeneratorConfiguration, pk=conf_id)
|
|
|
|
|
|
if not is_developer(request.user):
|
|
|
messages.error(request, 'You must be an developer to edit this configuration')
|
|
|
return redirect(conf.get_absolute_url())
|
|
|
|
|
|
if request.method=='GET':
|
|
|
|
|
|
form = GeneratorConfigurationForm(instance=conf)
|
|
|
|
|
|
elif request.method=='POST':
|
|
|
|
|
|
form = GeneratorConfigurationForm(request.POST, instance=conf)
|
|
|
|
|
|
if form.is_valid():
|
|
|
form.save()
|
|
|
|
|
|
messages.success(request, 'Generator configuration successfully updated')
|
|
|
|
|
|
return redirect(conf.get_absolute_url())
|
|
|
|
|
|
kwargs = {}
|
|
|
kwargs['dev_conf'] = conf
|
|
|
kwargs['form'] = form
|
|
|
kwargs['edit'] = True
|
|
|
|
|
|
kwargs['title'] = 'Generator Configuration'
|
|
|
kwargs['suptitle'] = 'Edit'
|
|
|
kwargs['button'] = 'Update'
|
|
|
|
|
|
return render(request, 'generator_conf_edit.html', kwargs)
|
|
|
|
|
|
def import_file(request, conf_id):
|
|
|
|
|
|
conf = get_object_or_404(GeneratorConfiguration, pk=conf_id)
|
|
|
if request.method=='POST':
|
|
|
form = GeneratorImportForm(request.POST, request.FILES)
|
|
|
if form.is_valid():
|
|
|
try:
|
|
|
data = conf.import_from_file(request.FILES['file_name'])
|
|
|
conf.dict_to_parms(data)
|
|
|
messages.success(request, 'Configuration "%s" loaded succesfully' % request.FILES['file_name'])
|
|
|
return redirect(conf.get_absolute_url_edit())
|
|
|
|
|
|
except Exception as e:
|
|
|
messages.error(request, 'Error parsing file: "%s" - %s' % (request.FILES['file_name'], repr(e)))
|
|
|
else:
|
|
|
messages.warning(request, 'Your current configuration will be replaced')
|
|
|
form = GeneratorImportForm()
|
|
|
|
|
|
kwargs = {}
|
|
|
kwargs['form'] = form
|
|
|
kwargs['title'] = 'Generator Configuration'
|
|
|
kwargs['suptitle'] = 'Import file'
|
|
|
kwargs['button'] = 'Upload'
|
|
|
kwargs['previous'] = conf.get_absolute_url()
|
|
|
|
|
|
return render(request, 'generator_import.html', kwargs)
|
|
|
|
|
|
def conf_raw(request, conf_id):
|
|
|
conf = get_object_or_404(GeneratorConfiguration, pk=conf_id)
|
|
|
raw = conf.write_device(raw=True)
|
|
|
return HttpResponse(raw, content_type='application/json')
|
