From ba1315d8a611f1b891cfd6610fbd053077357cb7 Mon Sep 17 00:00:00 2001
From: Filipp Lepalaan <filipp@mac.com>
Date: Thu, 8 Oct 2015 10:49:01 +0300
Subject: Add Dispatch & Close

---
 servo/views/invoices.py | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'servo/views')

diff --git a/servo/views/invoices.py b/servo/views/invoices.py
index 2fb7b26..6b77c8b 100644
--- a/servo/views/invoices.py
+++ b/servo/views/invoices.py
@@ -159,6 +159,11 @@ def create_invoice(request, order_id=None, numbers=None):
                 messages.success(request, _(u'Order %s dispatched') % order.code)
             except Exception as e:
                 messages.error(request, e)
+
+            if form.cleaned_data.get('close'):
+                order.close(request.user)
+                messages.success(request, _(u'Order %s closed') % order.code)
+
             return redirect(order)
         else:
             messages.error(request, form.errors)
-- 
cgit v1.2.3