def generate_cultivars(cn, l):
for d in l:
cv = Cultivar.get_or_create(d['name'], cn)
db.session.add(cv)
cv.visible = True
cv.active = True
cv.subtitle = d['subtitle']
cv.botanical_name = d['botanical_names']
cv.description = d['description']
if d['veg_info']:
if d['veg_info']['open_pollinated']:
cv.open_pollinated = True
if d['veg_info']['maturation']:
cv.maturation = d['veg_info']['maturation']
try:
cv.new_for = int(d['new_for'])
except ValueError:
pass
cv.featured = d['favorite']
cv.favorite = d['favorite']
cv.in_stock = d['packets'][0]['in_stock']
if 'organic' in cv.description.lower():
cv.organic = True
cv.taxable = d['packets'][0]['taxable']
cv.images = [download_image(i) for i in d['images']]
cv.thumbnail = cv.images[0]
cv.packets = list(generate_packets(d['packets']))
yield cv
def save_row_to_db(self, row, stream=sys.stdout):
"""Save a row from the Packets sheet to the database.
Args:
row: The number of the row to save.
stream: Optional IO stream to print messages to.
Returns:
bool: `True` if changes have been made, `False` if not.
"""
cultivar_json = self.cell(row, self.cols['Cultivar (JSON)']).value
cv_dict = json.loads(cultivar_json)
sku = self.cell(row, self.cols['SKU']).value
price = self.cell(row, self.cols['Price']).value
quantity = self.cell(row, self.cols['Quantity']).value
units = self.cell(row, self.cols['Units']).value
print('-- BEGIN editing/creating Packet with the SKU \'{0}\' from row '
'#{1}. --'.format(sku, row), file=stream)
edited = False
pkt = Packet.query.filter(Packet.sku == sku).one_or_none()
if pkt:
print('The Packet with SKU \'{0}\' has been loaded from the '
'database.'.format(pkt.sku), file=stream)
else:
edited = True
qty = Quantity.from_queryable_values(value=quantity, units=units)
if not qty:
qty = Quantity(value=quantity, units=units)
pkt = Packet(sku=sku, price=price, quantity=qty)
db.session.add(pkt)
pkt.cultivar = Cultivar.get_or_create(
name=dbify(cv_dict['Cultivar Name']),
common_name=dbify(cv_dict['Common Name']),
index=dbify(cv_dict['Index']),
stream=stream
)
print('The Packet with SKU \'{0}\' does not yet exist, so it has '
'been created.'.format(pkt.sku), file=stream)
if price != str(pkt.price):
edited = True
pkt.price = price
print('The price for Packet SKU \'{0}\' has been set to: ${1}.'
.format(pkt.sku, pkt.price), file=stream)
qty = Quantity.from_queryable_values(value=quantity, units=units)
if not qty:
qty = Quantity(value=quantity, units=units)
if qty is not pkt.quantity:
edited = True
pkt.quantity = qty
print('The quantity for the Packet SKU \'{0}\' has been set to: '
'{1} {2}'.format(pkt.sku, qty.value, qty.units), file=stream)
if edited:
db.session.flush()
print('Changes to the Packet \'{0}\' have been flushed to '
'the database.'.format(pkt.info), file=stream)
else:
print('No changes were made to the Packet \'{0}\'.'
.format(pkt.info), file=stream)
print('-- END editing/creating Packet with SKU \'{0}\' from row #{1}. '
'--'.format(pkt.sku, row), file=stream)
return edited
def save_row_to_db(self, row, stream=sys.stdout):
"""Save a row from the Cultivars sheet to the database.
Args:
row: The number of the row to save.
stream: Optional IO stream to print messages to.
Returns:
bool: `True` if changes have been made, `False` if not.
"""
index = dbify(self.cell(row, self.cols['Index']).value)
common_name = dbify(self.cell(row, self.cols['Common Name']).value)
cultivar = dbify(self.cell(row, self.cols['Cultivar Name']).value)
section = dbify(self.cell(row, self.cols['Section']).value)
if not section:
section = None
botanical_name = self.cell(row, self.cols['Botanical Name']).value
thumbnail = self.cell(row, self.cols['Thumbnail Filename']).value
description = self.cell(row, self.cols['Description']).value
synonyms = self.cell(row, self.cols['Synonyms']).value
if not synonyms:
synonyms = ''
nus = self.cell(row, self.cols['New Until']).value
if nus:
new_until = datetime.datetime.strptime(nus, '%m/%d/%Y').date()
else:
new_until = None
n_stk = self.cell(row, self.cols['In Stock']).value
if n_stk and 'true' in n_stk.lower():
in_stock = True
else:
in_stock = False
act = self.cell(row, self.cols['Active']).value
if act and 'true' in act.lower():
active = True
else:
active = False
vis = self.cell(row, self.cols['Visible']).value
if vis and 'true' in vis.lower():
visible = True
else:
visible = False
print('-- BEGIN editing/creating Cultivar \'{0}\' from row #{1}. '
'--'.format(cultivar + ' ' + common_name, row), file=stream)
edited = False
cv = Cultivar.get_or_create(name=cultivar,
index=index,
common_name=common_name,
stream=stream)
if cv.created:
edited = True
db.session.add(cv)
if section: # Section already exists if cv was not created.
sec = Section.query\
.join(CommonName, CommonName.id == Section.common_name_id)\
.join(Index, Index.id == CommonName.index_id)\
.filter(Section.name == section,
CommonName.name == common_name,
Index.name == index)\
.one_or_none()
if sec:
print('The Section \'{0}\' has been loaded from the '
'database.'.format(sec.name), file=stream)
else:
sec = Section(name=section)
sec.common_name = cv.common_name
print('The Section \'{0}\' does not yet exist, so it has '
'been created.'.format(sec.name), file=stream)
cv.section = sec
print('Section for the Cultivar \'{0}\' set to: {1}'
.format(cv.fullname, sec.name), file=stream)
if botanical_name:
if not BotanicalName.validate(botanical_name):
obn = botanical_name
words = botanical_name.strip().split(' ')
words[0] = words[0].capitalize()
botanical_name = ' '.join(words)
print('The BotanicalName \'{0}\' does not appear to be a '
'validly formatted botanical name. In an attempt to fix '
'it, it has been changed to: \'{1}\''
.format(obn, botanical_name), file=stream)
bn = BotanicalName.query\
.filter(BotanicalName.name == botanical_name)\
.one_or_none()
if bn and bn is not cv.botanical_name:
print('The BotanicalName \'{0}\' has been loaded from the '
'database.'.format(bn.name), file=stream)
elif not bn:
bn = BotanicalName(name=botanical_name)
bn.common_names.append(cv.common_name)
print('The BotanicalName \'{0}\' does not yet exist, so it '
'has been created.'.format(bn.name), file=stream)
if bn is not cv.botanical_name:
edited = True
cv.botanical_name = bn
print('BotanicalName for the Cultivar \'{0}\' set to: {1}'
.format(cv.fullname, bn.name), file=stream)
if thumbnail:
if not cv.thumbnail or cv.thumbnail.filename != thumbnail:
edited = True
tn = Image.query\
.filter(Image.filename == thumbnail)\
.one_or_none()
if tn:
print('The Image with the filename \'{0}\' has been '
'loaded from the database.'.format(tn.filename),
file=stream)
else:
tn = Image(filename=thumbnail)
print('The Image with the filename \'{0}\' does not yet '
'exist in the database, so it has been created.'
.format(tn.filename), file=stream)
cv.thumbnail = tn
print('The Image with the filename \'{0}\' has been set as '
'the thumbnail for the Cultivar \'{1}\'.'
.format(tn.filename, cv.fullname), file=stream)
if not tn.exists():
print('WARNING: The image file \'{0}\' set as the '
'thumbnail for the Cultivar \'{1}\' does not exist! '
'Please make sure you add the image file to the '
'images directory.'.format(tn.filename, cv.fullname),
file=stream)
if description != cv.description:
edited = True
if description:
cv.description = description
print('Description for the Cultivar \'{0}\' set to: {1}'
.format(cv.fullname, cv.description), file=stream)
else:
cv.description = None
print('Description for the Cultivar \'{0}\' has been cleared.'
.format(cv.fullname), file=stream)
if synonyms != cv.synonyms_string:
edited = True
cv.synonyms_string = synonyms
if synonyms:
print('Synonyms for the Cultivar \'{0}\' set to: {1}'
.format(cv.fullname, cv.synonyms_string),
file=stream)
else:
print('Synonyms for the Cultivar \'{0}\' have been cleared.'
.format(cv.fullname), file=stream)
if new_until != cv.new_until:
edited = True
if new_until:
cv.new_until = new_until
print('The Cultivar \'{0}\' has been set as new until {1}.'
.format(cv.fullname, cv.new_until.strftime('%m/%d/%Y')),
file=stream)
else:
cv.new_until = None
print('The Cultivar \'{0}\' is no longer set as new.'
.format(cv.fullname), file=stream)
if in_stock != cv.in_stock:
edited = True
cv.in_stock = in_stock
if cv.in_stock:
print('The Cultivar \'{0}\' is in stock.'.format(cv.fullname),
file=stream)
else:
print('The Cultivar \'{0}\' is out of stock.'
.format(cv.fullname), file=stream)
if active != cv.active:
edited = True
cv.active = active
if cv.active:
print('The Cultivar \'{0}\' is active.'.format(cv.fullname),
file=stream)
else:
print('The Cultivar \'{0}\' is inactive.'.format(cv.fullname),
file=stream)
if visible != cv.visible:
edited = True
cv.visible = visible
if cv.visible:
print('The Cultivar \'{0}\' will be shown on auto-generated '
'pages.'.format(cv.fullname), file=stream)
else:
print('The Cultivar \'{0}\' will not be shown on '
'auto-generated pages.'.format(cv.fullname), file=stream)
if edited:
db.session.flush()
print('Changes to the Cultivar \'{0}\' have been flushed to '
'the database.'.format(cv.fullname), file=stream)
else:
print('No changes were made to the Cultivar \'{0}\'.'
.format(cv.fullname), file=stream)
print('-- END editing/creating Cultivar \'{0}\' from row #{1}. '
'--'.format(cv.fullname, row), file=stream)
return edited
