def _create_product_variation(row):
"""
Manually create all objects for a new product option.
This includes (if not existing):
Option
OptionGroup
Product
ConfigurableProduct
ProductVariation
You could also use the create_subs flag from ConfigurableProduct which
creates the variations automatically for you, but you're then loosing the
ability to manually set the SKU etc.
"""
# get option group or create if not existing
try:
group = OptionGroup.objects.get(name=row[4])
except OptionGroup.DoesNotExist:
group = OptionGroup(site=Site.objects.get(id=1))
group.name = row[4]
group.description = ''
group.save()
# create the new option
option = Option(option_group=group)
option.name = row[3]
option.value = row[3]
option.sort_order = int(row[1])
option.save()
# now first create a Product...
product = _create_product(row)
# variants dont need a category, otherwise they show up as normal products
product.category.clear()
# then get or create the ConfigurableProduct and add the option group
# somehow the root product to which all variations belong must be
# identified, in this case the variation id is 00
main_product = Product.objects.get(sku='%s' % row[5])
try:
cp = ConfigurableProduct.objects.get(product=main_product)
except ConfigurableProduct.DoesNotExist:
cp = ConfigurableProduct(product=main_product)
cp.save()
cp.option_group.add(group)
cp.save()
# finally create the ProductVariation and add the option
variation = ProductVariation(product=product)
variation.parent = cp
variation.save()
variation.options.add(option)
variation.save()
def _create_product_variation(row):
"""
Manually create all objects for a new product option.
This includes (if not existing):
Option
OptionGroup
Product
ConfigurableProduct
ProductVariation
You could also use the create_subs flag from ConfigurableProduct which
creates the variations automatically for you, but you're then loosing the
ability to manually set the SKU etc.
"""
# get option group or create if not existing
try:
group = OptionGroup.objects.get(name=row[4])
except OptionGroup.DoesNotExist:
group = OptionGroup(site=Site.objects.get(id=1))
group.name = row[4]
group.description = ''
group.save()
# create the new option
option = Option(option_group=group)
option.name = row[3]
option.value = row[3]
option.sort_order = int(row[1])
option.save()
# now first create a Product...
product = _create_product(row)
# variants dont need a category, otherwise they show up as normal products
product.category.clear()
# then get or create the ConfigurableProduct and add the option group
# somehow the root product to which all variations belong must be
# identified, in this case the variation id is 00
main_product = Product.objects.get(sku='%s' % row[5])
try:
cp = ConfigurableProduct.objects.get(product=main_product)
except ConfigurableProduct.DoesNotExist:
cp = ConfigurableProduct(product=main_product)
cp.save()
cp.option_group.add(group)
cp.save()
# finally create the ProductVariation and add the option
variation = ProductVariation(product=product)
variation.parent = cp
variation.save()
variation.options.add(option)
variation.save()
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 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 get_product(self, day, schedule=None, tour_time=None, create=True):
"""
Get the TourProduct for the given day and time. If create is false and product does
not exist, return None.
"""
if schedule and tour_time:
assert False, 'Make up your mind, use a schedule or a tour_time, not both'
if not schedule and not tour_time:
assert False, 'Need a time buddy'
try:
if schedule:
tour_product = TourProduct.objects.filter(tour_type=self, day=day).filter(
Q(schedule=schedule) | Q(tour_time=schedule.tour_time))[0]
else:
tour_product = TourProduct.objects.get(tour_type=self, day=day, tour_time=tour_time)
if tour_product.tour_type.has_price_raise():
p = tour_product.product.price_set.all()[0]
p.price = self.get_raised_base_price(tour_product.day)
p.save()
return tour_product
except (TourProduct.DoesNotExist, IndexError):
if create:
# print 'creating product... ', day
product = Product()
product.site = Site.objects.get_current()
product.name = '%s on %s at ' % (self.name, day.strftime('%B %d, %Y'))
if schedule:
product.name += schedule.tour_time.strftime('%I:%M %p')
else:
product.name += tour_time.strftime('%I:%M %p')
product.items_in_stock = self.get_capacity_for_day(day)
product.shipclass = 'NO' # see product.models.SHIP_CLASS_CHOICES
product.save()
tour_product = TourProduct()
tour_product.product = product
tour_product.tour_type = self
tour_product.day = day
if schedule:
tour_product.schedule = schedule
tour_product.tour_time = schedule.tour_time
else:
tour_product.schedule = None
tour_product.tour_time = tour_time
tour_product.save()
p = Price()
p.product = product
if tour_product.tour_type.has_price_raise():
p.price = self.get_raised_base_price(tour_product.day)
else:
p.price = self.base_price
p.save()
# Create Variations
if self.option_group_id:
configurable_product = ConfigurableProduct(product=product)
configurable_product.save()
configurable_product.option_group.add(self.option_group)
configurable_product.create_subs = True
configurable_product.save()
return tour_product
else:
return None
