def attach_picture_url(table, item_id, location): """ A helper function used in populator.py to upload picture to the db from web Args: table: The category which the picture belongs to item_id: The category's id number which the picture should be uploaded to location: a web url of where the picture is found Returns: None """ try: item = session.query(table).filter_by(id=item_id).one() with store_context(fs_store): item.picture.from_file(urlopen(location)) session.commit() except Exception: session.rollback() raise
def delete(dbtype, item_id): """Delete either the item or category in the database""" # check login status if 'email' not in login_session: flash('Sorry, the page you tried to access is for members only. ' 'Please sign in first.') abort(401) # query the item user wants to delete deleteItem = (session.query(eval(dbtype[0].upper()+dbtype[1:])) .filter_by(id=item_id).one()) # make sure user is authorized to delete this item if login_session['user_id'] != deleteItem.user_id: flash('You are not authorized to modify items you did not create. ' 'Please create your own item in order to modify it.') return redirect(url_for(dbtype)) if request.method == 'POST': try: session.delete(deleteItem) session.commit() # handling legacy error when delete invovled cascade-delete except IntegrityError as detail: print 'Handling run-time error: ', detail session.rollback() flash('Delete Operation Failed') return redirect(url_for('home')) if dbtype.endswith('Lot'): flash('%s Lot Deleted' % dbtype[:-3].capitalize()) return redirect(url_for(dbtype[:-3])) else: flash('%s Deleted' % dbtype.capitalize()) return redirect(url_for(dbtype)) else: pass