def save_inventory(self):
prod = Product.load(self.request.POST.get('id'))
self.forbid_if(not prod or prod.company.enterprise_id != self.enterprise_id)
InventoryJournal.create_new(prod, 'Inventory Adjust', float(self.request.POST.get('inventory', 0)))
prod.name = self.request.POST.get('name')
prod.manufacturer = self.request.POST.get('manufacturer')
prod.unit_cost = util.nvl(self.request.POST.get('unit_cost'), 0.0)
prod.sku = self.request.POST.get('sku')
prod.inventory_par = util.nvl(self.request.POST.get('inventory_par'), None)
prod.save()
# save all the prices, prefixed by "cmp_"
for k in self.request.POST.keys():
if k.startswith('cmp_'):
match = re.search(r'^.*_(.*)', k)
if match:
campaign = Campaign.load(match.group(1))
price = self.request.POST.get(k)
if price:
price = util.float_(price)
prod.set_price_only(campaign, price)
else:
prod.remove_price(campaign)
prod.invalidate_caches(campaign_id=campaign.campaign_id)
return 'True'
def test_util(self):
d8e = util.today_date()
dtime = util.today()
assert util.format_rss_date(d8e) == d8e.strftime("%a, %d %b %Y %H:%M:%S EST")
assert util.words_date(dtime) == dtime.strftime("%B %d, %Y")
assert util.is_empty(" ") == True
assert util.float_("8") == None
assert util.page_list([1, 2, 3, 4, 5, 6, 7, 8, 9], 2, 2) == [3, 4]
assert util.page_list([1, 2, 3, 4, 5, 6, 7, 8, 9], None, None) == [1, 2, 3, 4, 5, 6, 7, 8, 9]
assert util.parse_date("2012-05-06") == datetime.datetime.strptime("2012-05-06", "%Y-%m-%d")
today_ = datetime.date.today()
assert [today_.year + 10, today_.year + 10] in util.year_list()
assert util.month_list()[0] == ["1", "January"]
assert util.this_year() == datetime.date.today().year
assert util.get_first_day(today_) == util.get_first_day(
today_
) # this is pretty dumb. it works, just get it covered.
assert util.get_last_day(today_) == util.get_last_day(today_)
assert util.to_uuid("ken") == None
assert int(util.average([1, 2, 3])) == 2
assert util.format_date(util.truncate_datetime(dtime)) == util.str_today()
assert util.is_today(d8e) == True
def save(self): #pylint: disable-msg=R0912,R0915
product_id = self.request.POST.get('product_id')
if product_id:
prod = Product.load(product_id)
else:
prod = Product()
prod.bind(self.request.POST, True)
prod.mod_dt = util.now()
prod.save()
self.db_flush()
new_children = {}
for k in self.request.POST.keys():
if k.startswith('campaign_price'):
match = re.search(r'^.*\[(.*)\]', k)
if match:
campaign = Campaign.load(match.group(1))
price = self.request.POST.get(k)
discount = self.request.POST.get('campaign_discount[%s]' % campaign.campaign_id)
if price:
price = util.float_(price)
discount = util.float_(util.nvl(discount, 0.0))
prod.set_price(campaign, price, discount)
else:
prod.remove_price(campaign)
if k.startswith('child_incl'):
child_product_id = self.request.POST.get(k)
child_product_quantity = self.request.POST.get('child_quantity_%s' % child_product_id)
new_children[child_product_id] = child_product_quantity
# KB: [2013-02-23]: Clear out old children that were just deselected and add the ones that are still selected.
for current_child in prod.get_children():
if current_child.child_id not in new_children.keys():
prod.clear_child(current_child.child_id)
for new_child_product_id in new_children.keys():
new_child_product_quantity = new_children[new_child_product_id]
prod.add_child(new_child_product_id, new_child_product_quantity)
prod.save()
self.db_flush()
redir_params = ''
if 'parent_id' in self.request.POST and self.request.POST['parent_id']:
parent = Product.load(self.request.POST['parent_id'])
if not parent.has_child(prod.product_id):
parent.add_child(prod.product_id)
parent.save()
redir_params = '?is_attribute=True&parent_id=%s' % parent.product_id
inventory = str(self.request.POST.get('prod_inventory', '0'))
if inventory and str(round(float(inventory), 2)) != str(round(util.nvl(InventoryJournal.total(prod), 0), 2)):
InventoryJournal.create_new(prod, 'Inventory Adjust', inventory)
self.db_flush()
self.flash('Inventory Adjusted to %s' % inventory)
prod.clear_attributes()
for i in range(30):
attr_name = self.request.POST.get('attr_name[%d]' % i)
attr_value = self.request.POST.get('attr_value[%d]' % i)
if attr_name and attr_value:
prod.set_attr(attr_name, attr_value)
category_id = self.request.POST.get('category_id')
if category_id:
category = ProductCategory.load(category_id)
self.forbid_if(not category)
category.add_product(prod.product_id)
self.flash('Successfully saved %s.' % prod.name)
return HTTPFound('/crm/product/edit/%s%s' % (prod.product_id, redir_params))