1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
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
import pandas as pd
from pandas.tseries.offsets import BDay
import os
import datetime
import redis
from .utils import load_counterparties, load_trades, add_triggers, bump_rev, simple_serialize
from PyPDF2 import PdfFileMerger
from io import BytesIO
from Dawn import app
def cp_choices():
return Counterparties.query.order_by('name').\
with_entities(Counterparties.code, Counterparties.name)
def get_queue():
q = getattr(g, 'queue', None)
if q is None:
q = g.queue = redis.Redis(host='debian')
return q
class CounterpartyForm(ModelForm):
class Meta:
model = Counterparties
class BondForm(ModelForm):
class Meta:
model = BondDeal
include_foreign_keys = True
exclude = ['dealid', 'lastupdate'] #we generate it with a trigger at the server level
@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()
bond_form = BondForm(obj = trade)
bond_form.cp_code.choices = cp_choices()
if tradeid:
bond_form.action.choices = [('UPDATE', 'UPDATE'), ('CANCEL', 'CANCEL')]
old_ticket_name = trade.ticket
bond_form.trade_date.data = pd.datetime.today().date()
bond_form.settle_date.data = pd.datetime.today().date()+ 3 * BDay()
if bond_form.is_submitted():
if bond_form.validate():
bond_form.populate_obj(trade)
session = bond_form.get_session()
ticket_file = trade.ticket
trade.ticket = None
if not tradeid:
session.add(trade)
if ticket_file.filename == '':
if tradeid:
trade.ticket = old_ticket_name
else:
if old_ticket_name:
trade.ticket = bump_rev(old_ticket_name)
else:
trade.ticket = "{0} {1}.pdf".format(str(trade.trade_date),
trade.description)
ticket_file.save(os.path.join(app.config['TICKETS_FOLDER'],
trade.ticket))
try:
session.commit()
except IntegrityError:
print("TODO: fix this")
finally:
q = get_queue()
q.rpush('trades', simple_serialize(trade))
return redirect(url_for('list_trades'))
else:
return str(bond_form.errors)
return render_template('trade_entry.html', form=bond_form, trade_id=tradeid)
@app.route('/blotter/')
def list_trades():
trade_list = BondDeal.query.order_by(BondDeal.trade_date, BondDeal.id)
return render_template('blotter.html', trades=trade_list.all())
@app.route('/tickets/<int:tradeid>')
def download_ticket(tradeid):
trade = BondDeal.query.get(tradeid)
pdf = PdfFileMerger()
pdf.append(os.path.join(app.config['TICKETS_FOLDER'], trade.ticket))
pdf.append(os.path.join(app.config['CP_FOLDER'], trade.counterparty.instructions))
fh = BytesIO()
pdf.write(fh)
pdf.close()
fh.seek(0)
return send_file(fh, mimetype='application/pdf')
@app.route('/counterparties/<path:instr>', methods=['GET'])
@app.route('/counterparties/', defaults={'instr': None}, methods=['GET'])
def list_counterparties(instr):
if instr:
return send_from_directory(filename = instr,
directory = app.config['CP_FOLDER'],
mimetype='application/pdf')
else:
cp_list = Counterparties.query.order_by(Counterparties.name)
return render_template('counterparties.html', counterparties = cp_list.all())
@app.route('/edit_cp/<cpcode>', methods=['GET', 'POST'])
def edit_counterparty(cpcode):
cp = Counterparties.query.get(cpcode)
cp_form = CounterpartyForm(obj = cp)
old_instructions = cp.instructions
if cp_form.is_submitted():
if cp_form.validate():
cp_form.populate_obj(cp)
session = cp_form.get_session()
instructions = cp.instructions
if instructions.filename == '':
cp.instructions = old_instructions
else:
cp.instructions = cp.name + '.pdf'
instructions.save(os.path.join(app.config['CP_FOLDER'],
cp.instructions))
session.commit()
return redirect(url_for('list_counterparties'))
return render_template('edit_cp.html', form=cp_form, code=cpcode)
|
