def save(self, contact=None, update_newsletter=True):
"""Save the contact info into the database.
Checks to see if contact exists. If not, creates a contact
and copies in the address and phone number."""
if not contact:
customer = Contact()
else:
customer = contact
data = self.cleaned_data
for field in customer.__dict__.keys():
try:
setattr(customer, field, data[field])
except KeyError:
pass
if update_newsletter and config_get_group('NEWSLETTER'):
from satchmo.newsletter import update_subscription
if 'newsletter' not in data:
subscribed = False
else:
subscribed = data['newsletter']
update_subscription(contact, subscribed)
if not customer.role:
customer.role = "Customer"
customer.save()
# we need to make sure we don't blindly add new addresses
# this isn't ideal, but until we have a way to manage addresses
# this will force just the two addresses, shipping and billing
# TODO: add address management like Amazon.
bill_address = customer.billing_address
if not bill_address:
bill_address = AddressBook(contact=customer)
address_keys = bill_address.__dict__.keys()
for field in address_keys:
try:
setattr(bill_address, field, data[field])
except KeyError:
pass
bill_address.is_default_billing = True
copy_address = data['copy_address']
ship_address = customer.shipping_address
if copy_address:
# make sure we don't have any other default shipping address
if ship_address and ship_address.id != bill_address.id:
ship_address.delete()
bill_address.is_default_shipping = True
bill_address.save()
if not copy_address:
if not ship_address or ship_address.id == bill_address.id:
ship_address = AddressBook()
for field in address_keys:
try:
setattr(ship_address, field, data['ship_' + field])
except KeyError:
pass
ship_address.is_default_shipping = True
ship_address.is_default_billing = False
ship_address.contact = customer
ship_address.country = bill_address.country
ship_address.save()
if not customer.primary_phone:
phone = PhoneNumber()
phone.primary = True
else:
phone = customer.primary_phone
phone.phone = data['phone']
phone.contact = customer
phone.save()
return customer.id
def register_handle_form(request, redirect=None):
"""
Handle all registration logic. This is broken out from "register" to allow easy overriding/hooks
such as a combined login/register page.
Returns:
- Success flag
- HTTPResponseRedirect (success) or form (fail)
"""
if request.method == 'POST':
form = RegistrationForm(request.POST)
if form.is_valid():
data = form.cleaned_data
password = data['password']
email = data['email']
first_name = data['first_name']
last_name = data['last_name']
username = generate_id(first_name, last_name)
verify = (config_value('SHOP', 'ACCOUNT_VERIFICATION') == 'EMAIL')
if verify:
from registration.models import RegistrationProfile
user = RegistrationProfile.objects.create_inactive_user(
username, password, email, send_email=True)
else:
user = User.objects.create_user(username, email, password)
user.first_name = first_name
user.last_name = last_name
user.save()
# If the user already has a contact, retrieve it.
# Otherwise, create a new one.
contact = Contact.from_request(request, create=False)
if contact is None:
contact = Contact()
contact.user = user
contact.first_name = first_name
contact.last_name = last_name
contact.email = email
contact.role = 'Customer'
contact.save()
if config_get_group('NEWSLETTER'):
from satchmo.newsletter import update_subscription
if 'newsletter' not in data:
subscribed = False
else:
subscribed = data['newsletter']
update_subscription(contact, subscribed)
if not verify:
user = authenticate(username=username, password=password)
login(request, user)
send_welcome_email(email, first_name, last_name)
if not redirect:
redirect = urlresolvers.reverse('registration_complete')
return (True, http.HttpResponseRedirect(redirect))
else:
initial_data = {}
contact = Contact.from_request(request, create=False)
if contact is not None:
initial_data = {
'email': contact.email,
'first_name': contact.first_name,
'last_name': contact.last_name}
if contact and config_get_group('NEWSLETTER'):
from satchmo.newsletter import is_subscribed
current_subscriber = is_subscribed(contact)
else:
current_subscriber = False
initial_data['newsletter'] = current_subscriber
form = RegistrationForm(initial=initial_data)
return (False, form)
def save(self, state=_NOTSET, attributes={}):
contact = self.get_contact()
if state == _NOTSET:
state = self.cleaned_data['subscribed']
return update_subscription(contact, state, attributes=attributes)
def register_handle_form(request, redirect=None):
"""
Handle all registration logic. This is broken out from "register" to allow easy overriding/hooks
such as a combined login/register page.
Returns:
- Success flag
- HTTPResponseRedirect (success) or form (fail)
"""
if request.method == 'POST':
form = RegistrationForm(request.POST)
if form.is_valid():
data = form.cleaned_data
password = data['password']
email = data['email']
first_name = data['first_name']
last_name = data['last_name']
username = generate_id(first_name, last_name)
verify = (config_value('SHOP', 'ACCOUNT_VERIFICATION') == 'EMAIL')
if verify:
from registration.models import RegistrationProfile
user = RegistrationProfile.objects.create_inactive_user(
username, password, email, send_email=True)
else:
user = User.objects.create_user(username, email, password)
user.first_name = first_name
user.last_name = last_name
user.save()
# If the user already has a contact, retrieve it.
# Otherwise, create a new one.
contact = Contact.from_request(request, create=False)
if contact is None:
contact = Contact()
contact.user = user
contact.first_name = first_name
contact.last_name = last_name
contact.email = email
contact.role = 'Customer'
contact.save()
if config_get_group('NEWSLETTER'):
from satchmo.newsletter import update_subscription
if 'newsletter' not in data:
subscribed = False
else:
subscribed = data['newsletter']
update_subscription(contact, subscribed)
if not verify:
user = authenticate(username=username, password=password)
login(request, user)
send_welcome_email(email, first_name, last_name)
if not redirect:
redirect = urlresolvers.reverse('registration_complete')
return (True, http.HttpResponseRedirect(redirect))
else:
initial_data = {}
contact = Contact.from_request(request, create=False)
if contact is not None:
initial_data = {
'email': contact.email,
'first_name': contact.first_name,
'last_name': contact.last_name
}
if contact and config_get_group('NEWSLETTER'):
from satchmo.newsletter import is_subscribed
current_subscriber = is_subscribed(contact)
else:
current_subscriber = False
initial_data['newsletter'] = current_subscriber
form = RegistrationForm(initial=initial_data)
return (False, form)
def save_info(self, contact=None, update_newsletter=True, **kwargs):
"""Save the contact info into the database.
Checks to see if contact exists. If not, creates a contact
and copies in the address and phone number."""
if not contact:
customer = Contact()
log.debug('creating new contact')
else:
customer = contact
log.debug('Saving contact info for %s', contact)
data = self.cleaned_data.copy()
country = data['country']
if not isinstance(country, Country):
country = Country.objects.get(pk=country)
data['country'] = country
data['country_id'] = country.id
shipcountry = data['ship_country']
if not isinstance(shipcountry, Country):
shipcountry = Country.objects.get(pk=shipcountry)
data['ship_country'] = shipcountry
data['ship_country_id'] = shipcountry.id
companyname = data.pop('company', None)
if companyname:
org = Organization.objects.by_name(companyname, create=True)
customer.organization = org
for field in customer.__dict__.keys():
try:
setattr(customer, field, data[field])
except KeyError:
pass
if update_newsletter and config_get_group('NEWSLETTER'):
from satchmo.newsletter import update_subscription
if 'newsletter' not in data:
subscribed = False
else:
subscribed = data['newsletter']
update_subscription(contact, subscribed)
if not customer.role:
customer.role = "Customer"
customer.save()
# we need to make sure we don't blindly add new addresses
# this isn't ideal, but until we have a way to manage addresses
# this will force just the two addresses, shipping and billing
# TODO: add address management like Amazon.
bill_address = customer.billing_address
if not bill_address:
bill_address = AddressBook(contact=customer)
changed_location = False
address_keys = bill_address.__dict__.keys()
for field in address_keys:
if (not changed_location) and field in ('state', 'country', 'city'):
if getattr(bill_address, field) != data[field]:
changed_location = True
try:
setattr(bill_address, field, data[field])
except KeyError:
pass
bill_address.is_default_billing = True
copy_address = data['copy_address']
ship_address = customer.shipping_address
if copy_address:
# make sure we don't have any other default shipping address
if ship_address and ship_address.id != bill_address.id:
ship_address.delete()
bill_address.is_default_shipping = True
bill_address.save()
if not copy_address:
if not ship_address or ship_address.id == bill_address.id:
ship_address = AddressBook()
for field in address_keys:
if (not changed_location) and field in ('state', 'country', 'city'):
if getattr(ship_address, field) != data[field]:
changed_location = True
try:
setattr(ship_address, field, data['ship_' + field])
except KeyError:
pass
ship_address.is_default_shipping = True
ship_address.is_default_billing = False
ship_address.contact = customer
ship_address.save()
if not customer.primary_phone:
phone = PhoneNumber()
phone.primary = True
else:
phone = customer.primary_phone
phone.phone = data['phone']
phone.contact = customer
phone.save()
signals.form_save.send(ContactInfoForm, object=customer, formdata=data, form=self)
if changed_location:
signals.satchmo_contact_location_changed.send(self, contact=customer)
return customer.id
