Nevar pievienot vairāk kā 25 tēmas Tēmai ir jāsākas ar burtu vai ciparu, tā var saturēt domu zīmes ('-') un var būt līdz 35 simboliem gara.

views.py 31KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782
  1. import datetime
  2. from decimal import Decimal as D, InvalidOperation
  3. from django.contrib import messages
  4. from django.utils.translation import ugettext_lazy as _
  5. from django.core.urlresolvers import reverse
  6. from django.core.exceptions import ObjectDoesNotExist
  7. from django.db.models import fields, Q, Sum, Count
  8. from django.http import HttpResponse, HttpResponseRedirect, Http404
  9. from django.shortcuts import get_object_or_404, redirect
  10. from django.utils.datastructures import SortedDict
  11. from django.views.generic import ListView, DetailView, UpdateView, FormView
  12. from django.conf import settings
  13. from oscar.core.loading import get_class, get_model
  14. from oscar.core.utils import format_datetime
  15. from oscar.core.compat import UnicodeCSVWriter
  16. from oscar.views import sort_queryset
  17. from oscar.views.generic import BulkEditMixin
  18. from oscar.apps.payment.exceptions import PaymentError
  19. from oscar.apps.order import exceptions as order_exceptions
  20. Partner = get_model('partner', 'Partner')
  21. Transaction = get_model('payment', 'Transaction')
  22. Order = get_model('order', 'Order')
  23. OrderNote = get_model('order', 'OrderNote')
  24. ShippingAddress = get_model('order', 'ShippingAddress')
  25. Line = get_model('order', 'Line')
  26. ShippingEventType = get_model('order', 'ShippingEventType')
  27. PaymentEventType = get_model('order', 'PaymentEventType')
  28. EventHandler = get_class('order.processing', 'EventHandler')
  29. OrderStatsForm = get_class('dashboard.orders.forms', 'OrderStatsForm')
  30. OrderSearchForm = get_class('dashboard.orders.forms', 'OrderSearchForm')
  31. OrderNoteForm = get_class('dashboard.orders.forms', 'OrderNoteForm')
  32. ShippingAddressForm = get_class(
  33. 'dashboard.orders.forms', 'ShippingAddressForm')
  34. OrderStatusForm = get_class('dashboard.orders.forms', 'OrderStatusForm')
  35. def queryset_orders_for_user(user):
  36. """
  37. Returns a queryset of all orders that a user is allowed to access.
  38. A staff user may access all orders.
  39. To allow access to an order for a non-staff user, at least one line's
  40. partner has to have the user in the partner's list.
  41. """
  42. queryset = Order._default_manager.select_related(
  43. 'billing_address', 'billing_address__country',
  44. 'shipping_address', 'shipping_address__country',
  45. 'user'
  46. ).prefetch_related('lines')
  47. if user.is_staff:
  48. return queryset
  49. else:
  50. partners = Partner._default_manager.filter(users=user)
  51. return queryset.filter(lines__partner__in=partners).distinct()
  52. def get_order_for_user_or_404(user, number):
  53. try:
  54. return queryset_orders_for_user(user).get(number=number)
  55. except ObjectDoesNotExist:
  56. raise Http404()
  57. class OrderStatsView(FormView):
  58. """
  59. Dashboard view for order statistics.
  60. Supports the permission-based dashboard.
  61. """
  62. template_name = 'dashboard/orders/statistics.html'
  63. form_class = OrderStatsForm
  64. def get(self, request, *args, **kwargs):
  65. return self.post(request, *args, **kwargs)
  66. def form_valid(self, form):
  67. ctx = self.get_context_data(form=form,
  68. filters=form.get_filters())
  69. return self.render_to_response(ctx)
  70. def get_form_kwargs(self):
  71. kwargs = super(OrderStatsView, self).get_form_kwargs()
  72. kwargs['data'] = self.request.GET
  73. return kwargs
  74. def get_context_data(self, **kwargs):
  75. ctx = super(OrderStatsView, self).get_context_data(**kwargs)
  76. filters = kwargs.get('filters', {})
  77. ctx.update(self.get_stats(filters))
  78. ctx['title'] = kwargs['form'].get_filter_description()
  79. return ctx
  80. def get_stats(self, filters):
  81. orders = queryset_orders_for_user(self.request.user).filter(**filters)
  82. stats = {
  83. 'total_orders': orders.count(),
  84. 'total_lines': Line.objects.filter(order__in=orders).count(),
  85. 'total_revenue': orders.aggregate(
  86. Sum('total_incl_tax'))['total_incl_tax__sum'] or D('0.00'),
  87. 'order_status_breakdown': orders.order_by('status').values(
  88. 'status').annotate(freq=Count('id'))
  89. }
  90. return stats
  91. class OrderListView(BulkEditMixin, ListView):
  92. """
  93. Dashboard view for a list of orders.
  94. Supports the permission-based dashboard.
  95. """
  96. model = Order
  97. context_object_name = 'orders'
  98. template_name = 'dashboard/orders/order_list.html'
  99. form_class = OrderSearchForm
  100. desc_template = _("%(main_filter)s %(name_filter)s %(title_filter)s"
  101. "%(upc_filter)s %(sku_filter)s %(date_filter)s"
  102. "%(voucher_filter)s %(payment_filter)s"
  103. "%(status_filter)s")
  104. paginate_by = 25
  105. description = ''
  106. actions = ('download_selected_orders', 'change_order_statuses')
  107. def dispatch(self, request, *args, **kwargs):
  108. # base_queryset is equal to all orders the user is allowed to access
  109. self.base_queryset = queryset_orders_for_user(
  110. request.user).order_by('-date_placed')
  111. return super(OrderListView, self).dispatch(request, *args, **kwargs)
  112. def get(self, request, *args, **kwargs):
  113. if 'order_number' in request.GET and request.GET.get(
  114. 'response_format', 'html') == 'html':
  115. # Redirect to Order detail page if valid order number is given
  116. try:
  117. order = self.base_queryset.get(
  118. number=request.GET['order_number'])
  119. except Order.DoesNotExist:
  120. pass
  121. else:
  122. return redirect(
  123. 'dashboard:order-detail', number=order.number)
  124. return super(OrderListView, self).get(request, *args, **kwargs)
  125. def get_desc_context(self, data=None): # noqa (too complex (16))
  126. """Update the title that describes the queryset"""
  127. desc_ctx = {
  128. 'main_filter': _('All orders'),
  129. 'name_filter': '',
  130. 'title_filter': '',
  131. 'upc_filter': '',
  132. 'sku_filter': '',
  133. 'date_filter': '',
  134. 'voucher_filter': '',
  135. 'payment_filter': '',
  136. 'status_filter': '',
  137. }
  138. if 'order_status' in self.request.GET:
  139. status = self.request.GET['order_status']
  140. if status.lower() == 'none':
  141. desc_ctx['main_filter'] = _("Orders without an order status")
  142. else:
  143. desc_ctx['main_filter'] = _("Orders with status '%s'") % status
  144. if data is None:
  145. return desc_ctx
  146. if data['order_number']:
  147. desc_ctx['main_filter'] = _('Orders with number starting with'
  148. ' "%(order_number)s"') % data
  149. if data['name']:
  150. desc_ctx['name_filter'] = _(" with customer name matching"
  151. " '%(name)s'") % data
  152. if data['product_title']:
  153. desc_ctx['title_filter'] \
  154. = _(" including an item with title matching"
  155. " '%(product_title)s'") % data
  156. if data['upc']:
  157. desc_ctx['upc_filter'] = _(" including an item with UPC"
  158. " '%(upc)s'") % data
  159. if data['partner_sku']:
  160. desc_ctx['upc_filter'] = _(" including an item with ID"
  161. " '%(partner_sku)s'") % data
  162. if data['date_from'] and data['date_to']:
  163. desc_ctx['date_filter'] \
  164. = _(" placed between %(start_date)s and %(end_date)s") \
  165. % {'start_date': format_datetime(data['date_from']),
  166. 'end_date': format_datetime(data['date_to'])}
  167. elif data['date_from']:
  168. desc_ctx['date_filter'] = _(" placed since %s") \
  169. % format_datetime(data['date_from'])
  170. elif data['date_to']:
  171. date_to = data['date_to'] + datetime.timedelta(days=1)
  172. desc_ctx['date_filter'] = _(" placed before %s") \
  173. % format_datetime(date_to)
  174. if data['voucher']:
  175. desc_ctx['voucher_filter'] = _(" using voucher '%(voucher)s'") \
  176. % data
  177. if data['payment_method']:
  178. desc_ctx['payment_filter'] = _(" paid for by %(payment_method)s") \
  179. % data
  180. if data['status']:
  181. desc_ctx['status_filter'] = _(" with status %(status)s") % data
  182. return desc_ctx
  183. def get_queryset(self): # noqa (too complex (19))
  184. """
  185. Build the queryset for this list.
  186. """
  187. queryset = sort_queryset(self.base_queryset, self.request,
  188. ['number', 'total_incl_tax'])
  189. # Look for shortcut query filters
  190. if 'order_status' in self.request.GET:
  191. self.form = self.form_class()
  192. status = self.request.GET['order_status']
  193. if status.lower() == 'none':
  194. status = None
  195. self.description = self.desc_template % self.get_desc_context()
  196. return self.base_queryset.filter(status=status)
  197. if 'order_number' not in self.request.GET:
  198. self.description = self.desc_template % self.get_desc_context()
  199. self.form = self.form_class()
  200. return queryset
  201. self.form = self.form_class(self.request.GET)
  202. if not self.form.is_valid():
  203. return queryset
  204. data = self.form.cleaned_data
  205. if data['order_number']:
  206. queryset = self.base_queryset.filter(
  207. number__istartswith=data['order_number'])
  208. if data['name']:
  209. # If the value is two words, then assume they are first name and
  210. # last name
  211. parts = data['name'].split()
  212. allow_anon = getattr(settings, 'OSCAR_ALLOW_ANON_CHECKOUT', False)
  213. if len(parts) == 1:
  214. parts = [data['name'], data['name']]
  215. else:
  216. parts = [parts[0], parts[1:]]
  217. filter = Q(user__first_name__istartswith=parts[0])
  218. filter |= Q(user__last_name__istartswith=parts[1])
  219. if allow_anon:
  220. filter |= Q(billing_address__first_name__istartswith=parts[0])
  221. filter |= Q(shipping_address__first_name__istartswith=parts[0])
  222. filter |= Q(billing_address__last_name__istartswith=parts[1])
  223. filter |= Q(shipping_address__last_name__istartswith=parts[1])
  224. queryset = queryset.filter(filter).distinct()
  225. if data['product_title']:
  226. queryset = queryset.filter(
  227. lines__title__istartswith=data['product_title']).distinct()
  228. if data['upc']:
  229. queryset = queryset.filter(lines__upc=data['upc'])
  230. if data['partner_sku']:
  231. queryset = queryset.filter(lines__partner_sku=data['partner_sku'])
  232. if data['date_from'] and data['date_to']:
  233. # Add 24 hours to make search inclusive
  234. date_to = data['date_to'] + datetime.timedelta(days=1)
  235. queryset = queryset.filter(date_placed__gte=data['date_from'])
  236. queryset = queryset.filter(date_placed__lt=date_to)
  237. elif data['date_from']:
  238. queryset = queryset.filter(date_placed__gte=data['date_from'])
  239. elif data['date_to']:
  240. date_to = data['date_to'] + datetime.timedelta(days=1)
  241. queryset = queryset.filter(date_placed__lt=date_to)
  242. if data['voucher']:
  243. queryset = queryset.filter(
  244. discounts__voucher_code=data['voucher']).distinct()
  245. if data['payment_method']:
  246. queryset = queryset.filter(
  247. sources__source_type__code=data['payment_method']).distinct()
  248. if data['status']:
  249. queryset = queryset.filter(status=data['status'])
  250. self.description = self.desc_template % self.get_desc_context(data)
  251. return queryset
  252. def get_context_data(self, **kwargs):
  253. ctx = super(OrderListView, self).get_context_data(**kwargs)
  254. ctx['queryset_description'] = self.description
  255. ctx['form'] = self.form
  256. ctx['order_statuses'] = Order.all_statuses()
  257. return ctx
  258. def is_csv_download(self):
  259. return self.request.GET.get('response_format', None) == 'csv'
  260. def get_paginate_by(self, queryset):
  261. return None if self.is_csv_download() else self.paginate_by
  262. def render_to_response(self, context, **response_kwargs):
  263. if self.is_csv_download():
  264. return self.download_selected_orders(
  265. self.request,
  266. context['object_list'])
  267. return super(OrderListView, self).render_to_response(
  268. context, **response_kwargs)
  269. def get_download_filename(self, request):
  270. return 'orders.csv'
  271. def download_selected_orders(self, request, orders):
  272. response = HttpResponse(content_type='text/csv')
  273. response['Content-Disposition'] = 'attachment; filename=%s' \
  274. % self.get_download_filename(request)
  275. writer = UnicodeCSVWriter(open_file=response)
  276. meta_data = (('number', _('Order number')),
  277. ('value', _('Order value')),
  278. ('date', _('Date of purchase')),
  279. ('num_items', _('Number of items')),
  280. ('status', _('Order status')),
  281. ('customer', _('Customer email address')),
  282. ('shipping_address_name', _('Deliver to name')),
  283. ('billing_address_name', _('Bill to name')),
  284. )
  285. columns = SortedDict()
  286. for k, v in meta_data:
  287. columns[k] = v
  288. writer.writerow(columns.values())
  289. for order in orders:
  290. row = columns.copy()
  291. row['number'] = order.number
  292. row['value'] = order.total_incl_tax
  293. row['date'] = format_datetime(order.date_placed, 'DATETIME_FORMAT')
  294. row['num_items'] = order.num_items
  295. row['status'] = order.status
  296. row['customer'] = order.email
  297. if order.shipping_address:
  298. row['shipping_address_name'] = order.shipping_address.name
  299. else:
  300. row['shipping_address_name'] = ''
  301. if order.billing_address:
  302. row['billing_address_name'] = order.billing_address.name
  303. else:
  304. row['billing_address_name'] = ''
  305. writer.writerow(row)
  306. return response
  307. def change_order_statuses(self, request, orders):
  308. for order in orders:
  309. self.change_order_status(request, order)
  310. return redirect('dashboard:order-list')
  311. def change_order_status(self, request, order):
  312. # This method is pretty similar to what
  313. # OrderDetailView.change_order_status does. Ripe for refactoring.
  314. new_status = request.POST['new_status'].strip()
  315. if not new_status:
  316. messages.error(request, _("The new status '%s' is not valid")
  317. % new_status)
  318. elif new_status not in order.available_statuses():
  319. messages.error(request, _("The new status '%s' is not valid for"
  320. " this order") % new_status)
  321. else:
  322. handler = EventHandler(request.user)
  323. old_status = order.status
  324. try:
  325. handler.handle_order_status_change(order, new_status)
  326. except PaymentError as e:
  327. messages.error(request, _("Unable to change order status due"
  328. " to payment error: %s") % e)
  329. else:
  330. msg = _("Order status changed from '%(old_status)s' to"
  331. " '%(new_status)s'") % {'old_status': old_status,
  332. 'new_status': new_status}
  333. messages.info(request, msg)
  334. order.notes.create(
  335. user=request.user, message=msg, note_type=OrderNote.SYSTEM)
  336. class OrderDetailView(DetailView):
  337. """
  338. Dashboard view to display a single order.
  339. Supports the permission-based dashboard.
  340. """
  341. model = Order
  342. context_object_name = 'order'
  343. template_name = 'dashboard/orders/order_detail.html'
  344. # These strings are method names that are allowed to be called from a
  345. # submitted form.
  346. order_actions = ('save_note', 'delete_note', 'change_order_status',
  347. 'create_order_payment_event')
  348. line_actions = ('change_line_statuses', 'create_shipping_event',
  349. 'create_payment_event')
  350. def get_object(self, queryset=None):
  351. return get_order_for_user_or_404(
  352. self.request.user, self.kwargs['number'])
  353. def post(self, request, *args, **kwargs):
  354. # For POST requests, we use a dynamic dispatch technique where a
  355. # parameter specifies what we're trying to do with the form submission.
  356. # We distinguish between order-level actions and line-level actions.
  357. order = self.object = self.get_object()
  358. # Look for order-level action first
  359. if 'order_action' in request.POST:
  360. return self.handle_order_action(
  361. request, order, request.POST['order_action'])
  362. # Look for line-level action
  363. if 'line_action' in request.POST:
  364. return self.handle_line_action(
  365. request, order, request.POST['line_action'])
  366. return self.reload_page(error=_("No valid action submitted"))
  367. def handle_order_action(self, request, order, action):
  368. if action not in self.order_actions:
  369. return self.reload_page(error=_("Invalid action"))
  370. return getattr(self, action)(request, order)
  371. def handle_line_action(self, request, order, action):
  372. if action not in self.line_actions:
  373. return self.reload_page(error=_("Invalid action"))
  374. # Load requested lines
  375. line_ids = request.POST.getlist('selected_line')
  376. if len(line_ids) == 0:
  377. return self.reload_page(error=_(
  378. "You must select some lines to act on"))
  379. lines = order.lines.filter(id__in=line_ids)
  380. if len(line_ids) != len(lines):
  381. return self.reload_page(error=_("Invalid lines requested"))
  382. # Build list of line quantities
  383. line_quantities = []
  384. for line in lines:
  385. qty = request.POST.get('selected_line_qty_%s' % line.id)
  386. try:
  387. qty = int(qty)
  388. except ValueError:
  389. qty = None
  390. if qty is None or qty <= 0:
  391. error_msg = _("The entered quantity for line #%s is not valid")
  392. return self.reload_page(error=error_msg % line.id)
  393. elif qty > line.quantity:
  394. error_msg = _(
  395. "The entered quantity for line #%(line_id)s "
  396. "should not be higher than %(quantity)s")
  397. kwargs = {'line_id': line.id, 'quantity': line.quantity}
  398. return self.reload_page(error=error_msg % kwargs)
  399. line_quantities.append(qty)
  400. return getattr(self, action)(
  401. request, order, lines, line_quantities)
  402. def reload_page(self, fragment=None, error=None):
  403. url = reverse('dashboard:order-detail',
  404. kwargs={'number': self.object.number})
  405. if fragment:
  406. url += '#' + fragment
  407. if error:
  408. messages.error(self.request, error)
  409. return HttpResponseRedirect(url)
  410. def get_context_data(self, **kwargs):
  411. ctx = super(OrderDetailView, self).get_context_data(**kwargs)
  412. ctx['active_tab'] = kwargs.get('active_tab', 'lines')
  413. # Forms
  414. ctx['note_form'] = self.get_order_note_form()
  415. ctx['order_status_form'] = self.get_order_status_form()
  416. ctx['line_statuses'] = Line.all_statuses()
  417. ctx['shipping_event_types'] = ShippingEventType.objects.all()
  418. ctx['payment_event_types'] = PaymentEventType.objects.all()
  419. ctx['payment_transactions'] = self.get_payment_transactions()
  420. return ctx
  421. # Data fetching methods for template context
  422. def get_payment_transactions(self):
  423. return Transaction.objects.filter(
  424. source__order=self.object)
  425. def get_order_note_form(self):
  426. kwargs = {
  427. 'order': self.object,
  428. 'user': self.request.user,
  429. 'data': None
  430. }
  431. if self.request.method == 'POST':
  432. kwargs['data'] = self.request.POST
  433. note_id = self.kwargs.get('note_id', None)
  434. if note_id:
  435. note = get_object_or_404(OrderNote, order=self.object, id=note_id)
  436. if note.is_editable():
  437. kwargs['instance'] = note
  438. return OrderNoteForm(**kwargs)
  439. def get_order_status_form(self):
  440. data = None
  441. if self.request.method == 'POST':
  442. data = self.request.POST
  443. return OrderStatusForm(order=self.object, data=data)
  444. # Order-level actions
  445. def save_note(self, request, order):
  446. form = self.get_order_note_form()
  447. if form.is_valid():
  448. form.save()
  449. messages.success(self.request, _("Note saved"))
  450. return self.reload_page(fragment='notes')
  451. ctx = self.get_context_data(note_form=form, active_tab='notes')
  452. return self.render_to_response(ctx)
  453. def delete_note(self, request, order):
  454. try:
  455. note = order.notes.get(id=request.POST.get('note_id', None))
  456. except ObjectDoesNotExist:
  457. messages.error(request, _("Note cannot be deleted"))
  458. else:
  459. messages.info(request, _("Note deleted"))
  460. note.delete()
  461. return self.reload_page()
  462. def change_order_status(self, request, order):
  463. form = self.get_order_status_form()
  464. if not form.is_valid():
  465. return self.reload_page(error=_("Invalid form submission"))
  466. old_status, new_status = order.status, form.cleaned_data['new_status']
  467. handler = EventHandler(request.user)
  468. success_msg = _(
  469. "Order status changed from '%(old_status)s' to "
  470. "'%(new_status)s'") % {'old_status': old_status,
  471. 'new_status': new_status}
  472. try:
  473. handler.handle_order_status_change(
  474. order, new_status, note_msg=success_msg)
  475. except PaymentError as e:
  476. messages.error(
  477. request, _("Unable to change order status due to "
  478. "payment error: %s") % e)
  479. except order_exceptions.InvalidOrderStatus as e:
  480. # The form should validate against this, so we should only end up
  481. # here during race conditions.
  482. messages.error(
  483. request, _("Unable to change order status as the requested "
  484. "new status is not valid"))
  485. else:
  486. messages.info(request, success_msg)
  487. return self.reload_page()
  488. def create_order_payment_event(self, request, order):
  489. """
  490. Create a payment event for the whole order
  491. """
  492. amount_str = request.POST.get('amount', None)
  493. try:
  494. amount = D(amount_str)
  495. except InvalidOperation:
  496. messages.error(request, _("Please choose a valid amount"))
  497. return self.reload_page()
  498. return self._create_payment_event(request, order, amount)
  499. # Line-level actions
  500. def change_line_statuses(self, request, order, lines, quantities):
  501. new_status = request.POST['new_status'].strip()
  502. if not new_status:
  503. messages.error(request, _("The new status '%s' is not valid")
  504. % new_status)
  505. return self.reload_page()
  506. errors = []
  507. for line in lines:
  508. if new_status not in line.available_statuses():
  509. errors.append(_("'%(status)s' is not a valid new status for"
  510. " line %(line_id)d") % {'status': new_status,
  511. 'line_id': line.id})
  512. if errors:
  513. messages.error(request, "\n".join(errors))
  514. return self.reload_page()
  515. msgs = []
  516. for line in lines:
  517. msg = _("Status of line #%(line_id)d changed from '%(old_status)s'"
  518. " to '%(new_status)s'") % {'line_id': line.id,
  519. 'old_status': line.status,
  520. 'new_status': new_status}
  521. msgs.append(msg)
  522. line.set_status(new_status)
  523. message = "\n".join(msgs)
  524. messages.info(request, message)
  525. order.notes.create(user=request.user, message=message,
  526. note_type=OrderNote.SYSTEM)
  527. return self.reload_page()
  528. def create_shipping_event(self, request, order, lines, quantities):
  529. code = request.POST['shipping_event_type']
  530. try:
  531. event_type = ShippingEventType._default_manager.get(code=code)
  532. except ShippingEventType.DoesNotExist:
  533. messages.error(request, _("The event type '%s' is not valid")
  534. % code)
  535. return self.reload_page()
  536. reference = request.POST.get('reference', None)
  537. try:
  538. EventHandler().handle_shipping_event(order, event_type, lines,
  539. quantities,
  540. reference=reference)
  541. except order_exceptions.InvalidShippingEvent as e:
  542. messages.error(request,
  543. _("Unable to create shipping event: %s") % e)
  544. except order_exceptions.InvalidStatus as e:
  545. messages.error(request,
  546. _("Unable to create shipping event: %s") % e)
  547. except PaymentError as e:
  548. messages.error(request, _("Unable to create shipping event due to"
  549. " payment error: %s") % e)
  550. else:
  551. messages.success(request, _("Shipping event created"))
  552. return self.reload_page()
  553. def create_payment_event(self, request, order, lines, quantities):
  554. """
  555. Create a payment event for a subset of order lines
  556. """
  557. amount_str = request.POST.get('amount', None)
  558. # If no amount passed, then we add up the total of the selected lines
  559. if not amount_str:
  560. amount = sum([line.line_price_incl_tax for line in lines])
  561. else:
  562. try:
  563. amount = D(amount_str)
  564. except InvalidOperation:
  565. messages.error(request, _("Please choose a valid amount"))
  566. return self.reload_page()
  567. return self._create_payment_event(request, order, amount, lines,
  568. quantities)
  569. def _create_payment_event(self, request, order, amount, lines=None,
  570. quantities=None):
  571. code = request.POST.get('payment_event_type')
  572. try:
  573. event_type = PaymentEventType._default_manager.get(code=code)
  574. except PaymentEventType.DoesNotExist:
  575. messages.error(
  576. request, _("The event type '%s' is not valid") % code)
  577. return self.reload_page()
  578. try:
  579. EventHandler().handle_payment_event(
  580. order, event_type, amount, lines, quantities)
  581. except PaymentError as e:
  582. messages.error(request, _("Unable to create payment event due to"
  583. " payment error: %s") % e)
  584. except order_exceptions.InvalidPaymentEvent as e:
  585. messages.error(
  586. request, _("Unable to create payment event: %s") % e)
  587. else:
  588. messages.info(request, _("Payment event created"))
  589. return self.reload_page()
  590. class LineDetailView(DetailView):
  591. """
  592. Dashboard view to show a single line of an order.
  593. Supports the permission-based dashboard.
  594. """
  595. model = Line
  596. context_object_name = 'line'
  597. template_name = 'dashboard/orders/line_detail.html'
  598. def get_object(self, queryset=None):
  599. order = get_order_for_user_or_404(self.request.user,
  600. self.kwargs['number'])
  601. try:
  602. return order.lines.get(pk=self.kwargs['line_id'])
  603. except self.model.DoesNotExist:
  604. raise Http404()
  605. def get_context_data(self, **kwargs):
  606. ctx = super(LineDetailView, self).get_context_data(**kwargs)
  607. ctx['order'] = self.object.order
  608. return ctx
  609. def get_changes_between_models(model1, model2, excludes=None):
  610. """
  611. Return a dict of differences between two model instances
  612. """
  613. if excludes is None:
  614. excludes = []
  615. changes = {}
  616. for field in model1._meta.fields:
  617. if (isinstance(field, (fields.AutoField,
  618. fields.related.RelatedField))
  619. or field.name in excludes):
  620. continue
  621. if field.value_from_object(model1) != field.value_from_object(model2):
  622. changes[field.verbose_name] = (field.value_from_object(model1),
  623. field.value_from_object(model2))
  624. return changes
  625. def get_change_summary(model1, model2):
  626. """
  627. Generate a summary of the changes between two address models
  628. """
  629. changes = get_changes_between_models(model1, model2, ['search_text'])
  630. change_descriptions = []
  631. for field, delta in changes.items():
  632. change_descriptions.append(_("%(field)s changed from '%(old_value)s'"
  633. " to '%(new_value)s'")
  634. % {'field': field,
  635. 'old_value': delta[0],
  636. 'new_value': delta[1]})
  637. return "\n".join(change_descriptions)
  638. class ShippingAddressUpdateView(UpdateView):
  639. """
  640. Dashboard view to update an order's shipping address.
  641. Supports the permission-based dashboard.
  642. """
  643. model = ShippingAddress
  644. context_object_name = 'address'
  645. template_name = 'dashboard/orders/shippingaddress_form.html'
  646. form_class = ShippingAddressForm
  647. def get_object(self, queryset=None):
  648. order = get_order_for_user_or_404(self.request.user,
  649. self.kwargs['number'])
  650. return get_object_or_404(self.model, order=order)
  651. def get_context_data(self, **kwargs):
  652. ctx = super(ShippingAddressUpdateView, self).get_context_data(**kwargs)
  653. ctx['order'] = self.object.order
  654. return ctx
  655. def form_valid(self, form):
  656. old_address = ShippingAddress.objects.get(id=self.object.id)
  657. response = super(ShippingAddressUpdateView, self).form_valid(form)
  658. changes = get_change_summary(old_address, self.object)
  659. if changes:
  660. msg = _("Delivery address updated:\n%s") % changes
  661. self.object.order.notes.create(user=self.request.user, message=msg,
  662. note_type=OrderNote.SYSTEM)
  663. return response
  664. def get_success_url(self):
  665. messages.info(self.request, _("Delivery address updated"))
  666. return reverse('dashboard:order-detail',
  667. kwargs={'number': self.object.order.number, })