diff options
Diffstat (limited to 'servo/views')
-rw-r--r-- | servo/views/note.py | 44 | ||||
-rw-r--r-- | servo/views/order.py | 53 |
2 files changed, 50 insertions, 47 deletions
diff --git a/servo/views/note.py b/servo/views/note.py index c3b0b99..9672a84 100644 --- a/servo/views/note.py +++ b/servo/views/note.py @@ -10,34 +10,34 @@ from django.http import HttpResponse from django.utils.translation import ugettext as _ from django.forms.models import modelformset_factory from django.views.decorators.csrf import csrf_exempt -from django.views.decorators.cache import cache_page from django.shortcuts import render, redirect, get_object_or_404 from django.contrib.auth.decorators import permission_required +from django.core.files.base import ContentFile + from reportlab.lib.units import mm from reportlab.graphics.shapes import Drawing from reportlab.graphics.barcode import createBarcodeDrawing from servo.lib.utils import paginate from servo.models import (Order, Template, Tag, Customer, Note, - Attachment, Escalation, Article,) + Attachment, Escalation, Article,) from servo.forms import NoteForm, NoteSearchForm, EscalationForm class BarcodeDrawing(Drawing): + """Pretty generic Reportlab drawing to render barcodes.""" def __init__(self, text_value, *args, **kwargs): barcode = createBarcodeDrawing("Code128", value=text_value.encode("utf-8"), - barHeight=10*mm, - width=80*mm) + barHeight=10 * mm, + width=80 * mm) Drawing.__init__(self, barcode.width, barcode.height, *args, **kwargs) self.add(barcode, name="barcode") def show_barcode(request, text): - """ - Returns text as a barcode - """ + """Return text as a barcode.""" if request.GET.get('f') == 'svg': import barcode output = StringIO.StringIO() @@ -52,9 +52,7 @@ def show_barcode(request, text): def prep_list_view(request, kind): - """ - Prepares the view for listing notes/messages - """ + """Prepare the view for listing notes/messages.""" data = {'title': _("Messages")} all_notes = Note.objects.all().order_by("-created_at") @@ -82,10 +80,7 @@ def prep_list_view(request, kind): @permission_required('servo.change_note') def copy(request, pk): - """ - Copies a note with its attachments and labels - """ - from servo.lib.shorturl import from_time + """Copy a note with its attachments and labels.""" note = get_object_or_404(Note, pk=pk) new_note = Note(created_by=request.user) @@ -96,7 +91,7 @@ def copy(request, pk): new_note.labels = note.labels.all() - for a in note.attachments.all(): # also copy the attachments + for a in note.attachments.all(): # also copy the attachments a.pk = None a.content_object = new_note a.save() @@ -106,10 +101,11 @@ def copy(request, pk): @permission_required('servo.change_note') -def edit(request, pk=None, order_id=None, parent=None, recipient=None, - customer=None): +def edit(request, pk=None, order_id=None, + parent=None, recipient=None, customer=None): """ - Edits a note + Edit a note. + @FIXME: Should split this up into smaller pieces """ to = [] @@ -220,6 +216,18 @@ def edit(request, pk=None, order_id=None, parent=None, recipient=None, return redirect(note) note.attachments.add(*files) + + if form.cleaned_data.get('attach_confirmation'): + from servo.views.order import put_on_paper + response = put_on_paper(request, note.order_id, fmt='pdf') + filename = response.filename + content = response.render().content + content = ContentFile(content, filename) + attachment = Attachment(content=content, content_object=note) + attachment.save() + attachment.content.save(filename, content) + note.attachments.add(attachment) + note.save() try: diff --git a/servo/views/order.py b/servo/views/order.py index 835aaf4..6853ee4 100644 --- a/servo/views/order.py +++ b/servo/views/order.py @@ -17,12 +17,13 @@ from django.db import DatabaseError from django.core.urlresolvers import reverse from django.utils.translation import ugettext as _ -from django.views.decorators.cache import cache_page from django.shortcuts import render, redirect, get_object_or_404 from django.views.decorators.csrf import csrf_exempt from django.contrib.auth.decorators import permission_required +from wkhtmltopdf.views import PDFTemplateView + from servo.lib.utils import paginate from servo.models.order import * @@ -31,7 +32,6 @@ from servo.forms.repairs import StatusForm from servo.models import Note, User, Device, Customer from servo.models.common import (Tag, - Configuration, FlaggedItem, GsxAccount,) from servo.models.repair import (Checklist, @@ -191,9 +191,7 @@ def prepare_detail_view(request, pk): @permission_required("servo.change_order") def close(request, pk): - """ - Closes this Service Order - """ + """Close this Service Order.""" order = get_object_or_404(Order, pk=pk) if request.method == 'POST': @@ -218,6 +216,7 @@ def close(request, pk): @permission_required("servo.delete_order") def reopen_order(request, pk): + """Open a closed order.""" order = get_object_or_404(Order, pk=pk) msg = order.reopen(request.user) messages.success(request, msg) @@ -225,7 +224,10 @@ def reopen_order(request, pk): @permission_required("servo.add_order") -def create(request, sn=None, device_id=None, product_id=None, note_id=None, customer_id=None): +def create(request, sn=None, device_id=None, + product_id=None, + note_id=None, + customer_id=None): """Create a new Service Order.""" order = Order(created_by=request.user) @@ -554,32 +556,29 @@ def update_order(request, pk, what, what_id): return redirect(order) -def put_on_paper(request, pk, kind="confirmation"): +def put_on_paper(request, pk, kind="confirmation", fmt='html'): """ 'Print' was taken? """ - conf = Configuration.conf() order = get_object_or_404(Order, pk=pk) - - title = _(u"Service Order #%s") % order.code - notes = order.note_set.filter(is_reported=True) - + data = order.get_print_dict(kind) template = order.get_print_template(kind) - - if kind == "receipt": - try: - invoice = order.invoice_set.latest() - except Exception as e: - pass - return render(request, template, locals()) + if fmt == 'pdf': + fn = data.get('title') + '.pdf' + view = PDFTemplateView(request=request, template_name=template, + filename=fn) + return view.render_to_response(data) + + return render(request, template, data) @permission_required("servo.change_order") def add_device(request, pk, device_id=None, sn=None): """ - Adds a device to a service order - using device_id with existing devices or + Add a device to a service order. + + Use device_id with existing devices or sn for new devices (which should have gone through GSX search) """ order = get_object_or_404(Order, pk=pk) @@ -670,9 +669,7 @@ def reserve_products(request, pk): @permission_required("servo.change_order") def edit_product(request, pk, item_id): - """ - Edits a product added to an order - """ + """Edit a product added to an order.""" order = Order.objects.get(pk=pk) item = get_object_or_404(ServiceOrderItem, pk=item_id) @@ -857,9 +854,7 @@ def choose_customer(request, pk): if len(query) > 2: customers = Customer.objects.filter( - Q(fullname__icontains=query) - | Q(email__icontains=query) - | Q(phone__contains=query) + Q(fullname__icontains=query) | Q(email__icontains=query) | Q(phone__contains=query) ) if kind == 'companies': @@ -936,8 +931,8 @@ def download_results(request): writer.writerow(header) for o in request.session['order_queryset']: - row = [o.code, o.customer, o.created_at, - o.user, o.checkin_location, o.location] + row = [o.code, o.customer, o.created_at, + o.user, o.checkin_location, o.location] coded = [unicode(s).encode('utf-8') for s in row] writer.writerow(coded) |