diff options
Diffstat (limited to 'python/Dawn')
| -rw-r--r-- | python/Dawn/views.py | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/python/Dawn/views.py b/python/Dawn/views.py index a609763c..efc375d9 100644 --- a/python/Dawn/views.py +++ b/python/Dawn/views.py @@ -1,8 +1,8 @@ from flask import (request, render_template, redirect, url_for, send_from_directory, send_file, g) from .models import ModelForm, BondDeal, Counterparties -from sqlalchemy import create_engine from sqlalchemy.exc import IntegrityError +from wtforms.fields import BooleanField import pandas as pd from pandas.tseries.offsets import BDay import os @@ -33,6 +33,7 @@ class CounterpartyForm(ModelForm): model = Counterparties class BondForm(ModelForm): + upload_globeop = BooleanField(label="Upload to globeop?") class Meta: model = BondDeal include_foreign_keys = True @@ -41,7 +42,7 @@ class BondForm(ModelForm): @app.route('/trades/<int:tradeid>', methods=['GET', 'POST']) @app.route('/trades/', defaults={'tradeid': None}, methods=['GET', 'POST']) def trade_manage(tradeid): - trade = BondDeal.query.get(tradeid) if tradeid else BondDeal() + trade = BondDeal.query.get(tradeid) if tradeid else BondDeal() if tradeid: bond_form = BondForm(obj = trade) #change default to UPDATE @@ -75,10 +76,11 @@ def trade_manage(tradeid): try: session.commit() except IntegrityError as e: - print(e) + app.logger.error(e) finally: - q = get_queue() - q.rpush('trades', simple_serialize(trade)) + if bond_form.upload_globeop.data: + q = get_queue() + q.rpush('trades', simple_serialize(trade)) return redirect(url_for('list_trades')) else: return str(bond_form.errors) |