def add():
form = AddShareForm()
if form.validate_on_submit():
ticker = form.ticker.data
quantity = form.quantity.data
dividends = form.dividends.data
portfolioid = form.portfolioid.data
if not Share.exists(ticker):
sharedata = share_data.JSONSharePrice(ticker)
sharename = sharedata['query']['results']['quote']['Name']
newshare = Share(ticker=ticker, name=sharename)
db.session.add(newshare)
# db.session.commit()
bm = Userownedshare(user=current_user.username, quantity=quantity, ticker=ticker, dividends=dividends, portfolioid=portfolioid)
db.session.add(bm)
db.session.commit()
flash("Added share '{}'".format(ticker))
return redirect(url_for('index'))
return render_template('add.html', form=form, portfolioids = Userownedshare.listportfolios())
