def make_test_order(country, state, site=None, orderitems=None):
if not orderitems:
orderitems = [
('formaggio-fresco-di-carmasciano', 5),
]
if not site:
site = Site.objects.get_current()
c = Contact(first_name="Order", last_name="Tester",
role=ContactRole.objects.get(pk='Customer'), email="*****@*****.**")
c.save()
if not isinstance(country, Country):
country = Country.objects.get(iso2_code__iexact = country)
ad = AddressBook(contact=c, description="home",
street1 = "via prova 123", state=state, city="Napoli",
country = country, is_default_shipping=True,
is_default_billing=True)
ad.save()
o = Order(contact=c, shipping_cost=Decimal('10.00'), site = site)
o.save()
for slug, qty in orderitems:
p = Product.objects.get(slug=slug)
price = p.unit_price
item = OrderItem(order=o, product=p, quantity=qty,
unit_price=price, line_item_price=price*qty)
item.save()
return o
def make_test_order(country, state, include_non_taxed=False, site=None):
if not site:
site = Site.objects.get_current()
c = Contact(first_name="Tax", last_name="Tester", role="Customer", email="*****@*****.**")
c.save()
if not isinstance(country, Country):
country = Country.objects.get(iso2_code__iexact=country)
ad = AddressBook(
contact=c,
description="home",
street1="test",
state=state,
city="Portland",
country=country,
is_default_shipping=True,
is_default_billing=True,
)
ad.save()
o = Order(contact=c, shipping_cost=Decimal("10.00"), site=site)
o.save()
p = Product.objects.get(slug="dj-rocks-s-b")
price = p.unit_price
item1 = OrderItem(order=o, product=p, quantity="5", unit_price=price, line_item_price=price * 5)
item1.save()
if include_non_taxed:
p = Product.objects.get(slug="neat-book-hard")
price = p.unit_price
item2 = OrderItem(order=o, product=p, quantity="1", unit_price=price, line_item_price=price)
item2.save()
return o
def make_test_order(country, state):
c = Contact(first_name="Gift", last_name="Tester",
role=ContactRole.objects.get(pk='Customer'), email="*****@*****.**")
c.save()
if not isinstance(country, Country):
country = Country.objects.get(iso2_code__iexact = country)
ad = AddressBook(contact=c, description="home",
street1 = "test", state=state, city="Portland",
country = country, is_default_shipping=True,
is_default_billing=True)
ad.save()
site = Site.objects.get_current()
o = Order(contact=c, shipping_cost=Decimal('0.00'), site=site)
o.save()
p = Product.objects.get(slug='GIFT10')
price = p.unit_price
log.debug("creating with price: %s", price)
item1 = OrderItem(order=o, product=p, quantity='2.0',
unit_price=price, line_item_price=price*2)
item1.save()
detl = OrderItemDetail(name = 'email', value='*****@*****.**', sort_order=0, item=item1)
detl.save()
detl = OrderItemDetail(name = 'message', value='hello there', sort_order=0, item=item1)
detl.save()
return o
def save(self, contact, address_entry=None, **kwargs):
data = self.cleaned_data.copy()
if not address_entry:
address_entry = AddressBook()
log.debug('creating new AddressBook entry')
else:
address_entry = address_entry
log.debug('Saving Addressbook info for %s', address_entry)
for field in data.keys():
# Getting around the issue where we normally want this auto created on the front end
if field <> 'addressee_name':
setattr(address_entry, field, data[field])
address_entry.addressee = data['addressee_name']
address_entry.contact = contact
address_entry.save()
def make_test_order(site, orderitems, shipping_cost='10.00', contact_info={}):
default_contact_info = {
'first_name': "Order",
'last_name': "Tester",
'role': ContactRole.objects.get(pk='Customer'),
'email': "*****@*****.**",
'address': {
'description': 'home',
'street1': 'Via delle Chiaie 21',
'city': 'Napoli',
'state': 'NA',
'country': Country.objects.get(iso2_code__iexact='IT'),
'is_default_shipping': True,
'is_default_billing': True
}
}
default_contact_info.update(contact_info)
address_info = default_contact_info.pop('address')
contact = Contact(**default_contact_info)
contact.save()
address_info['contact'] = contact
address_book = AddressBook(**address_info)
address_book.save()
order = Order(
contact=contact,
shipping_cost=Decimal(shipping_cost),
site = site
)
order.save()
for slug, quantity in orderitems:
product = Product.objects.get(slug=slug)
price = product.unit_price
item = OrderItem(
order=order,
product=product,
quantity=quantity,
unit_price=price,
line_item_price=price*quantity
)
item.save()
return order
def make_test_order(country, state):
c = Contact(first_name="Gift",
last_name="Tester",
role=ContactRole.objects.get(pk='Customer'),
email="*****@*****.**")
c.save()
if not isinstance(country, Country):
country = Country.objects.get(iso2_code__iexact=country)
ad = AddressBook(contact=c,
description="home",
street1="test",
state=state,
city="Portland",
country=country,
is_default_shipping=True,
is_default_billing=True)
ad.save()
site = Site.objects.get_current()
o = Order(contact=c, shipping_cost=Decimal('0.00'), site=site)
o.save()
p = Product.objects.get(slug='GIFT10')
price = p.unit_price
log.debug("creating with price: %s", price)
item1 = OrderItem(order=o,
product=p,
quantity='2.0',
unit_price=price,
line_item_price=price * 2)
item1.save()
detl = OrderItemDetail(name='email',
value='*****@*****.**',
sort_order=0,
item=item1)
detl.save()
detl = OrderItemDetail(name='message',
value='hello there',
sort_order=0,
item=item1)
detl.save()
return o
def make_test_order(country, state, site=None, orderitems=None):
if not orderitems:
orderitems = [
('formaggio-fresco-di-carmasciano', 5),
]
if not site:
site = Site.objects.get_current()
c = Contact(first_name="Order",
last_name="Tester",
role=ContactRole.objects.get(pk='Customer'),
email="*****@*****.**")
c.save()
if not isinstance(country, Country):
country = Country.objects.get(iso2_code__iexact=country)
ad = AddressBook(contact=c,
description="home",
street1="via prova 123",
state=state,
city="Napoli",
country=country,
is_default_shipping=True,
is_default_billing=True)
ad.save()
o = Order(contact=c, shipping_cost=Decimal('10.00'), site=site)
o.save()
for slug, qty in orderitems:
p = Product.objects.get(slug=slug)
price = p.unit_price
item = OrderItem(order=o,
product=p,
quantity=qty,
unit_price=price,
line_item_price=price * qty)
item.save()
return o
def make_test_order(site, orderitems, shipping_cost='10.00', contact_info={}):
default_contact_info = {
'first_name': "Order",
'last_name': "Tester",
'role': ContactRole.objects.get(pk='Customer'),
'email': "*****@*****.**",
'address': {
'description': 'home',
'street1': 'Via delle Chiaie 21',
'city': 'Napoli',
'state': 'NA',
'country': Country.objects.get(iso2_code__iexact='IT'),
'is_default_shipping': True,
'is_default_billing': True
}
}
default_contact_info.update(contact_info)
address_info = default_contact_info.pop('address')
contact = Contact(**default_contact_info)
contact.save()
address_info['contact'] = contact
address_book = AddressBook(**address_info)
address_book.save()
order = Order(contact=contact,
shipping_cost=Decimal(shipping_cost),
site=site)
order.save()
for slug, quantity in orderitems:
product = Product.objects.get(slug=slug)
price = product.unit_price
item = OrderItem(order=order,
product=product,
quantity=quantity,
unit_price=price,
line_item_price=price * quantity)
item.save()
return order
def setUp(self):
self.US = Country.objects.get(iso2_code__iexact="US")
self.site = Site.objects.get_current()
tax = config_get("TAX", "MODULE")
tax.update("tax.modules.no")
c = Contact(first_name="Jim", last_name="Tester", role="Customer", email="*****@*****.**")
c.save()
ad = AddressBook(
contact=c,
description="home",
street1="test",
state="OR",
city="Portland",
country=self.US,
is_default_shipping=True,
is_default_billing=True,
)
ad.save()
o = Order(contact=c, shipping_cost=Decimal("6.00"), site=self.site)
o.save()
small = Order(contact=c, shipping_cost=Decimal("6.00"), site=self.site)
small.save()
p = Product.objects.get(slug="neat-book-soft")
price = p.unit_price
item1 = OrderItem(order=o, product=p, quantity="1", unit_price=price, line_item_price=price)
item1.save()
item1s = OrderItem(order=small, product=p, quantity="1", unit_price=price, line_item_price=price)
item1s.save()
p = Product.objects.get(slug="neat-book-hard")
price = p.unit_price
item2 = OrderItem(order=o, product=p, quantity="1", unit_price=price, line_item_price=price)
item2.save()
self.order = o
self.small = small
def save(self, contact, address_entry=None, **kwargs):
data = self.cleaned_data.copy()
if not address_entry:
address_entry = AddressBook()
log.debug('creating new AddressBook entry')
else:
address_entry = address_entry
log.debug('Saving Addressbook info for %s', address_entry)
for field in data.keys():
# Getting around the issue where we normally want this auto created on the front end
if not field == 'addressee_name':
setattr(address_entry, field, data[field])
address_entry.addressee = data['addressee_name']
address_entry.contact = contact
address_entry.save()
def save_info(self, contact=None, **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
organization_name = data.pop('organization', None)
if organization_name:
org = Organization.objects.by_name(organization_name, create=True)
customer.organization = org
else:
# in case customer wants to remove organization name from their profile
customer.organization = None
for field in customer.__dict__.keys():
try:
setattr(customer, field, data[field])
except KeyError:
pass
if not customer.role:
customer.role = ContactRole.objects.get(pk='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_id',
'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:
ship_field = 'ship_' + field
if (not changed_location) and field in ('state', 'country_id',
'city'):
if getattr(ship_address, field) != data[ship_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()
form_postsave.send(ContactInfoForm,
object=customer,
formdata=data,
form=self)
if changed_location:
signals.satchmo_contact_location_changed.send(self,
contact=customer)
return customer.id
def handle_noargs(self, **options):
from satchmo_store.contact.models import (
AddressBook,
Contact,
ContactOrganization,
ContactOrganizationRole,
ContactRole,
Organization,
PhoneNumber,
)
from product.models import (
Category,
OptionGroup,
Option,
Price,
Product,
)
from product.modules.configurable.models import ConfigurableProduct
from satchmo_store.shop.models import Config
from django.conf import settings
from l10n.models import Country
from django.contrib.sites.models import Site
from django.contrib.auth.models import User
#idempotency test
print "Checking for existing sample data."
try:
p = Product.objects.get(slug="dj-rocks")
print "It looks like you already have loaded the sample store data, quitting."
import sys
sys.exit(1)
except Product.DoesNotExist:
pass
print "Loading sample store data."
#Load basic configuration information
print "Creating site..."
try:
site = Site.objects.get(id=settings.SITE_ID)
print "Using existing site #%i" % settings.SITE_ID
except Site.DoesNotExist:
print "Creating Example Store Site"
site = Site(domain="localhost", name="Sample Store")
site.domain = settings.SITE_DOMAIN
site.name = settings.SITE_NAME
site.save()
try:
store_country = Country.objects.get(iso3_code='USA')
except Country.DoesNotExist:
print "\nError: Country data should be first loaded by: python manage.py satchmo_load_l10n"
import sys
sys.exit(1)
config = Config(site=site,
store_name=settings.SITE_NAME,
country=store_country,
sales_country=store_country)
config.save()
config.shipping_countries.add(store_country)
config.save()
print "Creating Customers..."
# Import some customers
customer = ContactRole.objects.get(pk='Customer')
c1 = Contact(first_name="Chris",
last_name="Smith",
email="*****@*****.**",
role=customer,
notes="Really cool stuff")
c1.save()
p1 = PhoneNumber(contact=c1,
phone="601-555-5511",
type="Home",
primary=True)
p1.save()
c2 = Contact(first_name="John",
last_name="Smith",
email="*****@*****.**",
role=customer,
notes="Second user")
c2.save()
p2 = PhoneNumber(contact=c2,
phone="999-555-5111",
type="Work",
primary=True)
p2.save()
# Import some addresses for these customers
us = Country.objects.get(iso2_code='US')
a1 = AddressBook(description="Home",
street1="8235 Pike Street",
city="Anywhere Town",
state="TN",
postal_code="38138",
country=us,
is_default_shipping=True,
contact=c1)
a1.save()
a2 = AddressBook(description="Work",
street1="1245 Main Street",
city="Stillwater",
state="MN",
postal_code="55082",
country=us,
is_default_shipping=True,
contact=c2)
a2.save()
print "Creating Suppliers..."
#Import some suppliers
supplier = ContactOrganizationRole.objects.get(pk='Supplier')
company = ContactOrganization.objects.get(pk='Company')
contactsupplier = ContactRole.objects.get(pk='Supplier')
org1 = Organization(name="Rhinestone Ronny",
type=company,
role=supplier)
org1.save()
c4 = Contact(first_name="Fred",
last_name="Jones",
email="*****@*****.**",
role=contactsupplier,
organization=org1)
c4.save()
p4 = PhoneNumber(contact=c4,
phone="800-188-7611",
type="Work",
primary=True)
p4.save()
p5 = PhoneNumber(contact=c4, phone="755-555-1111", type="Fax")
p5.save()
a3 = AddressBook(contact=c4,
description="Mailing address",
street1="Receiving Dept",
street2="918 Funky Town St",
city="Fishkill",
state="NJ",
country=us,
postal_code="19010")
a3.save()
#s1 = Supplier(name="Rhinestone Ronny", address1="918 Funky Town St", address2="Suite 200",
# city="Fishkill", state="NJ", zip="19010", phone1="800-188-7611", fax="900-110-1909", email="*****@*****.**",
# notes="My main supplier")
#s1.save()
#s2 = Supplier(name="Shirt Sally", address1="9 ABC Lane",
# city="Happyville", state="MD", zip="190111", phone1="888-888-1111", fax="999-110-1909", email="*****@*****.**",
# notes="Shirt Supplier")
#s2.save()
print "Creating Categories..."
#Create some categories
cat1 = Category(site=site,
name="Shirts",
slug="shirts",
description="Women's Shirts")
cat1.save()
cat2 = Category(site=site,
name="Short Sleeve",
slug="shortsleeve",
description="Short sleeve shirts",
parent=cat1)
cat2.save()
cat3 = Category(site=site,
name="Books",
slug="book",
description="Books")
cat3.save()
cat4 = Category(site=site,
name="Fiction",
slug="fiction",
description="Fiction Books",
parent=cat3)
cat4.save()
cat5 = Category(site=site,
name="Science Fiction",
slug="scifi",
description="Science Fiction",
parent=cat4)
cat5.save()
cat6 = Category(site=site,
name="Non Fiction",
slug="nonfiction",
description="Non Fiction",
parent=cat3)
cat6.save()
cat7 = Category(site=site, name="Software", slug="software")
cat7.save()
print "Creating products..."
#Create some items
i1 = Product(site=site,
name="Django Rocks shirt",
slug="dj-rocks",
description="Really cool shirt",
active=True,
featured=True)
i1.save()
p1 = Price(price="20.00", product=i1)
p1.save()
i1.category.add(cat1)
i1.save()
i2 = Product(
site=site,
name="Python Rocks shirt",
slug="PY-Rocks",
description="Really cool python shirt - One Size Fits All",
active=True,
featured=True)
i2.save()
p2 = Price(price="19.50", product=i2)
p2.save()
i2.category.add(cat2)
i2.save()
i3 = Product(site=site,
name="A really neat book",
slug="neat-book",
description="A neat book. You should buy it.",
active=True,
featured=True)
i3.save()
p3 = Price(price="5.00", product=i3)
p3.save()
i3.category.add(cat4)
i3.save()
i4 = Product(site=site,
name="Robots Attack!",
slug="robot-attack",
description="Robots try to take over the world.",
active=True,
featured=True)
i4.save()
p4 = Price(price="7.99", product=i4)
p4.save()
i4.category.add(cat5)
i4.save()
# i5 = Product(site=site, name="Really Neat Software", slug="neat-software", description="Example Configurable/Downloadable product", active=True, featured=True)
# i5.save()
# i5.category.add(cat7)
# i5.save()
#Create an attribute set
optSet1 = OptionGroup(site=site, name="sizes", sort_order=1)
optSet2 = OptionGroup(site=site, name="colors", sort_order=2)
optSet1.save()
optSet2.save()
optSet3 = OptionGroup(site=site, name="Book type", sort_order=1)
optSet3.save()
optSet4 = OptionGroup(site=site, name="Full/Upgrade", sort_order=5)
optSet4.save()
optItem1a = Option(name="Small",
value="S",
sort_order=1,
option_group=optSet1)
optItem1a.save()
optItem1b = Option(name="Medium",
value="M",
sort_order=2,
option_group=optSet1)
optItem1b.save()
optItem1c = Option(name="Large",
value="L",
sort_order=3,
price_change="1.00",
option_group=optSet1)
optItem1c.save()
optItem2a = Option(name="Black",
value="B",
sort_order=1,
option_group=optSet2)
optItem2a.save()
optItem2b = Option(name="White",
value="W",
sort_order=2,
option_group=optSet2)
optItem2b.save()
optItem2c = Option(name="Blue",
value="BL",
sort_order=3,
price_change="2.00",
option_group=optSet2)
optItem2c.save()
optItem3a = Option(name="Hard cover",
value="hard",
sort_order=1,
option_group=optSet3)
optItem3a.save()
optItem3b = Option(name="Soft cover",
value="soft",
sort_order=2,
price_change="1.00",
option_group=optSet3)
optItem3b.save()
optItem3c = Option(name="On tape",
value="tape",
sort_order=3,
option_group=optSet3)
optItem3c.save()
optItem4a = Option(name="Full Version",
value="full",
option_group=optSet4,
sort_order=1)
optItem4a.save()
optItem4b = Option(name="Upgrade Version",
value="upgrade",
option_group=optSet4,
sort_order=2)
optItem4b.save()
#Add the option group to our items
pg1 = ConfigurableProduct(product=i1)
pg1.save()
pg1.option_group.add(optSet1)
pg1.save()
pg1.option_group.add(optSet2)
pg1.save()
pg3 = ConfigurableProduct(product=i3)
pg3.save()
pg3.option_group.add(optSet3)
pg3.save()
pg4 = ConfigurableProduct(product=i4)
pg4.save()
pg4.option_group.add(optSet3)
pg4.save()
# pg5 = ConfigurableProduct(product=i5)
# pg5.option_group.add(optSet4)
# pg5.save()
print "Creating product variations..."
#Create the required sub_items
pg1.create_all_variations()
pg3.create_all_variations()
pg4.create_all_variations()
#pg5.create_all_variations()
#set prices for full and upgrade versions of neat-software, this is an alternative to using the price_change in options, it allows for more flexability when required.
# pv1 = pg5.get_product_from_options([optItem4a])
# Price(product=pv1, price='5.00').save()
# Price(product=pv1, price='2.00', quantity='50.00').save()
# DownloadableProduct(product=pv1).save()
# pv2 = pg5.get_product_from_options([optItem4b])
# Price(product=pv2, price='1.00').save()
# DownloadableProduct(product=pv2).save()
print "Create a test user..."
#First see if our test user is still there, then use or create that user
try:
test_user = User.objects.get(username="******")
except:
test_user = User.objects.create_user('csmith',
'*****@*****.**', 'test')
test_user.save()
c1.user = test_user
c1.save()
def save_info(self, contact=None, **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
organization_name = data.pop('organization', None)
if organization_name:
org = Organization.objects.by_name(organization_name, create=True)
customer.organization = org
else:
# in case customer wants to remove organization name from their profile
customer.organization = None
for field in customer.__dict__.keys():
try:
setattr(customer, field, data[field])
except KeyError:
pass
if not customer.role:
customer.role = ContactRole.objects.get(pk='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_id', '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
try:
setattr(ship_address, "addressee",data.get('ship_addressee', ""))
setattr(bill_address, "addressee",data.get('addressee', ""))
except AttributeError:
pass
# If we are copying the address and one isn't in place for shipping
# copy it
if not getattr(ship_address, "addressee", False) and copy_address:
try:
ship_address.addressee = bill_address.addressee
except AttributeError:
pass
# Make sure not to overwrite a custom ship to name
if copy_address and getattr(ship_address, "addressee", "") == getattr(bill_address, "addressee", ""):
# 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 we have different ship to and bill to names, preserve them
if not copy_address or getattr(ship_address, "addressee", "") != getattr(bill_address, "addressee", ""):
if not ship_address or ship_address.id == bill_address.id:
ship_address = AddressBook()
for field in address_keys:
ship_field = 'ship_' + field
if (not changed_location) and field in ('state', 'country_id', 'city'):
if getattr(ship_address, field) != data[ship_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()
form_postsave.send(ContactInfoForm, object=customer, formdata=data, form=self)
if changed_location:
signals.satchmo_contact_location_changed.send(self, contact=customer)
return customer.id
def handle_noargs(self, **options):
from satchmo_store.contact.models import (
AddressBook,
Contact,
ContactOrganization,
ContactOrganizationRole,
ContactRole,
Organization,
PhoneNumber,
)
from product.models import (
Category,
OptionGroup,
Option,
Price,
Product,
)
from product.modules.configurable.models import ConfigurableProduct
from satchmo_store.shop.models import Config
from django.conf import settings
from l10n.models import Country
from django.contrib.sites.models import Site
from django.contrib.auth.models import User
#idempotency test
print "Checking for existing sample data."
try:
p = Product.objects.get(slug="dj-rocks")
print "It looks like you already have loaded the sample store data, quitting."
import sys
sys.exit(1)
except Product.DoesNotExist:
pass
print "Loading sample store data."
#Load basic configuration information
print "Creating site..."
try:
site = Site.objects.get(id=settings.SITE_ID)
print "Using existing site #%i" % settings.SITE_ID
except Site.DoesNotExist:
print "Creating Example Store Site"
site = Site(domain="localhost", name="Sample Store")
site.domain = settings.SITE_DOMAIN
site.name = settings.SITE_NAME
site.save()
try:
store_country = Country.objects.get(iso3_code='USA')
except Country.DoesNotExist:
print "\nError: Country data should be first loaded by: python manage.py satchmo_load_l10n"
import sys
sys.exit(1)
config = Config(site=site, store_name=settings.SITE_NAME, country=store_country, sales_country=store_country)
config.save()
config.shipping_countries.add(store_country)
config.save()
print "Creating Customers..."
# Import some customers
customer = ContactRole.objects.get(pk='Customer')
c1 = Contact(first_name="Chris", last_name="Smith", email="*****@*****.**", role=customer, notes="Really cool stuff")
c1.save()
p1 = PhoneNumber(contact=c1, phone="601-555-5511", type="Home",primary=True)
p1.save()
c2 = Contact(first_name="John", last_name="Smith", email="*****@*****.**", role=customer, notes="Second user")
c2.save()
p2 = PhoneNumber(contact=c2, phone="999-555-5111", type="Work",primary=True)
p2.save()
# Import some addresses for these customers
us = Country.objects.get(iso2_code='US');
a1 = AddressBook(description="Home", street1="8235 Pike Street", city="Anywhere Town", state="TN",
postal_code="38138", country=us, is_default_shipping=True, contact=c1)
a1.save()
a2 = AddressBook(description="Work", street1="1245 Main Street", city="Stillwater", state="MN",
postal_code="55082", country=us, is_default_shipping=True, contact=c2)
a2.save()
print "Creating Suppliers..."
#Import some suppliers
supplier = ContactOrganizationRole.objects.get(pk='Supplier')
company = ContactOrganization.objects.get(pk='Company')
contactsupplier = ContactRole.objects.get(pk='Supplier')
org1 = Organization(name="Rhinestone Ronny", type=company, role=supplier)
org1.save()
c4 = Contact(first_name="Fred", last_name="Jones", email="*****@*****.**",
role=contactsupplier, organization=org1)
c4.save()
p4 = PhoneNumber(contact=c4,phone="800-188-7611", type="Work", primary=True)
p4.save()
p5 = PhoneNumber(contact=c4,phone="755-555-1111",type="Fax")
p5.save()
a3 = AddressBook(contact=c4, description="Mailing address", street1="Receiving Dept",
street2="918 Funky Town St", city="Fishkill",
state="NJ", country=us, postal_code="19010")
a3.save()
#s1 = Supplier(name="Rhinestone Ronny", address1="918 Funky Town St", address2="Suite 200",
# city="Fishkill", state="NJ", zip="19010", phone1="800-188-7611", fax="900-110-1909", email="*****@*****.**",
# notes="My main supplier")
#s1.save()
#s2 = Supplier(name="Shirt Sally", address1="9 ABC Lane",
# city="Happyville", state="MD", zip="190111", phone1="888-888-1111", fax="999-110-1909", email="*****@*****.**",
# notes="Shirt Supplier")
#s2.save()
print "Creating Categories..."
#Create some categories
'''cat1 = Category(site=site, name="Shirts",slug="shirts",description="Women's Shirts")
cat1.save()
cat2 = Category(site=site, name="Short Sleeve",slug="shortsleeve",description="Short sleeve shirts", parent=cat1)
cat2.save()'''
cat3 = Category(name="Books",slug="book",description="Books")
cat3.save()
cat3.site.add(site)
'''cat4 = Category(site=site, name="Fiction",slug="fiction",description="Fiction Books", parent=cat3)
cat4.save()
cat5 = Category(site=site, name="Science Fiction",slug="scifi",description="Science Fiction",parent=cat4)
cat5.save()
cat6 = Category(site=site, name="Non Fiction",slug="nonfiction",description="Non Fiction",parent=cat3)
cat6.save()
cat7 = Category(site=site, name="Software", slug="software")
cat7.save()'''
print "Creating products..."
#Create some items
'''i1 = Product(name="Django Rocks shirt", slug="dj-rocks", description="Really cool shirt",
active=True, featured=True)
i1.save()
p1 = Price(price="20.00", product=i1)
p1.save()
i1.category.add(cat1)
i1.save()
i2 = Product(name="Python Rocks shirt", slug="PY-Rocks",
description="Really cool python shirt - One Size Fits All",
active=True, featured=True)
i2.save()
p2 = Price(price="19.50", product=i2)
p2.save()
i2.category.add(cat2)
i2.save()'''
i3 = Product(name="Purusha Suktam", slug="Advyata",
description="A neat book. You should buy it.",
active=True, featured=True)
i3.save()
i3.site.add(site)
p3 = Price(price="5.00", product=i3)
p3.save()
i3.category.add(cat3)
i3.save()
i3.site.add(site)
'''i4 = Product(name="Robots Attack!", slug="robot-attack",
description="Robots try to take over the world.",
active=True, featured=True)
i4.save()
p4 = Price(price="7.99", product=i4)
p4.save()
i4.category.add(cat5)
i4.save()
# i5 = Product(site=site, name="Really Neat Software", slug="neat-software", description="Example Configurable/Downloadable product", active=True, featured=True)
# i5.save()
# i5.category.add(cat7)
# i5.save()
#Create an attribute set
optSet1 = OptionGroup(site=site, name="sizes", sort_order=1)
optSet2 = OptionGroup(site=site, name="colors", sort_order=2)
optSet1.save()
optSet2.save()'''
optSet3 = OptionGroup(name="Book type", sort_order=1)
optSet3.save()
optSet3.site.add(site)
'''optSet4 = OptionGroup(site=site, name="Full/Upgrade", sort_order=5)
optSet4.save()
optItem1a = Option(name="Small", value="S", sort_order=1, option_group=optSet1)
optItem1a.save()
optItem1b = Option(name="Medium", value="M", sort_order=2, option_group=optSet1)
optItem1b.save()
optItem1c = Option(name="Large", value="L", sort_order=3, price_change="1.00", option_group=optSet1)
optItem1c.save()
optItem2a = Option(name="Black", value="B", sort_order=1, option_group=optSet2)
optItem2a.save()
optItem2b = Option(name="White", value="W", sort_order=2, option_group=optSet2)
optItem2b.save()
optItem2c = Option(name="Blue", value="BL", sort_order=3, price_change="2.00", option_group=optSet2)
optItem2c.save()'''
optItem3a = Option(name="Hard cover", value="hard", sort_order=1, option_group=optSet3)
optItem3a.save()
optItem3b = Option(name="Soft cover", value="soft", sort_order=2, price_change="1.00", option_group=optSet3)
optItem3b.save()
optItem3c = Option(name="On tape", value="tape", sort_order=3, option_group=optSet3)
optItem3c.save()
'''optItem4a = Option(name="Full Version", value="full", option_group=optSet4, sort_order=1)
optItem4a.save()
optItem4b = Option(name="Upgrade Version", value="upgrade", option_group=optSet4, sort_order=2)
optItem4b.save()'''
#Add the option group to our items
'''pg1 = ConfigurableProduct(product=i1)
pg1.save()
pg1.option_group.add(optSet1)
pg1.save()
pg1.option_group.add(optSet2)
pg1.save()'''
pg3 = ConfigurableProduct(product=i3)
pg3.save()
pg3.option_group.add(optSet3)
pg3.save()
'''pg4 = ConfigurableProduct(product=i4)
pg4.save()
pg4.option_group.add(optSet3)
pg4.save()'''
# pg5 = ConfigurableProduct(product=i5)
# pg5.option_group.add(optSet4)
# pg5.save()
print "Creating product variations..."
#Create the required sub_items
#pg1.create_all_variations()
pg3.create_all_variations()
#pg4.create_all_variations()
#pg5.create_all_variations()
#set prices for full and upgrade versions of neat-software, this is an alternative to using the price_change in options, it allows for more flexability when required.
# pv1 = pg5.get_product_from_options([optItem4a])
# Price(product=pv1, price='5.00').save()
# Price(product=pv1, price='2.00', quantity='50.00').save()
# DownloadableProduct(product=pv1).save()
# pv2 = pg5.get_product_from_options([optItem4b])
# Price(product=pv2, price='1.00').save()
# DownloadableProduct(product=pv2).save()
print "Create a test user..."
#First see if our test user is still there, then use or create that user
try:
test_user = User.objects.get(username="******")
except:
test_user = User.objects.create_user('csmith', '*****@*****.**', 'test')
test_user.save()
c1.user = test_user
c1.save()
def save_info(self, contact=None, **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 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
