def get_subcategory(json_data, category):
subcategory = json_data.get('group', '').replace('\\"', '"')
if category and subcategory:
subcategory_obs = Subcategory.all().filter(
'subcategory =', subcategory
)
if subcategory_obs.count():
subcategory = subcategory_obs.get()
else:
subcategory = Subcategory(
subcategory=subcategory, on_category=category
)
subcategory.put()
return subcategory
return None
def import_data(in_file='quick_export.csv'):
images_path = os.path.join(DEFAULT_PATH, 'images')
reader = csv.reader(open(os.path.join(DEFAULT_PATH, in_file)))
for i, line in enumerate(reader):
if not i:
continue
uid = line[0]
gift_imgs_path = os.path.join(images_path, uid)
try:
imgs_files = os.listdir(gift_imgs_path)
except OSError:
imgs_files = []
name = line[2].decode('utf-8')
category = line[3].decode('utf-8')
subcategory = line[4].decode('utf-8')
brand = line[5].decode('utf-8')
if line[6]:
price = float(line[6])
else:
price = 0.0
size = line[9].decode('utf-8')
if not size:
size = u''
desc = line[10].decode('utf-8')
if line[11]:
rating = float(line[11])
else:
rating = 3.0
if imgs_files:
imgs_files =\
[open(os.path.join(gift_imgs_path, img), 'rb').read() \
for img in imgs_files]
if brand:
brand_obj = Brand.all().filter('brand =', brand)
if not brand_obj.count():
brand_obj = Brand(brand=brand)
brand_obj.put()
else:
brand_obj = brand_obj[0]
else:
brand_obj = None
if category:
category_obj = Category.all().filter('category =', category)
if category_obj.count():
category_obj = category_obj[0]
else:
category_obj = Category(category=category)
category_obj.put()
else:
category_obj = None
if category and subcategory:
subcategory_obj = Subcategory.all().filter('subcategory =', subcategory).filter('on_category =', category_obj)
if subcategory_obj.count():
subcategory_obj = subcategory_obj[0]
else:
subcategory_obj = Subcategory(subcategory=subcategory, on_category=category_obj)
subcategory_obj.put()
else:
subcategory_obj = None
gift = Gift(name=name,
uid_5studio=uid,
brand=brand_obj,
category=category_obj,
subcategory=subcategory_obj,
description=desc,
rating=rating,
price=price,
size=size)
gift.put()
if gift.name:
title = gift.name.replace('"', '"')
else:
title = ''
content_type = 'image/jpeg'
for thumb in imgs_files:
thumb_img = ThumbImage()
thumb_img.add_new_thumb(blob_img=thumb, thumb_size=(700, 700, ),
title=title, content_type=content_type, is_using_pil=True)
thumb_img.add_new_thumb(blob_img=thumb, thumb_size=(400, 400, ),
title=title, content_type=content_type, is_using_pil=True)
thumb_img.add_new_thumb(blob_img=thumb, thumb_size=(200, 200, ),
title=title, content_type=content_type, is_using_pil=True)
thumb_img.add_new_thumb(blob_img=thumb, thumb_size=(100, 100, ),
title=title, content_type=content_type, is_using_pil=True)
if not gift.thumbs.count():
thumb_img.main_gift = gift
thumb_img.gift = gift
thumb_img.put()