c4d9d9489d
Need to use 'yeast' instead of 'sample' to keep thing consistent for the label generator.
79 lines
2.5 KiB
Python
79 lines
2.5 KiB
Python
from django.shortcuts import render, get_object_or_404
|
|
from django.views.generic import ListView
|
|
from django.views.generic.edit import CreateView
|
|
from django.http import HttpResponse, HttpRequest
|
|
from django.urls import reverse
|
|
|
|
from yeast.models import Yeast, Batch, Strain
|
|
from config.extras import AveryLabel
|
|
from yeast.forms import BatchAddForm, StrainAddForm
|
|
|
|
|
|
import logging
|
|
logger = logging.getLogger('django')
|
|
|
|
class YeastListView(ListView):
|
|
model = Yeast
|
|
|
|
class BatchListView(ListView):
|
|
model = Batch
|
|
|
|
def sample(request, yeast_id):
|
|
|
|
sample = get_object_or_404(Yeast, pk=yeast_id)
|
|
sample_batch = get_object_or_404(Batch, pk=sample.batch_id)
|
|
return render(request, 'yeast/sample.html', {'sample': sample, 'batch':sample_batch})
|
|
|
|
def home(request):
|
|
return render(request, 'home.html',{})
|
|
|
|
def batch(request, batch_id):
|
|
batch = get_object_or_404(Batch, pk=batch_id)
|
|
return render(request, 'yeast/batch.html', {'batch': batch})
|
|
|
|
def batch_labels(request, batch_id):
|
|
""" Create label PDF for samples in a batch
|
|
"""
|
|
skip_count = request.POST.get("skip_count", "")
|
|
samples = request.POST.getlist("samples", "")
|
|
batch = get_object_or_404(Batch, pk=batch_id)
|
|
to_print = list(filter(lambda d: str(d.id) in samples, batch.yeast_set.all()))
|
|
|
|
# Create the HttpResponse object with the appropriate PDF headers.
|
|
response = HttpResponse(content_type ='application/pdf')
|
|
response['Content-Disposition'] = 'attachment; filename=samplelabels.pdf'
|
|
labelSheet = AveryLabel(18294, debug=False)
|
|
|
|
logger.critical(samples)
|
|
logger.critical(to_print)
|
|
|
|
labels = []
|
|
for sample in to_print:
|
|
labels.append({
|
|
'id': sample.id,
|
|
'title': '{} {}'.format(sample.batch.strain.manufacturer.name, sample.batch.strain.name),
|
|
'data': ['ID: {}'.format(sample.id), 'Date: {}'.format(sample.batch.production_date)],
|
|
'blank': False,
|
|
'host': request.get_host(),
|
|
'template': 'yeast',
|
|
'ns': 'yeast'
|
|
})
|
|
labelSheet.render(labels, response, skip_count)
|
|
|
|
return response
|
|
|
|
class addBatch(CreateView):
|
|
model = Batch
|
|
form_class = BatchAddForm
|
|
|
|
def get_success_url(self):
|
|
id = self.object.id #gets id from created object
|
|
return reverse('yeast:batches', kwargs={'batch_id': id})
|
|
|
|
class addStrain(CreateView):
|
|
model = Strain
|
|
form_class = StrainAddForm
|
|
|
|
def get_success_url(self):
|
|
id = self.object.id #gets id from created object
|
|
return reverse('yeast:batches') |