def import_product_list(company_id, filename='/tmp/products/products.csv'):
company = Company.load(company_id)
default_campaign = company.default_campaign
products = []
with open(filename) as f:
products = f.readlines()
products = [p.rstrip() for p in products[1:]]
product_categories = {}
for pline in products:
log(pline)
(product_name, category_id, pic) = pline.split(',')
pic = pic.strip()
key = '%s%s' % (product_name.strip(), category_id.strip())
cat = ProductCategory.load(category_id.strip(), False)
prod = None
if key in product_categories:
prod = Product.load(product_categories[key][0], False)
else:
prod = Product()
prod.company = company
prod.name = product_name.strip()
prod.type = 'Parent or Child'
prod.save()
prod.flush()
product_categories[key] = [str(prod.product_id), str(cat.category_id)]
ass = Asset()
ass.fk_type = 'Product'
ass.fk_id = prod.product_id
ass.enterprise_id = company.enterprise_id
ass.name = os.path.basename(pic)
ass.extension = os.path.splitext(pic)[1]
ass.save()
ass.flush()
storage_root = Asset.get_storage_root()
if not storage_root:
storage_root = '/apps/pvs/storage'
fs_real_dir = "{root}/{reldir}".format(root=storage_root, reldir=ass.relative_dir)
util.mkdir_p(fs_real_dir)
fs_real_path = "{fs_real_dir}/{assid}{ext}".format(fs_real_dir=fs_real_dir,
assid=ass.id,
ext=ass.extension)
shutil.copyfile(pic, fs_real_path)
for pc in product_categories:
pcat = product_categories[pc]
cat = ProductCategory.load(pcat[1], False)
cat.add_product(pcat[0])
db.commit()
