B `0@sddlZddlmZddlmZddlmZddlmZm Z ddl m Z ddl m Z mZe dd Ze dd Ze d d Zed dZed dZe ddZed dZed dZed dZeddZeddZedZGdddeZdS)N)Site)ObjectDoesNotExist)HttpResponseRedirect)NoReverseMatchreverse) post_checkout) get_class get_modelz order.utils OrderCreatorOrderDispatcherzcheckout.sessionCheckoutSessionMixinorderBillingAddressShippingAddressOrderNumberGeneratorPaymentEventType PaymentEventPaymentEventQuantityaddress UserAddressbasketBasketzoscar.checkoutc@seZdZdZdZdZeZddZddZ d.dd Z d d Z d/d d Z d0ddZ ddZddZd1ddZddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-ZdS)2OrderPlacementMixinz Mixin which provides functionality for placing orders. Any view class which needs to place an order should use this mixin. NcKsdS)a Handle any payment processing and record payment sources and events. This method is designed to be overridden within your project. The default is to do nothing as payment is domain-specific. This method is responsible for handling payment and recording the payment sources (using the add_payment_source method) and payment events (using add_payment_event) so they can be linked to the order when it is saved later on. N)self order_numbertotalkwargsrrI/home/dcms/DCMS/lib/python3.7/site-packages/oscar/apps/checkout/mixins.pyhandle_payment0s z"OrderPlacementMixin.handle_paymentcCs |jdkrg|_|j|dS)z8 Record a payment source for this order N)_payment_sourcesappend)rsourcerrradd_payment_source>s z&OrderPlacementMixin.add_payment_sourcecCs@tjj|d\}}|jdkr"g|_t|||d}|j|dS)zN Record a payment event for creation once the order is placed )nameN) event_typeamount reference)robjectsZ get_or_create_payment_eventsrr!)rZevent_type_namer'r(r&__eventrrradd_payment_eventFs  z%OrderPlacementMixin.add_payment_eventcCs t|S)z+ Return a new order number )rr)rrrrrgenerate_order_numberWsz)OrderPlacementMixin.generate_order_numberc Ks6|jf||||||||| d | } ||| S)a  Write out the order models and return the appropriate HTTP response We deliberately pass the basket in here as the one tied to the request isn't necessarily the correct one to use in placing the order. This can happen when a basket gets frozen. ) ruserrshipping_addressshipping_methodshipping_charge order_totalbilling_address surcharges) place_orderZsubmithandle_successful_order) rrr/rr0r1r2r4r3r5rr rrrhandle_order_placement]s z*OrderPlacementMixin.handle_order_placementc Ks|||}|j|||f| }d| kr2||} n | d} d| krRt|dd} n | d} tjf||||||||| | | d | } || | S)zM Writes the order out to the DB including the payment models statusrequestN) r/rrr0r1r2rr4r9r:r5)create_shipping_addresscreate_billing_addressget_initial_order_statuspopgetattrr r6save_payment_details)rrr/rr0r1r2r3r4r5rr9r:r rrrr6ps0     zOrderPlacementMixin.place_ordercCs&|sdS||jr"||||S)z Create and return the shipping address for the current order. Compared to self.get_shipping_address(), ShippingAddress is saved and makes sure that appropriate UserAddress exists. N)saveis_authenticatedupdate_address_book)rr/r0rrrr;s  z+OrderPlacementMixin.create_shipping_addresscCs|y|jj|d}Wn(tk r>t|d}||YnXt|trX|jd7_t|t rp|j d7_ | dS)zR Update the user's address book based on the new shipping address )hash)r/N) addressesgetZ generate_hashrrZpopulate_alternative_model isinstancerZnum_orders_as_shipping_addressrZnum_orders_as_billing_addressrA)rr/addrZ user_addrrrrrCs   z'OrderPlacementMixin.update_address_bookcKs&|sdS||jr"||||S)zK Saves any relevant billing data (e.g. a billing address). N)rArBrC)rr/r4r0rrrrr<s  z*OrderPlacementMixin.create_billing_addresscCs||||dS)z Saves all payment-related details. This could include a billing address, payment sources and any order payment events. N)save_payment_eventssave_payment_sources)rr rrrr@s z(OrderPlacementMixin.save_payment_detailscCsT|js dSxD|jD]:}||_|x&|jD]}tjj|||jdq0WqWdS)zB Saves any relevant payment events for this order N)r,linequantity) r*r rAlinesallrr)createrM)rr r,rLrrrrJs z'OrderPlacementMixin.save_payment_eventscCs,|js dSx|jD]}||_|qWdS)z Saves any payment sources used in this order. When the payment sources are created, the order model does not exist and so they need to have it set before saving. N)r r rA)rr r"rrrrKs  z(OrderPlacementMixin.save_payment_sourcescCsdS)Nr)rrrrrr=sz,OrderPlacementMixin.get_initial_order_statuscCsB|||j|j|jjd<t|}||j|||S)z Handle the various steps required after an order has been successfully placed. Override this view if you want to perform custom actions when an order is submitted. Zcheckout_order_id) send_order_placed_emailcheckout_sessionflushidr:sessionrget_success_url send_signal)rr responserrrr7s   z+OrderPlacementMixin.handle_successful_ordercCs|jj|||j||ddS)N)Zsenderr r/r:rX) view_signalsendr/)rr:rXr rrrrWszOrderPlacementMixin.send_signalcCstdS)Nzcheckout:thank-you)r)rrrrrVsz#OrderPlacementMixin.get_success_urlcCs$||}ttd}|||dS)N)logger)get_message_contextr r[Z send_order_placed_email_for_user)rr Z extra_context dispatcherrrrrQ s  z+OrderPlacementMixin.send_order_placed_emailcCs|jj||jd}y:|jjjr6tdd|jid}ntd|j|dd}Wntk rdYn"Xt j |j}d|j |f|d<|S) N)r/r rNzcustomer:orderr)rzcustomer:anon-order)rrDz http://%s%sZ status_url) r:r/rNrOrBrnumberZverification_hashrrr)Z get_currentdomain)rr ctxpathsiterrrr\s z'OrderPlacementMixin.get_message_contextcCs|j}tjj|dS)N)pk)rRZget_submitted_basket_idrZ_default_managerrG)rZ basket_idrrrget_submitted_basket+s z(OrderPlacementMixin.get_submitted_basketcCs |dS)zC Freeze the basket so it can no longer be modified N)freeze)rrrrr freeze_basket/sz!OrderPlacementMixin.freeze_basketcCsby |}Wntjk r"Yns&