3

\G                 @   st   d dl mZ d dlmZ d dlmZ d dlmZ eddZeddZ	G dd	 d	ej
Z
G d
d dejZdd ZdS )    )forms)	get_model)settingsaddressCountryorderZshippingaddressc                   s2   e Zd Z fddZdd ZG dd dZ  ZS )ShippingAddressFormc                s2   t  j|| | j  d| jd _d| jd _d S )NTphone_numberpostcode)super__init__adjust_country_fieldfieldsrequired)selfargskwargs)	__class__ ./var/www/html/oscar2019/apps/checkout/forms.pyr      s    zShippingAddressForm.__init__c             C   sR   t jjdd}t|dkr6| jjdd  |d | j_n|| jd _d | jd _	d S )NT)is_shipping_country   countryr   )
r   _default_managerfilterlenr   popinstancer   querysetempty_label)r   	countriesr   r   r   r      s    z(ShippingAddressForm.adjust_country_fieldc            
   @   s(   e Zd ZeZddddddddd	d
g
ZdS )zShippingAddressForm.Metatitle
first_name	last_nameline1line2line4stater
   r   r	   N)__name__
__module____qualname__ShippingAddressmodelr   r   r   r   r   Meta   s
   r-   )r(   r)   r*   r   r   r-   __classcell__r   r   )r   r   r   
   s   r   c               @   s&   e Zd ZdZejdejej dZ	dS )PaymentMethodFormz3
    Extra form for the custom payment method.
    zSelect a payment method)labelchoiceswidgetN)
r(   r)   r*   __doc__r   ChoiceFieldr   OSCAR_PAYMENT_METHODSRadioSelectpayment_methodr   r   r   r   r/   (   s
   r/   c             C   s   t tjj| S )N)dictr   r5   get)r7   r   r   r   get_payment_method_display3   s    r:   N)oscar.apps.checkoutr   
base_formsoscar.core.loadingr   djangodjango.confr   r   r+   r   Formr/   r:   r   r   r   r   <module>   s   

