417 lines
14 KiB
Python
417 lines
14 KiB
Python
from collections import namedtuple
|
|
from datetime import datetime
|
|
from datetime import timedelta
|
|
from decimal import Decimal
|
|
from django.contrib.auth.models import Group
|
|
from django.core.exceptions import ObjectDoesNotExist
|
|
from django.core.management.base import BaseCommand, CommandError
|
|
|
|
from registrasion.models import inventory as inv
|
|
from registrasion.models import conditions as cond
|
|
from symposion import proposals
|
|
|
|
class Command(BaseCommand):
|
|
help = 'Populates the inventory with the NBPy2017 inventory model'
|
|
|
|
def add_arguments(self, parser):
|
|
pass
|
|
|
|
def handle(self, *args, **options):
|
|
|
|
kinds = []
|
|
for i in ("talk", ):
|
|
kinds.append(proposals.models.ProposalKind.objects.get(name=i))
|
|
self.main_conference_proposals = kinds
|
|
|
|
self.populate_groups()
|
|
self.populate_inventory()
|
|
self.populate_restrictions()
|
|
self.populate_discounts()
|
|
|
|
def populate_groups(self):
|
|
self.group_team = self.find_or_make(
|
|
Group,
|
|
("name", ),
|
|
name="Conference organisers",
|
|
)
|
|
self.group_volunteers = self.find_or_make(
|
|
Group,
|
|
("name", ),
|
|
name="Conference volunteers",
|
|
)
|
|
self.group_unpublish = self.find_or_make(
|
|
Group,
|
|
("name", ),
|
|
name="Can see unpublished products",
|
|
)
|
|
|
|
def populate_inventory(self):
|
|
# Categories
|
|
|
|
self.ticket = self.find_or_make(
|
|
inv.Category,
|
|
("name",),
|
|
name="Ticket",
|
|
description="Each type of ticket has different included products. "
|
|
"For details of what products are included, see our "
|
|
"<a href='/attend'>ticket sales page</a>.",
|
|
required = True,
|
|
render_type=inv.Category.RENDER_TYPE_RADIO,
|
|
limit_per_user=1,
|
|
order=1,
|
|
)
|
|
self.t_shirt = self.find_or_make(
|
|
inv.Category,
|
|
("name",),
|
|
name="T-Shirt",
|
|
description="Commemorative conference t-shirts, featuring secret "
|
|
"North Bay Python 2017 artwork. Details of sizing and "
|
|
"manufacturer are on our <a href='/attend/tshirt'>"
|
|
"t-shirts page</a>",
|
|
required = False,
|
|
render_type=inv.Category.RENDER_TYPE_ITEM_QUANTITY,
|
|
order=40,
|
|
)
|
|
self.extras = self.find_or_make(
|
|
inv.Category,
|
|
("name",),
|
|
name="Extras",
|
|
description="Other items that can improve your conference "
|
|
"experience.",
|
|
required = False,
|
|
render_type=inv.Category.RENDER_TYPE_QUANTITY,
|
|
order=60,
|
|
)
|
|
|
|
# Tickets
|
|
|
|
self.ticket_ind_sponsor = self.find_or_make(
|
|
inv.Product,
|
|
("name", "category",),
|
|
category=self.ticket,
|
|
name="Individual Sponsor",
|
|
price=Decimal("500.00"),
|
|
reservation_duration=hours(24),
|
|
order=1,
|
|
)
|
|
self.ticket_corporate = self.find_or_make(
|
|
inv.Product,
|
|
("name", "category",),
|
|
category=self.ticket,
|
|
name="Corporate",
|
|
price=Decimal("200.00"),
|
|
reservation_duration=hours(24),
|
|
order=10,
|
|
)
|
|
self.ticket_supporter = self.find_or_make(
|
|
inv.Product,
|
|
("name", "category",),
|
|
category=self.ticket,
|
|
name="Individual Supporter",
|
|
price=Decimal("100.00"),
|
|
reservation_duration=hours(24),
|
|
order=20,
|
|
)
|
|
self.ticket_unaffiliated = self.find_or_make(
|
|
inv.Product,
|
|
("name", "category",),
|
|
category=self.ticket,
|
|
name="Unaffiliated Individual",
|
|
price=Decimal("50.00"),
|
|
reservation_duration=hours(24),
|
|
order=30,
|
|
)
|
|
self.ticket_speaker = self.find_or_make(
|
|
inv.Product,
|
|
("name", "category",),
|
|
category=self.ticket,
|
|
name="Speaker",
|
|
price=Decimal("00.00"),
|
|
reservation_duration=hours(24),
|
|
order=50,
|
|
)
|
|
self.ticket_media = self.find_or_make(
|
|
inv.Product,
|
|
("name", "category",),
|
|
category=self.ticket,
|
|
name="Media",
|
|
price=Decimal("00.00"),
|
|
reservation_duration=hours(24),
|
|
order=60,
|
|
)
|
|
self.ticket_sponsor = self.find_or_make(
|
|
inv.Product,
|
|
("name", "category",),
|
|
category=self.ticket,
|
|
name="Sponsor",
|
|
price=Decimal("00.00"),
|
|
reservation_duration=hours(24),
|
|
order=70,
|
|
)
|
|
self.ticket_team = self.find_or_make(
|
|
inv.Product,
|
|
("name", "category",),
|
|
category=self.ticket,
|
|
name="Conference Organiser",
|
|
price=Decimal("00.00"),
|
|
reservation_duration=hours(24),
|
|
order=80,
|
|
)
|
|
self.ticket_volunteer = self.find_or_make(
|
|
inv.Product,
|
|
("name", "category",),
|
|
category=self.ticket,
|
|
name="Conference Volunteer",
|
|
price=Decimal("00.00"),
|
|
reservation_duration=hours(24),
|
|
order=90,
|
|
)
|
|
|
|
# Shirts
|
|
ShirtGroup = namedtuple("ShirtGroup", ("prefix", "sizes"))
|
|
shirt_names = {
|
|
"mens": ShirtGroup(
|
|
"Men's/Straight Cut Size",
|
|
("S", "M", "L", "XL", "2XL", "3XL", "5XL"),
|
|
),
|
|
"womens_classic": ShirtGroup(
|
|
"Women's Relaxed Fit",
|
|
("XS", "S", "M", "L", "XL", "2XL", "3XL"),
|
|
),
|
|
"womens_semi": ShirtGroup(
|
|
"Women's Semi-Fitted",
|
|
("S", "M", "L", "XL", "2XL", "3XL"),
|
|
),
|
|
}
|
|
|
|
self.shirts = {}
|
|
order = 0
|
|
for name, group in shirt_names.items():
|
|
self.shirts[name] = {}
|
|
prefix = group.prefix
|
|
for size in group.sizes:
|
|
product_name = "%s %s" % (prefix, size)
|
|
order += 10
|
|
self.shirts[name][size] = self.find_or_make(
|
|
inv.Product,
|
|
("name", "category",),
|
|
name=product_name,
|
|
category=self.t_shirt,
|
|
price=Decimal("30.00"),
|
|
reservation_duration=hours(1),
|
|
order=order,
|
|
)
|
|
|
|
def populate_restrictions(self):
|
|
|
|
# Hide the products that will eventually need a voucher
|
|
hide_voucher_products = self.find_or_make(
|
|
cond.GroupMemberFlag,
|
|
("description", ),
|
|
description="Can see hidden products",
|
|
condition=cond.FlagBase.ENABLE_IF_TRUE,
|
|
)
|
|
hide_voucher_products.group.set([self.group_unpublish])
|
|
hide_voucher_products.products.set([
|
|
self.ticket_media,
|
|
self.ticket_sponsor,
|
|
])
|
|
|
|
# Set limits.
|
|
public_ticket_cap = self.find_or_make(
|
|
cond.TimeOrStockLimitFlag,
|
|
("description", ),
|
|
description="Public ticket cap",
|
|
condition=cond.FlagBase.DISABLE_IF_FALSE,
|
|
limit=350,
|
|
)
|
|
public_ticket_cap.products.set([
|
|
self.ticket_ind_sponsor,
|
|
self.ticket_corporate,
|
|
self.ticket_supporter,
|
|
self.ticket_unaffiliated,
|
|
])
|
|
|
|
non_public_ticket_cap = self.find_or_make(
|
|
cond.TimeOrStockLimitFlag,
|
|
("description", ),
|
|
description="Non-public ticket cap",
|
|
condition=cond.FlagBase.DISABLE_IF_FALSE,
|
|
limit=200,
|
|
)
|
|
non_public_ticket_cap.products.set([
|
|
self.ticket_speaker,
|
|
self.ticket_sponsor,
|
|
self.ticket_media,
|
|
self.ticket_team,
|
|
self.ticket_volunteer,
|
|
])
|
|
|
|
# Volunteer tickets are for volunteers only
|
|
volunteers = self.find_or_make(
|
|
cond.GroupMemberFlag,
|
|
("description", ),
|
|
description="Volunteer tickets",
|
|
condition=cond.FlagBase.ENABLE_IF_TRUE,
|
|
)
|
|
volunteers.group.set([self.group_volunteers])
|
|
volunteers.products.set([
|
|
self.ticket_volunteer,
|
|
])
|
|
|
|
# Team tickets are for team members only
|
|
team = self.find_or_make(
|
|
cond.GroupMemberFlag,
|
|
("description", ),
|
|
description="Team tickets",
|
|
condition=cond.FlagBase.ENABLE_IF_TRUE,
|
|
)
|
|
team.group.set([self.group_team])
|
|
team.products.set([
|
|
self.ticket_team,
|
|
])
|
|
|
|
# Speaker tickets are for primary speakers only
|
|
speaker_tickets = self.find_or_make(
|
|
cond.SpeakerFlag,
|
|
("description", ),
|
|
description="Speaker tickets",
|
|
condition=cond.FlagBase.ENABLE_IF_TRUE,
|
|
is_presenter=True,
|
|
is_copresenter=False,
|
|
)
|
|
speaker_tickets.proposal_kind.set(self.main_conference_proposals)
|
|
speaker_tickets.products.set([self.ticket_speaker, ])
|
|
|
|
def populate_discounts(self):
|
|
|
|
def add_early_birds(discount):
|
|
self.find_or_make(
|
|
cond.DiscountForProduct,
|
|
("discount", "product"),
|
|
discount=discount,
|
|
product=self.ticket_ind_sponsor,
|
|
price=Decimal("50.00"),
|
|
quantity=1, # Per user
|
|
)
|
|
self.find_or_make(
|
|
cond.DiscountForProduct,
|
|
("discount", "product"),
|
|
discount=discount,
|
|
product=self.ticket_corporate,
|
|
price=Decimal("20.00"),
|
|
quantity=1, # Per user
|
|
)
|
|
self.find_or_make(
|
|
cond.DiscountForProduct,
|
|
("discount", "product"),
|
|
discount=discount,
|
|
product=self.ticket_supporter,
|
|
price=Decimal("20.00"),
|
|
quantity=1, # Per user
|
|
)
|
|
self.find_or_make(
|
|
cond.DiscountForProduct,
|
|
("discount", "product"),
|
|
discount=discount,
|
|
product=self.ticket_unaffiliated,
|
|
price=Decimal("25.00"),
|
|
quantity=1, # Per user
|
|
)
|
|
|
|
def free_category(parent_discount, category, quantity=1):
|
|
self.find_or_make(
|
|
cond.DiscountForCategory,
|
|
("discount", "category",),
|
|
discount=parent_discount,
|
|
category=category,
|
|
percentage=Decimal("100.00"),
|
|
quantity=quantity,
|
|
)
|
|
|
|
# Early Bird Discount (general public)
|
|
early_bird = self.find_or_make(
|
|
cond.TimeOrStockLimitDiscount,
|
|
("description", ),
|
|
description="Early Bird",
|
|
end_time=datetime(year=2017, month=10, day=20),
|
|
limit=100, # Across all users
|
|
)
|
|
add_early_birds(early_bird)
|
|
|
|
# Early bird rates for speakers
|
|
speaker_ticket_discounts = self.find_or_make(
|
|
cond.SpeakerDiscount,
|
|
("description", ),
|
|
description="Speaker Ticket Discount",
|
|
is_presenter=True,
|
|
is_copresenter=True,
|
|
)
|
|
speaker_ticket_discounts.proposal_kind.set(
|
|
self.main_conference_proposals,
|
|
)
|
|
add_early_birds(speaker_ticket_discounts)
|
|
|
|
# Professional-Like ticket inclusions
|
|
ticket_prolike_inclusions = self.find_or_make(
|
|
cond.IncludedProductDiscount,
|
|
("description", ),
|
|
description="Complimentary for ticket holder (Supporter-level and above)",
|
|
)
|
|
ticket_prolike_inclusions.enabling_products.set([
|
|
self.ticket_ind_sponsor,
|
|
self.ticket_corporate,
|
|
self.ticket_supporter,
|
|
self.ticket_sponsor,
|
|
self.ticket_speaker,
|
|
])
|
|
free_category(ticket_prolike_inclusions, self.t_shirt)
|
|
|
|
# Team & volunteer ticket inclusions
|
|
ticket_staff_inclusions = self.find_or_make(
|
|
cond.IncludedProductDiscount,
|
|
("description", ),
|
|
description="Complimentary for ticket holder (staff/volunteer)",
|
|
)
|
|
ticket_staff_inclusions.enabling_products.set([
|
|
self.ticket_team,
|
|
self.ticket_volunteer,
|
|
])
|
|
|
|
# Team & volunteer t-shirts, regardless of ticket type
|
|
staff_t_shirts = self.find_or_make(
|
|
cond.GroupMemberDiscount,
|
|
("description", ),
|
|
description="T-shirts complimentary for staff and volunteers",
|
|
)
|
|
staff_t_shirts.group.set([
|
|
self.group_team,
|
|
self.group_volunteers,
|
|
])
|
|
free_category(staff_t_shirts, self.t_shirt, quantity=2)
|
|
|
|
def find_or_make(self, model, search_keys, **k):
|
|
''' Either makes or finds an object of type _model_, with the given
|
|
kwargs.
|
|
|
|
Arguments:
|
|
search_keys ([str, ...]): A sequence of keys that are used to search
|
|
for an existing version in the database. The remaining arguments are
|
|
only used when creating a new object.
|
|
'''
|
|
|
|
try:
|
|
keys = dict((key, k[key]) for key in search_keys)
|
|
a = model.objects.get(**keys)
|
|
self.stdout.write("FOUND : " + str(keys))
|
|
model.objects.filter(id=a.id).update(**k)
|
|
a.refresh_from_db()
|
|
return a
|
|
except ObjectDoesNotExist:
|
|
a = model.objects.create(**k)
|
|
self.stdout.write("CREATED: " + str(k))
|
|
return a
|
|
|
|
|
|
def hours(n):
|
|
return timedelta(hours=n)
|