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, 'yeast/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') |