views.py
385 lines
| 13.1 KiB
| text/x-python
|
PythonLexer
Juan C. Espinoza
|
r79 | |||
|
Juan C. Espinoza
|
r23 | import json | ||
| from django.contrib import messages | ||||
|
Juan C. Espinoza
|
r45 | from django.utils.safestring import mark_safe | ||
|
Juan C. Espinoza
|
r25 | from django.shortcuts import render, redirect, get_object_or_404, HttpResponse | ||
|
Juan C. Espinoza
|
r23 | |||
|
Juan C. Espinoza
|
r45 | from apps.main.models import Configuration, Experiment, Device | ||
|
Juan C. Espinoza
|
r25 | from apps.main.views import sidebar | ||
|
Juan C. Espinoza
|
r23 | |||
|
Juan C. Espinoza
|
r45 | from .models import RCConfiguration, RCLine, RCLineType, RCLineCode | ||
|
Juan C. Espinoza
|
r79 | from .forms import RCConfigurationForm, RCLineForm, RCLineViewForm, RCLineEditForm, RCImportForm, RCLineCodesForm | ||
| from .utils import plot_pulses | ||||
|
Juan C. Espinoza
|
r23 | |||
|
Juan C. Espinoza
|
r45 | def conf(request, conf_id): | ||
|
Juan C. Espinoza
|
r23 | |||
|
Juan C. Espinoza
|
r45 | conf = get_object_or_404(RCConfiguration, pk=conf_id) | ||
|
Juan C. Espinoza
|
r23 | |||
| lines = RCLine.objects.filter(rc_configuration=conf).order_by('channel') | ||||
| for line in lines: | ||||
|
Juan C. Espinoza
|
r45 | params = json.loads(line.params) | ||
| line.form = RCLineViewForm(extra_fields=params, line=line) | ||||
| if 'params' in params: | ||||
| line.subforms = [RCLineViewForm(extra_fields=fields, line=line, subform=True) for fields in params['params']] | ||||
|
Juan C. Espinoza
|
r23 | |||
| kwargs = {} | ||||
| kwargs['dev_conf'] = conf | ||||
| kwargs['rc_lines'] = lines | ||||
|
Juan C. Espinoza
|
r79 | kwargs['dev_conf_keys'] = ['name', 'ipp', 'ntx', 'clock_in', 'clock_divider', 'clock', | ||
| 'time_before', 'time_after', 'sync', 'sampling_reference', 'control_tx', 'control_sw'] | ||||
|
Juan C. Espinoza
|
r23 | |||
| kwargs['title'] = 'RC Configuration' | ||||
| kwargs['suptitle'] = 'Details' | ||||
| kwargs['button'] = 'Edit Configuration' | ||||
| ###### SIDEBAR ###### | ||||
|
Juan C. Espinoza
|
r85 | kwargs.update(sidebar(conf=conf)) | ||
|
Juan C. Espinoza
|
r23 | |||
| return render(request, 'rc_conf.html', kwargs) | ||||
|
Juan C. Espinoza
|
r45 | |||
| def conf_edit(request, conf_id): | ||||
|
Juan C. Espinoza
|
r23 | |||
|
Juan C. Espinoza
|
r45 | conf = get_object_or_404(RCConfiguration, pk=conf_id) | ||
|
Juan C. Espinoza
|
r23 | |||
|
Juan C. Espinoza
|
r45 | lines = RCLine.objects.filter(rc_configuration=conf).order_by('channel') | ||
|
Juan C. Espinoza
|
r79 | for line in lines: | ||
| params = json.loads(line.params) | ||||
| line.form = RCLineEditForm(conf=conf, line=line, extra_fields=params) | ||||
| line.subform = False | ||||
|
Juan C. Espinoza
|
r45 | |||
| if 'params' in params: | ||||
|
Juan C. Espinoza
|
r79 | line.subforms = [RCLineEditForm(extra_fields=fields, line=line, subform=i) for i, fields in enumerate(params['params'])] | ||
|
Juan C. Espinoza
|
r45 | line.subform = True | ||
|
Juan C. Espinoza
|
r23 | |||
| if request.method=='GET': | ||||
|
Juan C. Espinoza
|
r45 | |||
|
Juan C. Espinoza
|
r23 | form = RCConfigurationForm(instance=conf) | ||
|
Juan C. Espinoza
|
r45 | |||
| elif request.method=='POST': | ||||
|
Juan C. Espinoza
|
r23 | |||
|
Juan C. Espinoza
|
r45 | line_data = {} | ||
| conf_data = {} | ||||
| extras = [] | ||||
|
Juan C. Espinoza
|
r23 | |||
|
Juan C. Espinoza
|
r45 | #classified post fields | ||
| for label,value in request.POST.items(): | ||||
| if label=='csrfmiddlewaretoken': | ||||
| continue | ||||
|
Juan C. Espinoza
|
r79 | |||
|
Juan C. Espinoza
|
r45 | if label.count('|')==0: | ||
| conf_data[label] = value | ||||
| continue | ||||
|
Juan C. Espinoza
|
r79 | |||
| elif label.split('|')[0]<>'-1': | ||||
|
Juan C. Espinoza
|
r45 | extras.append(label) | ||
| continue | ||||
|
Juan C. Espinoza
|
r79 | x, pk, name = label.split('|') | ||
| if name=='codes': | ||||
| value = [s for s in value.split('\r\n') if s] | ||||
|
Juan C. Espinoza
|
r45 | if pk in line_data: | ||
| line_data[pk][name] = value | ||||
| else: | ||||
| line_data[pk] = {name:value} | ||||
| #update conf | ||||
| form = RCConfigurationForm(conf_data, instance=conf) | ||||
|
Juan C. Espinoza
|
r23 | if form.is_valid(): | ||
|
Juan C. Espinoza
|
r45 | |||
|
Juan C. Espinoza
|
r23 | form.save() | ||
|
Juan C. Espinoza
|
r45 | |||
| #update lines fields | ||||
| extras.sort() | ||||
| for label in extras: | ||||
| x, pk, name = label.split('|') | ||||
| if pk not in line_data: | ||||
| line_data[pk] = {} | ||||
| if 'params' not in line_data[pk]: | ||||
| line_data[pk]['params'] = [] | ||||
| if len(line_data[pk]['params'])<int(x)+1: | ||||
| line_data[pk]['params'].append({}) | ||||
| line_data[pk]['params'][int(x)][name] = float(request.POST[label]) | ||||
| for pk, params in line_data.items(): | ||||
| line = RCLine.objects.get(pk=pk) | ||||
| if line.line_type.name in ('windows', 'prog_pulses'): | ||||
| if 'params' not in params: | ||||
| params['params'] = [] | ||||
| line.params = json.dumps(params) | ||||
| line.save() | ||||
| #update pulses field | ||||
|
Juan C. Espinoza
|
r107 | conf.update_pulses() | ||
|
Juan C. Espinoza
|
r45 | |||
| messages.success(request, 'RC Configuration successfully updated') | ||||
| return redirect(conf.get_absolute_url()) | ||||
|
Juan C. Espinoza
|
r23 | kwargs = {} | ||
|
Juan C. Espinoza
|
r25 | kwargs['dev_conf'] = conf | ||
|
Juan C. Espinoza
|
r23 | kwargs['form'] = form | ||
|
Juan C. Espinoza
|
r45 | kwargs['rc_lines'] = lines | ||
| kwargs['edit'] = True | ||||
| kwargs['title'] = 'RC Configuration' | ||||
|
Juan C. Espinoza
|
r23 | kwargs['suptitle'] = 'Edit' | ||
| kwargs['button'] = 'Update' | ||||
| kwargs['previous'] = conf.get_absolute_url() | ||||
|
Juan C. Espinoza
|
r25 | |||
|
Juan C. Espinoza
|
r23 | return render(request, 'rc_conf_edit.html', kwargs) | ||
|
Juan C. Espinoza
|
r79 | def add_line(request, conf_id, line_type_id=None, code_id=None): | ||
|
Juan C. Espinoza
|
r23 | |||
|
Juan C. Espinoza
|
r25 | conf = get_object_or_404(RCConfiguration, pk=conf_id) | ||
|
Juan C. Espinoza
|
r23 | |||
| if request.method=='GET': | ||||
|
Juan C. Espinoza
|
r25 | if line_type_id: | ||
| line_type = get_object_or_404(RCLineType, pk=line_type_id) | ||||
|
Juan C. Espinoza
|
r79 | |||
| if code_id: | ||||
| form = RCLineForm(initial={'rc_configuration':conf_id, 'line_type': line_type_id, 'code_id': code_id}, | ||||
| extra_fields=json.loads(line_type.params)) | ||||
| else: | ||||
| form = RCLineForm(initial={'rc_configuration':conf_id, 'line_type': line_type_id}, | ||||
| extra_fields=json.loads(line_type.params)) | ||||
|
Juan C. Espinoza
|
r25 | else: | ||
|
Juan C. Espinoza
|
r79 | line_type = {'id':0} | ||
|
Juan C. Espinoza
|
r25 | form = RCLineForm(initial={'rc_configuration':conf_id}) | ||
|
Juan C. Espinoza
|
r23 | |||
| if request.method=='POST': | ||||
|
Juan C. Espinoza
|
r79 | |||
|
Juan C. Espinoza
|
r25 | line_type = get_object_or_404(RCLineType, pk=line_type_id) | ||
|
Juan C. Espinoza
|
r79 | form = RCLineForm(request.POST, | ||
| extra_fields=json.loads(line_type.params)) | ||||
|
Juan C. Espinoza
|
r23 | |||
| if form.is_valid(): | ||||
| form.save() | ||||
|
Juan C. Espinoza
|
r45 | form.instance.update_pulses() | ||
| return redirect('url_edit_rc_conf', conf.id) | ||||
|
Juan C. Espinoza
|
r23 | |||
| kwargs = {} | ||||
| kwargs['form'] = form | ||||
| kwargs['title'] = 'RC Configuration' | ||||
| kwargs['suptitle'] = 'Add Line' | ||||
| kwargs['button'] = 'Add' | ||||
|
Juan C. Espinoza
|
r45 | kwargs['previous'] = conf.get_absolute_url_edit() | ||
|
Juan C. Espinoza
|
r25 | kwargs['dev_conf'] = conf | ||
|
Juan C. Espinoza
|
r79 | kwargs['line_type'] = line_type | ||
|
Juan C. Espinoza
|
r25 | |||
|
Juan C. Espinoza
|
r23 | return render(request, 'rc_add_line.html', kwargs) | ||
|
Juan C. Espinoza
|
r79 | def edit_codes(request, conf_id, line_id, code_id=None): | ||
| conf = get_object_or_404(RCConfiguration, pk=conf_id) | ||||
| line = get_object_or_404(RCLine, pk=line_id) | ||||
| params = json.loads(line.params) | ||||
| if request.method=='GET': | ||||
| if code_id: | ||||
| code = get_object_or_404(RCLineCode, pk=code_id) | ||||
| form = RCLineCodesForm(instance=code) | ||||
| else: | ||||
| initial = {'code': params['code'], | ||||
| 'codes': params['codes'] if 'codes' in params else [], | ||||
| 'number_of_codes': len(params['codes']) if 'codes' in params else 0, | ||||
| 'bits_per_code': len(params['codes'][0]) if 'codes' in params else 0, | ||||
| } | ||||
| form = RCLineCodesForm(initial=initial) | ||||
| if request.method=='POST': | ||||
| form = RCLineCodesForm(request.POST) | ||||
| if form.is_valid(): | ||||
| params['code'] = request.POST['code'] | ||||
| params['codes'] = [s for s in request.POST['codes'].split('\r\n') if s] | ||||
| line.params = json.dumps(params) | ||||
| line.save() | ||||
| messages.success(request, 'Line: "%s" has been updated.' % line) | ||||
| return redirect('url_edit_rc_conf', conf.id) | ||||
| kwargs = {} | ||||
| kwargs['form'] = form | ||||
| kwargs['title'] = line | ||||
| kwargs['suptitle'] = 'Edit' | ||||
| kwargs['button'] = 'Update' | ||||
| kwargs['dev_conf'] = conf | ||||
| kwargs['previous'] = conf.get_absolute_url_edit() | ||||
| kwargs['line'] = line | ||||
| return render(request, 'rc_edit_codes.html', kwargs) | ||||
|
Juan C. Espinoza
|
r45 | def add_subline(request, conf_id, line_id): | ||
| conf = get_object_or_404(RCConfiguration, pk=conf_id) | ||||
| line = get_object_or_404(RCLine, pk=line_id) | ||||
| if request.method == 'POST': | ||||
| if line: | ||||
| params = json.loads(line.params) | ||||
|
Juan C. Espinoza
|
r79 | subparams = json.loads(line.line_type.params) | ||
| if 'params' in subparams: | ||||
| dum = {} | ||||
| for key, value in subparams['params'].items(): | ||||
| dum[key] = value['value'] | ||||
| params['params'].append(dum) | ||||
|
Juan C. Espinoza
|
r45 | line.params = json.dumps(params) | ||
| line.save() | ||||
| return redirect('url_edit_rc_conf', conf.id) | ||||
| kwargs = {} | ||||
| kwargs['title'] = 'Add new' | ||||
| kwargs['suptitle'] = '%s to %s' % (line.line_type.name, line) | ||||
| return render(request, 'confirm.html', kwargs) | ||||
|
Juan C. Espinoza
|
r23 | |||
| def remove_line(request, conf_id, line_id): | ||||
| conf = get_object_or_404(RCConfiguration, pk=conf_id) | ||||
| line = get_object_or_404(RCLine, pk=line_id) | ||||
| if request.method == 'POST': | ||||
| if line: | ||||
| try: | ||||
| channel = line.channel | ||||
| line.delete() | ||||
| for ch in range(channel+1, RCLine.objects.filter(rc_configuration=conf).count()+1): | ||||
| l = RCLine.objects.get(rc_configuration=conf, channel=ch) | ||||
| l.channel = l.channel-1 | ||||
| l.save() | ||||
| messages.success(request, 'Line: "%s" has been deleted.' % line) | ||||
| except: | ||||
| messages.error(request, 'Unable to delete line: "%s".' % line) | ||||
|
Juan C. Espinoza
|
r45 | return redirect('url_edit_rc_conf', conf.id) | ||
|
Juan C. Espinoza
|
r23 | |||
| kwargs = {} | ||||
| kwargs['object'] = line | ||||
| kwargs['delete_view'] = True | ||||
| kwargs['title'] = 'Confirm delete' | ||||
|
Juan C. Espinoza
|
r45 | kwargs['previous'] = conf.get_absolute_url_edit() | ||
|
Juan C. Espinoza
|
r23 | return render(request, 'confirm.html', kwargs) | ||
|
Juan C. Espinoza
|
r79 | |||
|
Juan C. Espinoza
|
r45 | def remove_subline(request, conf_id, line_id, subline_id): | ||
|
Juan C. Espinoza
|
r23 | |||
| conf = get_object_or_404(RCConfiguration, pk=conf_id) | ||||
|
Juan C. Espinoza
|
r45 | line = get_object_or_404(RCLine, pk=line_id) | ||
|
Juan C. Espinoza
|
r23 | |||
|
Juan C. Espinoza
|
r45 | if request.method == 'POST': | ||
| if line: | ||||
| params = json.loads(line.params) | ||||
| params['params'].remove(params['params'][int(subline_id)-1]) | ||||
|
Juan C. Espinoza
|
r25 | line.params = json.dumps(params) | ||
|
Juan C. Espinoza
|
r45 | line.save() | ||
| return redirect('url_edit_rc_conf', conf.id) | ||||
|
Juan C. Espinoza
|
r23 | |||
|
Juan C. Espinoza
|
r45 | kwargs = {} | ||
|
Juan C. Espinoza
|
r25 | |||
|
Juan C. Espinoza
|
r45 | kwargs['object'] = line | ||
| kwargs['object_name'] = line.line_type.name | ||||
| kwargs['delete_view'] = True | ||||
| kwargs['title'] = 'Confirm delete' | ||||
|
Juan C. Espinoza
|
r25 | |||
|
Juan C. Espinoza
|
r45 | return render(request, 'confirm.html', kwargs) | ||
|
Juan C. Espinoza
|
r25 | |||
|
Juan C. Espinoza
|
r23 | |||
|
Juan C. Espinoza
|
r45 | def update_lines_position(request, conf_id): | ||
|
Juan C. Espinoza
|
r23 | |||
| conf = get_object_or_404(RCConfiguration, pk=conf_id) | ||||
|
Juan C. Espinoza
|
r25 | |||
| if request.method=='POST': | ||||
| ch = 0 | ||||
| for item in request.POST['items'].split('&'): | ||||
| line = RCLine.objects.get(pk=item.split('=')[-1]) | ||||
| line.channel = ch | ||||
|
Juan C. Espinoza
|
r23 | line.save() | ||
|
Juan C. Espinoza
|
r25 | ch += 1 | ||
| lines = RCLine.objects.filter(rc_configuration=conf).order_by('channel') | ||||
| for line in lines: | ||||
|
Juan C. Espinoza
|
r79 | params = json.loads(line.params) | ||
| line.form = RCLineEditForm(conf=conf, line=line, extra_fields=params) | ||||
| if 'params' in params: | ||||
| line.subform = True | ||||
| line.subforms = [RCLineEditForm(extra_fields=fields, line=line, subform=i) for i, fields in enumerate(params['params'])] | ||||
|
Juan C. Espinoza
|
r25 | |||
|
Juan C. Espinoza
|
r79 | html = render(request, 'rc_lines.html', {'dev_conf':conf, 'rc_lines':lines, 'edit':True}) | ||
|
Juan C. Espinoza
|
r25 | data = {'html': html.content} | ||
| return HttpResponse(json.dumps(data), content_type="application/json") | ||||
|
Juan C. Espinoza
|
r45 | return redirect('url_edit_rc_conf', conf.id) | ||
|
Juan C. Espinoza
|
r79 | |||
|
Juan C. Espinoza
|
r45 | |||
| def import_file(request, conf_id): | ||||
|
Juan C. Espinoza
|
r25 | |||
|
Juan C. Espinoza
|
r45 | conf = get_object_or_404(RCConfiguration, pk=conf_id) | ||
| if request.method=='POST': | ||||
| form = RCImportForm(request.POST, request.FILES) | ||||
| if form.is_valid(): | ||||
| #try: | ||||
| if True: | ||||
|
Juan C. Espinoza
|
r79 | conf.update_from_file(request.FILES['file_name']) | ||
| conf.save() | ||||
|
Juan C. Espinoza
|
r45 | for line in conf.get_lines(): | ||
| if line.line_type.name=='tr': | ||||
| continue | ||||
| line.update_pulses() | ||||
| for tr in conf.get_lines('tr'): | ||||
|
Juan C. Espinoza
|
r79 | tr.update_pulses() | ||
|
Juan C. Espinoza
|
r45 | |||
| messages.success(request, 'Configuration "%s" loaded succesfully' % request.FILES['file_name']) | ||||
| return redirect(conf.get_absolute_url()) | ||||
| #except Exception as e: | ||||
| # messages.error(request, 'Error parsing file: "%s" - %s' % (request.FILES['file_name'], e)) | ||||
| else: | ||||
| messages.warning(request, 'Your current configuration will be replaced') | ||||
|
Juan C. Espinoza
|
r79 | form = RCImportForm() | ||
|
Juan C. Espinoza
|
r45 | |||
| kwargs = {} | ||||
| kwargs['form'] = form | ||||
| kwargs['title'] = 'RC Configuration' | ||||
| kwargs['suptitle'] = 'Import file' | ||||
| kwargs['button'] = 'Upload' | ||||
| kwargs['previous'] = conf.get_absolute_url() | ||||
| return render(request, 'rc_import.html', kwargs) | ||||
|
Juan C. Espinoza
|
r79 | |||
|
Juan C. Espinoza
|
r0 | |||
|
Juan C. Espinoza
|
r45 | def view_pulses(request, conf_id): | ||
| conf = get_object_or_404(RCConfiguration, pk=conf_id) | ||||
| lines = RCLine.objects.filter(rc_configuration=conf) | ||||
| unit = (conf.clock/conf.clock_divider)*3./20 | ||||
|
Juan C. Espinoza
|
r107 | N = conf.ipp*conf.km2unit*conf.ntx | ||
|
Juan C. Espinoza
|
r45 | |||
|
Juan C. Espinoza
|
r79 | script, div = plot_pulses(unit, N, lines) | ||
|
Juan C. Espinoza
|
r45 | |||
|
Juan C. Espinoza
|
r79 | kwargs = {} | ||
|
Juan C. Espinoza
|
r45 | |||
|
Juan C. Espinoza
|
r79 | kwargs['title'] = 'RC Pulses' | ||
| kwargs['suptitle'] = conf.name | ||||
| kwargs['div'] = mark_safe(div) | ||||
| kwargs['script'] = mark_safe(script) | ||||
| return render(request, 'rc_pulses.html', kwargs) | ||||
|
Juan C. Espinoza
|
r45 | |||